gpu-dev 0.5.7__tar.gz → 0.5.8__tar.gz

{gpu_dev-0.5.7 → gpu_dev-0.5.8}/PKG-INFO

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

{gpu_dev-0.5.7 → gpu_dev-0.5.8}/cli-tools/gpu-dev-cli/gpu_dev.egg-info/PKG-INFO

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

{gpu_dev-0.5.7 → gpu_dev-0.5.8}/pyproject.toml

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

{gpu_dev-0.5.7 → gpu_dev-0.5.8}/terraform-gpu-devservers/lambda/availability_updater/index.py

@@ -24,6 +24,45 @@ RESERVATIONS_TABLE = os.environ.get("RESERVATIONS_TABLE", "pytorch-gpu-dev-reser
 SUPPORTED_GPU_TYPES = json.loads(os.environ["SUPPORTED_GPU_TYPES"])
 
 
+def _parse_expires_at(value):
+    """Parse the reservations table's `expires_at` field to a unix epoch (int).
+
+    DDB stores it as either an ISO-8601 datetime string ("2026-05-02T00:12:03.674845") OR
+    occasionally a numeric epoch — handle both. Returns None if unparseable.
+    """
+    if value is None:
+        return None
+    # Numeric (Decimal/int/float) → epoch seconds directly.
+    if not isinstance(value, str):
+        try:
+            return int(float(value))
+        except (ValueError, TypeError):
+            return None
+    s = value.strip()
+    if not s:
+        return None
+    # ISO-8601 first (the actual production format).
+    try:
+        from datetime import datetime, timezone
+        # `fromisoformat` accepts microseconds; tolerate optional 'Z' suffix.
+        if s.endswith("Z"):
+            s2 = s[:-1] + "+00:00"
+        else:
+            s2 = s
+        dt = datetime.fromisoformat(s2)
+        if dt.tzinfo is None:
+            # Convention in this codebase: timestamps written via datetime.utcnow().isoformat() are UTC.
+            dt = dt.replace(tzinfo=timezone.utc)
+        return int(dt.timestamp())
+    except (ValueError, TypeError):
+        pass
+    # Numeric-as-string fallback.
+    try:
+        return int(float(s))
+    except (ValueError, TypeError):
+        return None
+
+
 def get_gpu_resource_name(gpu_type: str) -> str:
     return SUPPORTED_GPU_TYPES.get(gpu_type, {}).get("k8s_resource", "nvidia.com/gpu")
 
@@ -542,9 +581,8 @@ def compute_size_etas(v1, gpu_type, node_label_value, resource_name, gpus_per_in
             continue
         if pod_name not in pod_to_info:
             continue
-        try:
-            ts = int(float(expires_at))
-        except (ValueError, TypeError):
+        ts = _parse_expires_at(expires_at)
+        if ts is None:
             continue
         node_name, gpus = pod_to_info[pod_name]
         node_state[node_name]["expirations"].append((ts, gpus))

{gpu_dev-0.5.7 → gpu_dev-0.5.8}/terraform-gpu-devservers/lambda.tf

@@ -180,7 +180,7 @@ 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.5.7"
+      LAMBDA_VERSION                     = "0.5.8"
       MIN_CLI_VERSION                    = "0.5.5"
       DISK_CONTENTS_BUCKET               = aws_s3_bucket.disk_contents.bucket
       OPERATIONS_TABLE                   = aws_dynamodb_table.operations.name