Modified stream to make returning results optional

radiantone · Mar 13, 2022 · 2304268 · 2304268
1 parent c9ae393
commit 2304268
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/blazer/cli/__init__.py b/blazer/cli/__init__.py
@@ -90,14 +90,17 @@ def run_cmd(cmd):
         return result
 
     if mpi:
-        count = 1
+        count = 0
         with blazer.begin():
             _results = stream(getjobs(), run_cmd, results=results) 
             if results:
                 for result in _results:
+                    if not result:
+                        continue
                     if results:
                         blazer.print("RESULT[{}]:".format(count),result)
-                    count += 1
+                    count += len(result)
+                blazer.print("Total computations: {}".format(count))
             else:
                 blazer.print("Total computations: {}".format(list(_results)))
     else: