Merge pull request #24 from zijwang/fix_attrlist

Fix attrList no-effect issue. Close #22 #27
anderskm · Aug 16, 2019 · 42ef071 · 42ef071
2 parents 47e1e0f + a4bf18d
commit 42ef071
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/GPUtil/GPUtil.py b/GPUtil/GPUtil.py
@@ -234,7 +234,8 @@ def showUtilization(all=False, attrList=None, useOldCode=False):
             print('--------------')
             for gpu in GPUs:
                 print(' {0:2d} {1:3.0f}% {2:3.0f}%'.format(gpu.id, gpu.load*100, gpu.memoryUtil*100))
-        else:
+        elif attrList is None:
+            # if `attrList` was not specified, use the default one
             attrList = [[{'attr':'id','name':'ID'},
                          {'attr':'load','name':'GPU','suffix':'%','transform': lambda x: x*100,'precision':0},
                          {'attr':'memoryUtil','name':'MEM','suffix':'%','transform': lambda x: x*100,'precision':0}],