podstack 1.3.8__tar.gz → 1.3.10__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {podstack-1.3.8 → podstack-1.3.10}/PKG-INFO +1 -1
- {podstack-1.3.8 → podstack-1.3.10}/podstack/registry/client.py +16 -4
- {podstack-1.3.8 → podstack-1.3.10}/podstack.egg-info/PKG-INFO +1 -1
- {podstack-1.3.8 → podstack-1.3.10}/pyproject.toml +1 -1
- {podstack-1.3.8 → podstack-1.3.10}/LICENSE +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/README.md +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/__init__.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/annotations.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/client.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/exceptions.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/execution.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/gpu_runner.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/models.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/notebook.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/registry/__init__.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/registry/exceptions.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/registry/experiment.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/registry/model.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack/registry/model_utils.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack.egg-info/SOURCES.txt +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack.egg-info/dependency_links.txt +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack.egg-info/requires.txt +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack.egg-info/top_level.txt +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack_gpu/__init__.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack_gpu/app.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack_gpu/exceptions.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack_gpu/image.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack_gpu/runner.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack_gpu/secret.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack_gpu/utils.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/podstack_gpu/volume.py +0 -0
- {podstack-1.3.8 → podstack-1.3.10}/setup.cfg +0 -0
|
@@ -411,7 +411,13 @@ class RegistryClient:
|
|
|
411
411
|
if tags:
|
|
412
412
|
body["tags"] = tags
|
|
413
413
|
|
|
414
|
-
|
|
414
|
+
try:
|
|
415
|
+
data = self._request("POST", "/models", json=body)
|
|
416
|
+
except RegistryError as e:
|
|
417
|
+
if "already exists" in str(e).lower():
|
|
418
|
+
data = self._request("GET", f"/models/{name}")
|
|
419
|
+
else:
|
|
420
|
+
raise
|
|
415
421
|
model_data = data.get("model", data)
|
|
416
422
|
model = RegisteredModel.from_dict(model_data, client=self)
|
|
417
423
|
|
|
@@ -421,9 +427,15 @@ class RegistryClient:
|
|
|
421
427
|
if run_id:
|
|
422
428
|
artifact_dir = self._get_artifact_dir(run_id)
|
|
423
429
|
source = artifact_dir if os.path.isdir(artifact_dir) else None
|
|
424
|
-
|
|
425
|
-
|
|
426
|
-
|
|
430
|
+
try:
|
|
431
|
+
self.create_model_version(
|
|
432
|
+
model.id, run_id=run_id, source=source
|
|
433
|
+
)
|
|
434
|
+
except RegistryError as e:
|
|
435
|
+
# If a version already exists for this model, that's fine —
|
|
436
|
+
# the caller can query list_model_versions() to see what exists.
|
|
437
|
+
if "already exists" not in str(e).lower():
|
|
438
|
+
raise
|
|
427
439
|
|
|
428
440
|
return model
|
|
429
441
|
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|