nvidia-probe.py should exit 1 if there is an error

Former-commit-id: a45849c9

roles/slurm-common/files/scripts/nvidia-probe.py

@@ -23,7 +23,7 @@ import os
 import json
 
 try:
-	#run nvidia-smi -L to parse output
+    #run nvidia-smi -L to parse output
         p = subprocess.Popen(['nvidia-smi', '-L'], stdout=subprocess.PIPE,
                                    stderr=subprocess.PIPE)
         out, err = p.communicate()
@@ -48,9 +48,9 @@ try:
                 if os.path.exists(devicePath):
                         #print "OK"
                         deviceList.append( { 'name':'gpu' , 'file': devicePath, 'type':typeOfDevice } )
- 		else:
-                        #print "Error file not found ",devicePath
-			sys.exit(0)
+                else:
+                    print("error looking for nvidia device")
+                    sys.exit(1)
         #now convert list to json 
         output=json.dumps(deviceList)
         print output