podstack 1.3.15__tar.gz → 1.3.17__tar.gz

{podstack-1.3.15 → podstack-1.3.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: podstack
-Version: 1.3.15
+Version: 1.3.17
 Summary: Official Python SDK for Podstack GPU Notebook Platform
 Author-email: Podstack <support@podstack.ai>
 License-Expression: MIT

{podstack-1.3.15 → podstack-1.3.17}/podstack/registry/__init__.py

@@ -347,25 +347,47 @@ def load_model(model_name: str, version: int = None, stage: str = None, framewor
 def log_dataset(
     name: str,
     path: str = None,
+    source_path: str = None,
+    df=None,
+    context: str = "training",
+    split: str = None,
+    digest: str = None,
+    source_type: str = "local",
+    tags: dict = None,
     version: str = None,
     description: str = None,
-    digest: str = None,
     num_rows: int = None,
-    num_features: int = None
+    num_features: int = None,
 ):
     """
-    Log dataset metadata for the active run.
+    Log a dataset to the active run.
 
     Args:
         name: Dataset name.
-        path: Dataset path or URI.
-        version: Dataset version string.
-        description: Dataset description.
-        digest: Hash/digest of the dataset.
-        num_rows: Number of rows/samples.
-        num_features: Number of features/columns.
-    """
-    _get_client().log_dataset(name, path, version, description, digest, num_rows, num_features)
+        path: Local file path or URI.
+        source_path: Alias for path.
+        df: Optional pandas DataFrame for auto schema/profile computation.
+        context: One of "training", "validation", "test".
+        split: Alias for context ("train", "val", "test").
+        digest: SHA-256 hex digest (auto-computed from path if not provided).
+        source_type: One of "local", "s3", "gcs", "url".
+        tags: Optional dict of string tags.
+    """
+    return _get_client().log_dataset(
+        name=name,
+        path=path,
+        source_path=source_path,
+        df=df,
+        context=context,
+        split=split,
+        digest=digest,
+        source_type=source_type,
+        tags=tags,
+        version=version,
+        description=description,
+        num_rows=num_rows,
+        num_features=num_features,
+    )
 
 
 def compare_runs(run_ids: list, metric_keys: list = None) -> dict:

{podstack-1.3.15 → podstack-1.3.17}/podstack/registry/client.py

@@ -977,8 +977,10 @@ class RegistryClient:
         self,
         name: str,
         path: str = None,
+        source_path: str = None,
         df=None,
         context: str = "training",
+        split: str = None,
         digest: str = None,
         source_type: str = "local",
         tags: dict = None,
@@ -1021,6 +1023,13 @@ class RegistryClient:
         if not self._active_run:
             raise NoActiveRunError()
 
+        # Aliases
+        if source_path and not path:
+            path = source_path
+        if split:
+            _split_map = {"train": "training", "val": "validation", "valid": "validation", "test": "test"}
+            context = _split_map.get(split.lower(), split)
+
         schema: Dict[str, str] = {}
         profile: Dict[str, Any] = {}
 

{podstack-1.3.15 → podstack-1.3.17}/podstack/registry/experiment.py

@@ -98,6 +98,26 @@ class Dataset:
     tags: Dict[str, str] = field(default_factory=dict)
     created_at: Optional[str] = None
 
+    def __getitem__(self, key: str):
+        # Convenience aliases
+        if key == "rows":
+            return self.profile.get("num_rows")
+        if key == "num_features":
+            return self.profile.get("num_features")
+        if key == "path":
+            return self.source
+        try:
+            return getattr(self, key)
+        except AttributeError:
+            raise KeyError(key)
+
+    def get(self, key: str, default=None):
+        try:
+            val = self[key]
+            return val if val is not None else default
+        except KeyError:
+            return default
+
     @classmethod
     def from_dict(cls, data: Dict[str, Any]) -> "Dataset":
         """Create a Dataset from a dict."""

{podstack-1.3.15 → podstack-1.3.17}/podstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: podstack
-Version: 1.3.15
+Version: 1.3.17
 Summary: Official Python SDK for Podstack GPU Notebook Platform
 Author-email: Podstack <support@podstack.ai>
 License-Expression: MIT

{podstack-1.3.15 → podstack-1.3.17}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "podstack"
-version = "1.3.15"
+version = "1.3.17"
 description = "Official Python SDK for Podstack GPU Notebook Platform"
 readme = "README.md"
 license = "MIT"