File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed
tests/unitary/default_setup/model Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -100,20 +100,19 @@ def test_downaload_small_artifact(self):
100
100
expected_artifact_bytes_content
101
101
)
102
102
with tempfile .TemporaryDirectory () as tmp_dir :
103
- target_dir = os .path .join (tmp_dir , "model_artifacts/" )
104
103
SmallArtifactDownloader (
105
104
dsc_model = self .mock_dsc_model ,
106
- target_dir = target_dir ,
105
+ target_dir = tmp_dir ,
107
106
force_overwrite = True ,
108
107
).download ()
109
108
110
109
self .mock_dsc_model .get_model_artifact_content .assert_called ()
111
110
112
111
test_files = list (
113
- glob .iglob (os .path .join (target_dir , "**" ), recursive = True )
112
+ glob .iglob (os .path .join (tmp_dir , "**" ), recursive = True )
114
113
)
115
114
expected_files = [
116
- os .path .join (target_dir , file_name )
115
+ os .path .join (tmp_dir , file_name )
117
116
for file_name in ["" , "runtime.yaml" , "score.py" ]
118
117
]
119
118
assert sorted (test_files ) == sorted (expected_files )
You can’t perform that action at this time.
0 commit comments