gpu-dev 0.4.0__tar.gz → 0.4.1__tar.gz

{gpu_dev-0.4.0 → gpu_dev-0.4.1}/PKG-INFO

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

{gpu_dev-0.4.0 → gpu_dev-0.4.1}/cli-tools/gpu-dev-cli/gpu_dev.egg-info/PKG-INFO

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

{gpu_dev-0.4.0 → gpu_dev-0.4.1}/cli-tools/gpu-dev-cli/gpu_dev_cli/reservations.py

@@ -1364,10 +1364,18 @@ class ReservationManager:
                             time.sleep(2)
                             continue
 
+                        # Check for error written by Lambda
+                        add_user_error = reservation.get("add_user_error", "")
+                        if add_user_error:
+                            live.stop()
+                            console.print(
+                                f"[red]❌ {add_user_error}[/red]"
+                            )
+                            return False
+
                         current_secondary_users = reservation.get(
                             "secondary_users", [])
 
-                        # Check if the user has been added
                         if github_username in current_secondary_users:
                             live.stop()
                             console.print(

{gpu_dev-0.4.0 → gpu_dev-0.4.1}/pyproject.toml

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

{gpu_dev-0.4.0 → gpu_dev-0.4.1}/terraform-gpu-devservers/lambda/reservation_processor/index.py

@@ -8153,12 +8153,14 @@ def process_add_user_action(record: dict[str, Any]) -> bool:
             logger.info(
                 f"Successfully added user {github_username} to reservation {full_reservation_id}"
             )
+            # Clear any previous error
+            update_reservation_fields(full_reservation_id, add_user_error="")
             return True
         else:
-            logger.error(
-                f"Failed to add user {github_username} to reservation {full_reservation_id}"
-            )
-            return False  # Retry on failure
+            error_msg = f"Failed to add user {github_username} - check that the GitHub user has public SSH keys at github.com/{github_username}.keys"
+            logger.error(error_msg)
+            update_reservation_fields(full_reservation_id, add_user_error=error_msg)
+            return True  # Don't retry - write error for CLI to detect
 
     except Exception as e:
         logger.error(f"Error processing add user action: {str(e)}")

{gpu_dev-0.4.0 → gpu_dev-0.4.1}/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.4.0"
+      LAMBDA_VERSION                     = "0.4.1"
       MIN_CLI_VERSION                    = "0.4.0"
       DISK_CONTENTS_BUCKET               = aws_s3_bucket.disk_contents.bucket
       OPERATIONS_TABLE                   = aws_dynamodb_table.operations.name