diff --git a/aidge_export_arm_cortexm/export.py b/aidge_export_arm_cortexm/export.py index d1f6595e89eb327ddb67b39a21d708c9078ea3ab..be332bc6043fe853e723382f1df0516934c82bda 100644 --- a/aidge_export_arm_cortexm/export.py +++ b/aidge_export_arm_cortexm/export.py @@ -37,7 +37,8 @@ def supported_boards() -> list[str]: def gen_board_files(path:str, board:str)->None: if board not in supported_boards(): - raise ValueError(f"Board {board} is not supported, supported board are:\n\t-{'\n\t-'.join(supported_boards())}") + joint_board_str = "\n\t-".join(supported_boards()) + raise ValueError(f"Board {board} is not supported, supported board are:\n\t-{joint_board_str}") if isinstance(path, str): path = Path(path) # Create dnn directory is not exist diff --git a/aidge_export_arm_cortexm/export_registry.py b/aidge_export_arm_cortexm/export_registry.py index 46c0653ad05895d13ca0d9be7fdb5f4314f2f775..65e867335d2ab061df1257915813bc4509598e69 100644 --- a/aidge_export_arm_cortexm/export_registry.py +++ b/aidge_export_arm_cortexm/export_registry.py @@ -3,11 +3,5 @@ from aidge_core.export_utils import ExportLib class ExportLibAidgeARM(ExportLib): _name="aidge_arm" - -# TODO ugly fix for Tensor registration issue... -import aidge_core -aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.float32], - aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.float32])) - class ExportLibCMSISNN(ExportLib): _name="export_cmsisnn" diff --git a/aidge_export_arm_cortexm/unit_tests/dummy_test.py b/aidge_export_arm_cortexm/unit_tests/test_dummy.py similarity index 100% rename from aidge_export_arm_cortexm/unit_tests/dummy_test.py rename to aidge_export_arm_cortexm/unit_tests/test_dummy.py