Skip to content

Commit e16a2b2

Browse files
committed
changed export-dir to Path object
1 parent 85f8a5f commit e16a2b2

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/test_ml_funcs.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -121,13 +121,12 @@ def test_run_segmentation():
121121

122122

123123
def test_download_model(tmpdir):
124-
print(tmpdir)
124+
tmpdir = Path(tmpdir)
125125
export_dir = Path(tmpdir / 'export')
126126
export_dir.mkdir(parents=True, exist_ok=True)
127127

128128
ml_model = sa.search_models(include_global=True)[0]
129-
Path.rmdir(Path(tmpdir) / './export')
130-
model = sa.download_model(ml_model, './export')
129+
model = sa.download_model(ml_model, tmpdir / 'export')
131130
assert model['name']
132-
model = sa.download_model(ml_model['name'], './export')
131+
model = sa.download_model(ml_model['name'], tmpdir / 'export')
133132
assert model['name']

0 commit comments

Comments
 (0)