Skip to content

Commit 36e295c

Browse files
test_ssr fix
1 parent 6fc7511 commit 36e295c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

models/experimental/SSR/tests/test_ssr.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,10 @@
66
# Import reference and TTNN models
77
from models.experimental.SSR.reference.SSR.model.ssr import SSR, SSR_wo_conv
88
from models.experimental.SSR.tt.ssr import TTSSR, TTSSR_wo_conv
9-
from models.experimental.SSR.tests.test_upsample import create_upsample_preprocessor
9+
from models.experimental.SSR.tests.tile_refinement.test_upsample import create_upsample_preprocessor
1010
from models.experimental.SSR.tests.tile_selection.test_tile_selection import create_tile_selection_preprocessor
11-
from models.experimental.SSR.tests.test_tile_refinement import create_tile_refinement_preprocessor
12-
from models.experimental.SSR.tests.test_HAB import create_relative_position_index
11+
from models.experimental.SSR.tests.tile_refinement.test_tile_refinement import create_tile_refinement_preprocessor
12+
from models.experimental.SSR.tests.tile_refinement.test_HAB import create_relative_position_index
1313

1414
from ttnn.model_preprocessing import preprocess_model_parameters
1515
from models.utility_functions import (

0 commit comments

Comments
 (0)