fixes #10

AntreasAntoniou · Dec 18, 2023 · e30a3c7 · e30a3c7
1 parent 5e86bfe
commit e30a3c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kubejobs/jobs.py b/kubejobs/jobs.py
@@ -126,12 +126,12 @@ def __init__(
         self.cpu_request = (
             cpu_request
             if cpu_request
-            else MAX_CPU // (MAX_GPU - gpu_limit + 1)
+            else MAX_CPU // (MAX_GPU - (1 if gpu_limit is None else gpu_limit) + 1)
         )
         self.ram_request = (
             ram_request
             if ram_request
-            else f"{MAX_RAM // (MAX_GPU - gpu_limit + 1)}G"
+            else f"{MAX_RAM // (MAX_GPU - (1 if gpu_limit is None else gpu_limit) + 1)}G"
         )
         self.storage_request = storage_request
         self.gpu_type = gpu_type