Skip to content

Commit 7982070

Browse files
committed
Fix type error
Signed-off-by: Timothy Lee <ctiml@infuseai.io>
1 parent 204afcb commit 7982070

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

primehub_admission/resources_validation.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -119,13 +119,13 @@ def get_user_and_group_from_request_info(self):
119119

120120
def _get_gpu_resource_from_limits(self, limits):
121121
if 'nvidia.com/gpu' in limits:
122-
return 'nvidia.com/gpu', limits['nvidia.com/gpu']
122+
return limits['nvidia.com/gpu']
123123
if 'amd.com/gpu' in limits:
124-
return 'amd.com/gpu', limits['amd.com/gpu']
124+
return limits['amd.com/gpu']
125125
for k, v in limits.items():
126126
if k.startswith('gpu.intel.com/'):
127-
return k, v
128-
return None, 0
127+
return v
128+
return 0
129129

130130
def aggregate_resource_usage(self, containers, cpu, gpu, mem):
131131
for container in containers:

0 commit comments

Comments
 (0)