From ddd14df0f529844ff5c476fb0b76a3d0b0e0a6fc Mon Sep 17 00:00:00 2001 From: cmoineau <cyril.moineau@cea.fr> Date: Wed, 24 Jul 2024 13:38:18 +0000 Subject: [PATCH] Move utils in file instead of folder. --- .../{utils/__init__.py => utils.py} | 2 -- .../utils/operator_registry.py | 18 ------------------ 2 files changed, 20 deletions(-) rename aidge_core/aidge_export_aidge/{utils/__init__.py => utils.py} (92%) delete mode 100644 aidge_core/aidge_export_aidge/utils/operator_registry.py 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 ecdf2aec2..31a0d6ee7 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 dd6fbaace..000000000 --- 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()) -- GitLab