From 23042685d36f639ec2a21b1a3e35f8e19162519b Mon Sep 17 00:00:00 2001 From: Darren Govoni Date: Sun, 13 Mar 2022 12:44:43 -0400 Subject: [PATCH] Modified stream to make returning results optional --- blazer/cli/__init__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/blazer/cli/__init__.py b/blazer/cli/__init__.py index 91630d0..36cb694 100644 --- 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: