gpu-dev 0.6.3__tar.gz → 0.6.4__tar.gz

{gpu_dev-0.6.3 → gpu_dev-0.6.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gpu-dev
-Version: 0.6.3
+Version: 0.6.4
 Summary: CLI tool for PyTorch GPU developer server reservations
 Author: PyTorch Team
 Requires-Python: >=3.10

{gpu_dev-0.6.3 → gpu_dev-0.6.4}/cli-tools/gpu-dev-cli/gpu_dev.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gpu-dev
-Version: 0.6.3
+Version: 0.6.4
 Summary: CLI tool for PyTorch GPU developer server reservations
 Author: PyTorch Team
 Requires-Python: >=3.10

{gpu_dev-0.6.3 → gpu_dev-0.6.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "gpu-dev"
-version = "0.6.3"
+version = "0.6.4"
 description = "CLI tool for PyTorch GPU developer server reservations"
 authors = [{name = "PyTorch Team"}]
 readme = "cli-tools/gpu-dev-cli/README.md"

{gpu_dev-0.6.3 → gpu_dev-0.6.4}/sdk/python/src/gpu_dev/_backend/aws.py

@@ -189,13 +189,17 @@ class AwsBackend:
                 return self._item_to_info(item)
             return None
 
-        resp = self._reservations.query(
-            IndexName="UserIndex",
-            KeyConditionExpression="user_id = :uid",
-            FilterExpression="begins_with(reservation_id, :rid)",
-            ExpressionAttributeValues={":uid": user_id, ":rid": reservation_id},
-        )
+        query_kwargs = {
+            "IndexName": "UserIndex",
+            "KeyConditionExpression": "user_id = :uid",
+            "FilterExpression": "begins_with(reservation_id, :rid)",
+            "ExpressionAttributeValues": {":uid": user_id, ":rid": reservation_id},
+        }
+        resp = self._reservations.query(**query_kwargs)
         items = resp.get("Items", [])
+        while not items and "LastEvaluatedKey" in resp:
+            resp = self._reservations.query(**query_kwargs, ExclusiveStartKey=resp["LastEvaluatedKey"])
+            items = resp.get("Items", [])
         if len(items) == 1:
             return self._item_to_info(items[0])
         return None

{gpu_dev-0.6.3 → gpu_dev-0.6.4}/sdk/python/src/gpu_dev/_sync/sandbox.py

@@ -136,11 +136,6 @@ class Sandbox:
         """Owner's user ID."""
         return self._info.user_id
 
-    @property
-    def node_ip(self) -> str | None:
-        """Node public IP."""
-        return self._info.node_ip
-
     @property
     def is_active(self) -> bool:
         """Whether the sandbox is running and ready for commands."""
@@ -322,6 +317,62 @@ class Sandbox:
         except Exception:
             return []
 
+    def timing(self) -> list[dict[str, str | float]]:
+        """Get a performance breakdown of reservation setup.
+
+        Computes time deltas between status transitions to show
+        where time was spent.
+
+        Returns:
+            List of ``{"step": "...", "duration": seconds, "timestamp": "..."}`` dicts.
+
+        Example::
+
+            for step in sandbox.timing():
+                print(f"{step['duration']:5.1f}s  {step['step']}")
+        """
+        from datetime import datetime
+
+        entries = self.logs()
+        if not entries:
+            return []
+
+        result = []
+        prev_ts = None
+        for entry in entries:
+            ts_str = entry.get("timestamp", "")
+            msg = entry.get("message", "")
+            try:
+                ts = datetime.fromisoformat(ts_str)
+            except (ValueError, TypeError):
+                continue
+            if prev_ts:
+                delta = (ts - prev_ts).total_seconds()
+                if delta >= 0.05:
+                    result.append({
+                        "step": msg,
+                        "duration": round(delta, 2),
+                        "timestamp": ts_str,
+                    })
+            else:
+                result.append({
+                    "step": msg,
+                    "duration": 0.0,
+                    "timestamp": ts_str,
+                })
+            prev_ts = ts
+
+        if result and len(entries) >= 2:
+            first_ts = datetime.fromisoformat(entries[0]["timestamp"])
+            last_ts = datetime.fromisoformat(entries[-1]["timestamp"])
+            result.append({
+                "step": "TOTAL",
+                "duration": round((last_ts - first_ts).total_seconds(), 2),
+                "timestamp": "",
+            })
+
+        return result
+
     def pod_logs(self, lines: int = 50) -> str:
         """Fetch container stdout from the running pod via SSH.
 

{gpu_dev-0.6.3 → gpu_dev-0.6.4}/terraform-gpu-devservers/lambda.tf

@@ -186,8 +186,8 @@ resource "aws_lambda_function" "reservation_processor" {
       HOSTED_ZONE_ID                     = local.effective_domain_name != "" ? local.hosted_zone_id : ""
       SSH_DOMAIN_MAPPINGS_TABLE          = local.effective_domain_name != "" ? aws_dynamodb_table.ssh_domain_mappings.name : ""
       SSL_CERTIFICATE_ARN                = local.effective_domain_name != "" ? aws_acm_certificate.wildcard[0].arn : ""
-      LAMBDA_VERSION                     = "0.6.0"
-      MIN_CLI_VERSION                    = "0.6.0"
+      LAMBDA_VERSION                     = "0.6.4"
+      MIN_CLI_VERSION                    = "0.6.4"
       # Comma-separated GPU types that require --spot flag, or "all" for every type.
       # Empty = no spot types (on-demand / reserved). Set per-workspace.
       ASG_NAME_PREFIX                    = "${var.prefix}-gpu-nodes"