diff --git a/SwanSpawner/swanspawner/swankubespawner.py b/SwanSpawner/swanspawner/swankubespawner.py index 9d5de73..0c5fe33 100644 --- a/SwanSpawner/swanspawner/swankubespawner.py +++ b/SwanSpawner/swanspawner/swankubespawner.py @@ -27,8 +27,6 @@ async def start(self): """ if self._gpu_requested(): - self._check_gpu_access() - self.extra_resource_guarantees["nvidia.com/gpu"] = "1" self.extra_resource_limits["nvidia.com/gpu"] = "1" elif "nvidia.com/gpu" in self.extra_resource_guarantees: @@ -116,12 +114,4 @@ def get_env(self): def _gpu_requested(self): """Returns true if the user requested a GPU""" - return "cu" in self.user_options[self.lcg_rel_field] - - def _check_gpu_access(self): - """Checks if the user is allowed to start a session with a GPU""" - if "swan-gpu" not in self.user_roles: - raise ValueError( - """Access to GPUs is not granted; - please open a Support Ticket""" - ) + return "cu" in self.user_options[self.lcg_rel_field] \ No newline at end of file