From 4b7830823c9dd45a0e79ef0962960d7530992cbf Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Mon, 23 Oct 2023 10:49:43 +0200
Subject: [PATCH] Hotfix for OperatorImpl binding test
---
aidge_core/unit_tests/test_operator_binding.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/aidge_core/unit_tests/test_operator_binding.py b/aidge_core/unit_tests/test_operator_binding.py
index c7279afed..7bd1e730a 100644
--- a/aidge_core/unit_tests/test_operator_binding.py
+++ b/aidge_core/unit_tests/test_operator_binding.py
@@ -107,8 +107,8 @@ class test_operator_binding(unittest.TestCase):
class PythonCustomImpl(aidge_core.OperatorImpl):
"""Dummy implementation to test that C++ call python code
"""
- def __init__(self):
- aidge_core.OperatorImpl.__init__(self) # Recquired to avoid type error !
+ def __init__(self, op: aidge_core.Operator):
+ aidge_core.OperatorImpl.__init__(self, op) # Recquired to avoid type error !
self.idx = 0
def forward(self):
@@ -117,8 +117,8 @@ class test_operator_binding(unittest.TestCase):
self.idx += 1
generic_node = aidge_core.GenericOperator("Relu", 1, 1, 1, name="myReLu")
- customImpl = PythonCustomImpl()
generic_op = generic_node.get_operator()
+ customImpl = PythonCustomImpl(generic_op)
generic_op.forward() # Do nothing, no implementation set
generic_op.set_impl(customImpl)
--
GitLab