assign result correctly

apache · Dec 17, 2024 · 922bb92 · 922bb92
1 parent f75c1e3
commit 922bb92
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...pache_beam/testing/benchmarks/inference/tensorflow_mnist_classification_cost_benchmark.py b/...pache_beam/testing/benchmarks/inference/tensorflow_mnist_classification_cost_benchmark.py
@@ -31,7 +31,7 @@ def test(self):
     extra_opts['input'] = self.pipeline.get_option('input_file')
     extra_opts['output'] = self.pipeline.get_option('output_file')
     extra_opts['model_path'] = self.pipeline.get_option('model')
-    tensorflow_mnist_classification.run(
+    self.result = tensorflow_mnist_classification.run(
         self.pipeline.get_full_options_as_args(**extra_opts),
         save_main_session=False)