podstack 1.3.7__tar.gz → 1.3.9__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.7 → podstack-1.3.9}/PKG-INFO +1 -1
- {podstack-1.3.7 → podstack-1.3.9}/podstack/registry/client.py +11 -5
- {podstack-1.3.7 → podstack-1.3.9}/podstack.egg-info/PKG-INFO +1 -1
- {podstack-1.3.7 → podstack-1.3.9}/pyproject.toml +1 -1
- {podstack-1.3.7 → podstack-1.3.9}/LICENSE +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/README.md +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/__init__.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/annotations.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/client.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/exceptions.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/execution.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/gpu_runner.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/models.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/notebook.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/registry/__init__.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/registry/exceptions.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/registry/experiment.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/registry/model.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack/registry/model_utils.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack.egg-info/SOURCES.txt +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack.egg-info/dependency_links.txt +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack.egg-info/requires.txt +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack.egg-info/top_level.txt +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack_gpu/__init__.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack_gpu/app.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack_gpu/exceptions.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack_gpu/image.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack_gpu/runner.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack_gpu/secret.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack_gpu/utils.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/podstack_gpu/volume.py +0 -0
- {podstack-1.3.7 → podstack-1.3.9}/setup.cfg +0 -0
|
@@ -411,16 +411,22 @@ 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
|
|
|
418
|
-
# Auto-create version 1 when run_id is provided
|
|
424
|
+
# Auto-create version 1 when run_id is provided.
|
|
425
|
+
# Only pass source when the artifact dir actually exists locally;
|
|
426
|
+
# otherwise let the backend derive it from the run's artifact URI.
|
|
419
427
|
if run_id:
|
|
420
428
|
artifact_dir = self._get_artifact_dir(run_id)
|
|
421
|
-
source =
|
|
422
|
-
if os.path.isdir(artifact_dir):
|
|
423
|
-
source = artifact_dir
|
|
429
|
+
source = artifact_dir if os.path.isdir(artifact_dir) else None
|
|
424
430
|
self.create_model_version(
|
|
425
431
|
model.id, run_id=run_id, source=source
|
|
426
432
|
)
|
|
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
|