From 7fc25a300d30b15da3818d8254567c6c001c73a9 Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Thu, 30 Nov 2023 16:33:16 +0100
Subject: [PATCH] fix removeDropout test
---
aidge_core/unit_tests/test_recipies.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/aidge_core/unit_tests/test_recipies.py b/aidge_core/unit_tests/test_recipies.py
index 235d6f683..d2406b28d 100644
--- a/aidge_core/unit_tests/test_recipies.py
+++ b/aidge_core/unit_tests/test_recipies.py
@@ -22,7 +22,7 @@ class test_recipies(unittest.TestCase):
def test_remove_dropout(self):
graph_view = aidge_core.sequential([
- aidge_core.GenericOperator("Conv", 1, 1, 1, "Conv0");
+ aidge_core.GenericOperator("Conv", 1, 0, 1, "Conv0");
aidge_core.GenericOperator("Dropout", 1, 1, 1, name="Dropout0")
])
old_nodes = graph_view.get_nodes()
--
GitLab