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 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
@@ -126,12 +125,10 @@ def preprocess( # noqa: C901
126
125
127
126
if artifact_path :
128
127
tag = _get_first_delegation_tag (graph_module )
129
- et_version = version ("executorch" )
130
128
dbg_tosa_dump (
131
129
tosa_graph ,
132
130
artifact_path ,
133
- suffix = "{}" .format (f"_{ tag } " if tag else "" )
134
- + (f"_{ tosa_spec } " + (f"_{ et_version } " )),
131
+ suffix = "{}" .format (f"_{ tag } " if tag else "" ),
135
132
)
136
133
137
134
# Serialize and return the TOSA flatbuffer.
You can’t perform that action at this time.
0 commit comments