diff --git a/backends/arm/test/misc/test_debug_feats.py b/backends/arm/test/misc/test_debug_feats.py index 156d646cb6e..690549d7174 100644 --- a/backends/arm/test/misc/test_debug_feats.py +++ b/backends/arm/test/misc/test_debug_feats.py @@ -9,7 +9,6 @@ import shutil import tempfile import unittest -from importlib.metadata import version import torch from executorch.backends.arm.test import common @@ -193,16 +192,15 @@ def test_collate_tosa_BI_tests(self): .to_edge_transform_and_lower() .to_executorch() ) - et_version = version("executorch") # test that the output directory is created and contains the expected files assert os.path.exists( "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests" ) assert os.path.exists( - f"test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests/output_tag6_TOSA-0.80+BI_{et_version}.tosa" + "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests/output_tag6.tosa" ) assert os.path.exists( - f"test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests/desc_tag6_TOSA-0.80+BI_{et_version}.json" + "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests/desc_tag6.json" ) os.environ.pop("TOSA_TESTCASES_BASE_PATH") diff --git a/backends/arm/tosa_backend.py b/backends/arm/tosa_backend.py index 18c39d133b7..f64b3681098 100644 --- a/backends/arm/tosa_backend.py +++ b/backends/arm/tosa_backend.py @@ -12,7 +12,6 @@ # import logging import os -from importlib.metadata import version from typing import cast, final, List import serializer.tosa_serializer as ts # type: ignore @@ -126,12 +125,10 @@ def preprocess( # noqa: C901 if artifact_path: tag = _get_first_delegation_tag(graph_module) - et_version = version("executorch") dbg_tosa_dump( tosa_graph, artifact_path, - suffix="{}".format(f"_{tag}" if tag else "") - + (f"_{tosa_spec}" + (f"_{et_version}")), + suffix="{}".format(f"_{tag}" if tag else ""), ) # Serialize and return the TOSA flatbuffer.