Skip to content

Commit a6c41f3

Browse files
Add test for attachments in build_spec
1 parent e7f7486 commit a6c41f3

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/build_spec/test_build_spec.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,11 @@ def _test_build_spec(
8383
loaded_config = loaded_model.config
8484
assert "deepimagej" in loaded_config
8585

86+
attachments = loaded_model.attachments
87+
if "files" in attachments:
88+
for attached_file in attachments["files"]:
89+
assert attached_file.exists()
90+
8691

8792
def test_build_spec_pytorch(any_torch_model, tmp_path):
8893
_test_build_spec(any_torch_model, tmp_path / "model.zip", "pytorch_state_dict")

0 commit comments

Comments
 (0)