The logic only gets more complicated

NERSC · Jun 28, 2020 · 5fc3c46 · 5fc3c46
1 parent bb26b27
commit 5fc3c46
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/jupyter-nersc/web-jupyterhub/nerscspawner.py b/jupyter-nersc/web-jupyterhub/nerscspawner.py
@@ -63,9 +63,10 @@ def check_role_staff(self, auth_state):
         return False
 
     def default_gpu_repo(self, auth_state):
-        for allocation in self.user_allocations(auth_state, ["nstaff", "m1759", "dasrepo", "gpu4sci"]):
+#       for allocation in self.user_allocations(auth_state, ["nstaff", "m1759", "dasrepo", "gpu4sci"]):
+        for allocation in self.user_allocations(auth_state):
             for qos in allocation["userAllocationQos"]:
-                if qos["qos"]["qos"] == "gpu":
+                if qos["qos"]["qos"] in ["gpu", "gpu_special_m1759"]:
                     return allocation["computeAllocation"]["repoName"]
         return None