fix: call list_gpus in deepssm mock task

MedVIC-Lab · Mar 27, 2024 · 4f1ce4f · 4f1ce4f
1 parent f59ce71
commit 4f1ce4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shapeworks_cloud/core/tasks.py b/shapeworks_cloud/core/tasks.py
@@ -357,7 +357,7 @@ def deepssm(progress_id):
     try:
         from ngpuinfo import NGPUInfo
 
-        gpus = NGPUInfo.list_gpus
+        gpus = NGPUInfo.list_gpus()
         message += f' GPU available. Found {[gpu.name for gpu in gpus]}.'
     except Exception:
         message += ' GPU not available.'