Skip to content

Commit 067f24d

Browse files
committed
use ZipPath instead of zipfile.Path
1 parent 29ed684 commit 067f24d

File tree

1 file changed

+3
-10
lines changed

1 file changed

+3
-10
lines changed

bioimageio/core/cli.py

Lines changed: 3 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
from typing_extensions import assert_never
4646

4747
from bioimageio.spec import AnyModelDescr, InvalidDescr, load_description
48+
from bioimageio.spec._internal.io_basics import ZipPath
4849
from bioimageio.spec._internal.types import NotEmpty
4950
from bioimageio.spec.dataset import DatasetDescr
5051
from bioimageio.spec.model import ModelDescr, v0_4, v0_5
@@ -298,20 +299,12 @@ def _example(self):
298299

299300
for t, src in zip(input_ids, example_inputs):
300301
local = download(src).path
301-
if isinstance(local, Path):
302-
suffixes = local.suffixes
303-
elif isinstance(local, zipfile.Path):
304-
# .suffixes for zipfile.Path only added in py 3.11
305-
suffixes = PurePosixPath(local.at).suffixes
306-
else:
307-
assert_never(local)
308-
309-
dst = Path(f"{example_path}/{t}/001{''.join(suffixes)}")
302+
dst = Path(f"{example_path}/{t}/001{''.join(local.suffixes)}")
310303
dst.parent.mkdir(parents=True, exist_ok=True)
311304
inputs001.append(dst.as_posix())
312305
if isinstance(local, Path):
313306
shutil.copy(local, dst)
314-
elif isinstance(local, zipfile.Path):
307+
elif isinstance(local, ZipPath):
315308
_ = local.root.extract(local.at, path=dst)
316309
else:
317310
assert_never(local)

0 commit comments

Comments
 (0)