gpu-dev 0.5.9__tar.gz → 0.5.11__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (115) hide show
  1. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/PKG-INFO +1 -1
  2. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev.egg-info/PKG-INFO +1 -1
  3. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/pyproject.toml +1 -1
  4. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/reservation_processor/index.py +3 -3
  5. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda.tf +2 -2
  6. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/.github/workflows/no-gitlinks.yml +0 -0
  7. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/.github/workflows/publish.yml +0 -0
  8. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/.gitignore +0 -0
  9. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/CLAUDE.md +0 -0
  10. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/PROGRESS.md +0 -0
  11. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/PR_DESCRIPTION.md +0 -0
  12. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/TODO.md +0 -0
  13. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/admin/README.md +0 -0
  14. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/admin/generate_stats.py +0 -0
  15. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/admin/requirements.txt +0 -0
  16. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/README.md +0 -0
  17. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/ZERO_CONFIG_SETUP.md +0 -0
  18. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev.egg-info/SOURCES.txt +0 -0
  19. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev.egg-info/dependency_links.txt +0 -0
  20. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev.egg-info/entry_points.txt +0 -0
  21. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev.egg-info/requires.txt +0 -0
  22. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev.egg-info/top_level.txt +0 -0
  23. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev_cli/__init__.py +0 -0
  24. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev_cli/auth.py +0 -0
  25. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev_cli/cli.py +0 -0
  26. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev_cli/config.py +0 -0
  27. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev_cli/disks.py +0 -0
  28. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev_cli/interactive.py +0 -0
  29. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev_cli/name_generator.py +0 -0
  30. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev_cli/reservations.py +0 -0
  31. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/gpu_dev_cli/ssh_proxy.py +0 -0
  32. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/gpu-dev-cli/minimal-iam-policy.json +0 -0
  33. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/cli-tools/scripts/clear_stale_disk_locks.py +0 -0
  34. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/docs/USER_GUIDE.md +0 -0
  35. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/docs/devgpu-features.html +0 -0
  36. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/docs/docker-mark-blue.svg +0 -0
  37. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/docs/icons8-cursor-ai.svg +0 -0
  38. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/post.md +0 -0
  39. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/setup.cfg +0 -0
  40. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/.claude/skills/deploy.md +0 -0
  41. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/.terraform.lock.hcl +0 -0
  42. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/README.md +0 -0
  43. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/alb.tf +0 -0
  44. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/availability.tf +0 -0
  45. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/backend.tf +0 -0
  46. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/.dockerignore +0 -0
  47. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/Dockerfile +0 -0
  48. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/backup-dotfiles +0 -0
  49. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/bash_profile +0 -0
  50. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/bashrc +0 -0
  51. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/bashrc_ext +0 -0
  52. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/build-with-efa.sh +0 -0
  53. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/dotfiles-shutdown-handler +0 -0
  54. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/list-dotfile-versions +0 -0
  55. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/motd_script +0 -0
  56. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/nproc_wrapper +0 -0
  57. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/profile +0 -0
  58. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/restore-dotfiles +0 -0
  59. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/restore-dotfiles-version +0 -0
  60. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/setup-dotfiles-persistence +0 -0
  61. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/shell_env +0 -0
  62. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/ssh_config +0 -0
  63. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/zprofile +0 -0
  64. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/zshrc +0 -0
  65. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker/zshrc_ext +0 -0
  66. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker-build.tf +0 -0
  67. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker-example/Dockerfile +0 -0
  68. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/docker-example/hello.txt +0 -0
  69. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/ecr.tf +0 -0
  70. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/efs.tf +0 -0
  71. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/eks.tf +0 -0
  72. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/expiry.tf +0 -0
  73. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/git-cache.tf +0 -0
  74. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/kubernetes.tf +0 -0
  75. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/availability_updater/index.py +0 -0
  76. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/availability_updater/requirements.txt +0 -0
  77. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/migration/tag_largest_snapshots.py +0 -0
  78. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/reservation_expiry/index.py +0 -0
  79. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/reservation_expiry/requirements.txt +0 -0
  80. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/reservation_processor/buildkit_job.py +0 -0
  81. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/reservation_processor/requirements.txt +0 -0
  82. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/shared/__init__.py +0 -0
  83. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/shared/alb_utils.py +0 -0
  84. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/shared/dns_utils.py +0 -0
  85. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/shared/k8s_client.py +0 -0
  86. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/shared/k8s_resource_tracker.py +0 -0
  87. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/shared/requirements.txt +0 -0
  88. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/lambda/shared/snapshot_utils.py +0 -0
  89. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/main.tf +0 -0
  90. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/migrations/backfill_snapshot_contents.py +0 -0
  91. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/migrations/backfill_snapshot_contents.py.bak +0 -0
  92. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/migrations/check_snapshots.py +0 -0
  93. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/migrations/migrate_disks_to_named.py +0 -0
  94. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/migrations/run_backfill.sh +0 -0
  95. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/monitoring.tf +0 -0
  96. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/outputs.tf +0 -0
  97. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/pyproject.toml +0 -0
  98. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/queue.tf +0 -0
  99. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/route53.tf +0 -0
  100. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/s3-disk-contents.tf +0 -0
  101. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/scripts/CLEANUP_GUIDE.md +0 -0
  102. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/scripts/detect_empty_volumes.sh +0 -0
  103. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/scripts/ec2_avail_probe.sh +0 -0
  104. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/scripts/inspect_user_data.sh +0 -0
  105. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/ssh-proxy/Dockerfile +0 -0
  106. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/ssh-proxy/proxy.py +0 -0
  107. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/ssh-proxy/requirements.txt +0 -0
  108. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/ssh-proxy-service.tf +0 -0
  109. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/ssh-proxy.tf +0 -0
  110. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/switch-to.sh +0 -0
  111. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/templates/al2023-cpu-user-data.sh +0 -0
  112. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/templates/al2023-user-data.sh +0 -0
  113. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/templates/user-data-self-managed.sh +0 -0
  114. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/templates/user-data.sh +0 -0
  115. {gpu_dev-0.5.9 → gpu_dev-0.5.11}/terraform-gpu-devservers/variables.tf +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: gpu-dev
3
- Version: 0.5.9
3
+ Version: 0.5.11
4
4
  Summary: CLI tool for PyTorch GPU developer server reservations
5
5
  Author: PyTorch Team
6
6
  Requires-Python: >=3.10
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: gpu-dev
3
- Version: 0.5.9
3
+ Version: 0.5.11
4
4
  Summary: CLI tool for PyTorch GPU developer server reservations
5
5
  Author: PyTorch Team
6
6
  Requires-Python: >=3.10
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "gpu-dev"
7
- version = "0.5.9"
7
+ version = "0.5.11"
8
8
  description = "CLI tool for PyTorch GPU developer server reservations"
9
9
  authors = [{name = "PyTorch Team"}]
10
10
  readme = "cli-tools/gpu-dev-cli/README.md"
@@ -323,12 +323,12 @@ def get_target_az_for_reservation(gpu_type, gpus_requested):
323
323
  return target_az
324
324
 
325
325
  logger.warning(f"No ready/schedulable {gpu_type} nodes found in cluster")
326
- return None
326
+ return None, None
327
327
 
328
328
  except Exception as e:
329
329
  logger.error(f"Error determining target AZ for {gpu_type}: {str(e)}")
330
- # Fallback to primary AZ if detection fails
331
- return PRIMARY_AVAILABILITY_ZONE
330
+ # Fallback to primary AZ if detection fails (no node hint — let k8s pick).
331
+ return PRIMARY_AVAILABILITY_ZONE, None
332
332
 
333
333
 
334
334
  def check_for_multiple_volumes(user_id):
@@ -180,8 +180,8 @@ resource "aws_lambda_function" "reservation_processor" {
180
180
  HOSTED_ZONE_ID = local.effective_domain_name != "" ? local.hosted_zone_id : ""
181
181
  SSH_DOMAIN_MAPPINGS_TABLE = local.effective_domain_name != "" ? aws_dynamodb_table.ssh_domain_mappings.name : ""
182
182
  SSL_CERTIFICATE_ARN = local.effective_domain_name != "" ? aws_acm_certificate.wildcard[0].arn : ""
183
- LAMBDA_VERSION = "0.5.9"
184
- MIN_CLI_VERSION = "0.5.5"
183
+ LAMBDA_VERSION = "0.5.11"
184
+ MIN_CLI_VERSION = "0.5.9"
185
185
  DISK_CONTENTS_BUCKET = aws_s3_bucket.disk_contents.bucket
186
186
  OPERATIONS_TABLE = aws_dynamodb_table.operations.name
187
187
  }, local.alb_env_vars)
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes