Operator register now return an object instead of a function.

aidge_core/export_utils/export_registry.py

@@ -111,8 +111,9 @@ def operator_register(lib: ExportLib, key: str, *args):
     """Helper decorator to register an :py:class:`ExportNode` to an :py:class:`ExportLib`
     """
     def decorator(operator):
-        def wrapper(*args, **kwargs):
-            return operator(*args, **kwargs)
+        class Wrapper(operator):
+            def __init__(self, *args, **kwargs):
+                return operator(*args, **kwargs)
         lib.add_export_node(key, operator)
-        return wrapper
+        return Wrapper
     return decorator