File tree Expand file tree Collapse file tree 2 files changed +3
-8
lines changed Expand file tree Collapse file tree 2 files changed +3
-8
lines changed Original file line number Diff line number Diff line change 99import shutil
1010import tempfile
1111import unittest
12- from importlib .metadata import version
1312
1413import torch
1514from executorch .backends .arm .test import common
@@ -193,16 +192,15 @@ def test_collate_tosa_BI_tests(self):
193192 .to_edge_transform_and_lower ()
194193 .to_executorch ()
195194 )
196- et_version = version ("executorch" )
197195 # test that the output directory is created and contains the expected files
198196 assert os .path .exists (
199197 "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests"
200198 )
201199 assert os .path .exists (
202- f "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests/output_tag6_TOSA-0.80+BI_ { et_version } .tosa"
200+ "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests/output_tag6 .tosa"
203201 )
204202 assert os .path .exists (
205- f "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests/desc_tag6_TOSA-0.80+BI_ { et_version } .json"
203+ "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests/desc_tag6 .json"
206204 )
207205
208206 os .environ .pop ("TOSA_TESTCASES_BASE_PATH" )
Original file line number Diff line number Diff line change 1212#
1313import logging
1414import os
15- from importlib .metadata import version
1615from typing import cast , final , List
1716
1817import serializer .tosa_serializer as ts # type: ignore
@@ -123,12 +122,10 @@ def preprocess( # noqa: C901
123122
124123 if artifact_path :
125124 tag = _get_first_delegation_tag (graph_module )
126- et_version = version ("executorch" )
127125 dbg_tosa_dump (
128126 tosa_graph ,
129127 artifact_path ,
130- suffix = "{}" .format (f"_{ tag } " if tag else "" )
131- + (f"_{ tosa_spec } " + (f"_{ et_version } " )),
128+ suffix = "{}" .format (f"_{ tag } " if tag else "" ),
132129 )
133130
134131 # Serialize and return the TOSA flatbuffer.
You can’t perform that action at this time.
0 commit comments