File tree 2 files changed +3
-8
lines changed
2 files changed +3
-8
lines changed Original file line number Diff line number Diff line change 9
9
import shutil
10
10
import tempfile
11
11
import unittest
12
- from importlib .metadata import version
13
12
14
13
import torch
15
14
from executorch .backends .arm .test import common
@@ -193,16 +192,15 @@ def test_collate_tosa_BI_tests(self):
193
192
.to_edge_transform_and_lower ()
194
193
.to_executorch ()
195
194
)
196
- et_version = version ("executorch" )
197
195
# test that the output directory is created and contains the expected files
198
196
assert os .path .exists (
199
197
"test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests"
200
198
)
201
199
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"
203
201
)
204
202
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"
206
204
)
207
205
208
206
os .environ .pop ("TOSA_TESTCASES_BASE_PATH" )
Original file line number Diff line number Diff line change 12
12
#
13
13
import logging
14
14
import os
15
- from importlib .metadata import version
16
15
from typing import cast , final , List
17
16
18
17
import serializer .tosa_serializer as ts # type: ignore
@@ -123,12 +122,10 @@ def preprocess( # noqa: C901
123
122
124
123
if artifact_path :
125
124
tag = _get_first_delegation_tag (graph_module )
126
- et_version = version ("executorch" )
127
125
dbg_tosa_dump (
128
126
tosa_graph ,
129
127
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 "" ),
132
129
)
133
130
134
131
# Serialize and return the TOSA flatbuffer.
You can’t perform that action at this time.
0 commit comments