aipmodel 0.2.72__tar.gz → 0.2.73__tar.gz

{aipmodel-0.2.72 → aipmodel-0.2.73}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aipmodel
-Version: 0.2.72
+Version: 0.2.73
 Summary: SDK for model registration, versioning, and storage
 Home-page: https://github.com/AIP-MLOPS/model-registry
 Author: AIP MLOPS Team

{aipmodel-0.2.72 → aipmodel-0.2.73}/aipmodel/__init__.py

@@ -11,5 +11,5 @@ if _os.environ.get("AIPMODEL_UPDATE_CHECK", "").lower() in ("1", "true", "yes"):
 
 __all__ = ["MLOpsManager", "CephS3Manager"]
 
-__version__ = "0.2.72"
+__version__ = "0.2.73"
 __description__ = "SDK for model registration, versioning, and storage"

{aipmodel-0.2.72 → aipmodel-0.2.73}/aipmodel/model_registry.py

@@ -1677,7 +1677,8 @@ class MLOpsManager:
 
         Returns:
             {
-              "model_id", "model_name", "status": "ok"|"missing_files"|"orphan_files",
+              "model_id", "model_name",
+              "status": "ok"|"missing_files"|"orphan_files"|"empty",
               "versions": [{"version","path","files_exist","status"}...],
               "orphan_folders": [<S3 folders under the model not referenced by any version>],
             }
@@ -1733,6 +1734,15 @@ class MLOpsManager:
             report["status"] = "missing_files"
         elif report["orphan_folders"]:
             report["status"] = "orphan_files"
+        elif not versions_map:
+            # Registry holds the model container but NO versions and NO files.
+            # This is the shell an interrupted add leaves behind when the process
+            # is hard-killed mid-upload (pod rollout, OOM, node eviction) — the
+            # in-process rollback in add_model never runs in that case. Surface it
+            # (it was previously reported "ok") so it can be cleaned up or the add
+            # retried. Note: a version add still in flight shows its temp folder as
+            # an orphan_folder above, so it reports "orphan_files", not "empty".
+            report["status"] = "empty"
         return report
 
     def verify_all_models(self):
@@ -1813,6 +1823,23 @@ class MLOpsManager:
         self.delete_model(model_name=model_name, version=version)
         print(f"[SDK_SUCCESS] Synced: removed zombie version '{version}' of '{model_name}' from registry.")
 
+    def remove_empty_model(self, model_name=None, model_id=None):
+        """
+        Sync-repair for an EMPTY model container: the registry holds the model
+        but it has no versions and no stored files — the shell an interrupted /
+        hard-killed add leaves behind. Deletes the container so the registry is
+        clean. Refuses to run if the model actually has versions or files (so a
+        real model, or an add still in flight, is never removed by mistake).
+        """
+        report = self.verify_model_integrity(model_name=model_name, model_id=model_id)
+        if report["status"] != "empty":
+            raise ValueError(
+                f"Model '{report['model_name']}' is not empty (status={report['status']}); "
+                f"refusing to remove it. Use delete_model for a real delete."
+            )
+        self.delete_model(model_id=report["model_id"])
+        print(f"[SDK_SUCCESS] Removed empty model shell '{report['model_name']}'.")
+
     def purge_orphan_folders(self, model_name=None, model_id=None):
         """
         Sync-repair for ORPHAN folders inside a registered model: S3 folders

{aipmodel-0.2.72 → aipmodel-0.2.73}/aipmodel.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aipmodel
-Version: 0.2.72
+Version: 0.2.73
 Summary: SDK for model registration, versioning, and storage
 Home-page: https://github.com/AIP-MLOPS/model-registry
 Author: AIP MLOPS Team

{aipmodel-0.2.72 → aipmodel-0.2.73}/setup.py

@@ -31,7 +31,7 @@ version, description = read_meta()
 
 setup(
     name="aipmodel",
-    version="0.2.72",
+    version="0.2.73",
     description=description,
     author="AIP MLOPS Team",
     author_email="mohmmadweb@gmail.com",