We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e7f7486 commit a6c41f3Copy full SHA for a6c41f3
tests/build_spec/test_build_spec.py
@@ -83,6 +83,11 @@ def _test_build_spec(
83
loaded_config = loaded_model.config
84
assert "deepimagej" in loaded_config
85
86
+ attachments = loaded_model.attachments
87
+ if "files" in attachments:
88
+ for attached_file in attachments["files"]:
89
+ assert attached_file.exists()
90
+
91
92
def test_build_spec_pytorch(any_torch_model, tmp_path):
93
_test_build_spec(any_torch_model, tmp_path / "model.zip", "pytorch_state_dict")
0 commit comments