diff --git a/aidge_core/aidge_export_aidge/utils/__init__.py b/aidge_core/aidge_export_aidge/utils.py similarity index 92% rename from aidge_core/aidge_export_aidge/utils/__init__.py rename to aidge_core/aidge_export_aidge/utils.py index ecdf2aec2692a48e108d5f4ad05ed05803319525..31a0d6ee77eafb0883f2882bfaae999307a0fa4b 100644 --- a/aidge_core/aidge_export_aidge/utils/__init__.py +++ b/aidge_core/aidge_export_aidge/utils.py @@ -1,5 +1,3 @@ -from .operator_registry import * - def parse_node_input(node_inputs: list) -> list: """Parse node intputs in order to adapt the list for Jinja. diff --git a/aidge_core/aidge_export_aidge/utils/operator_registry.py b/aidge_core/aidge_export_aidge/utils/operator_registry.py deleted file mode 100644 index dd6fbaaceeba9c2125b38354eca9cc116acd29b1..0000000000000000000000000000000000000000 --- a/aidge_core/aidge_export_aidge/utils/operator_registry.py +++ /dev/null @@ -1,18 +0,0 @@ -OPERATORS_REGISTRY = {} - -def operator_register(*args): - - key_list = [arg for arg in args] - - def decorator(operator): - def wrapper(*args, **kwargs): - return operator(*args, **kwargs) - - for key in key_list: - OPERATORS_REGISTRY[key] = operator - - return wrapper - return decorator - -def supported_operators(): - return list(OPERATORS_REGISTRY.keys())