corrections

Author: dkazanc

docs/source/howto/run_httomo.rst

@@ -407,7 +407,7 @@ method to execute in the pipeline, which will look something like the following:
       Total methods CPU time:     19.376s
       Total methods GPU time:     19.042s
       Total host2device time:      0.013s
-      Total device2host time:      0.54 8s
+      Total device2host time:      0.548s
       Total sources time    :      0.063s
       Total sinks time      :      0.028s
       Other overheads       :      0.362s

httomo/method_wrappers/images.py

@@ -7,7 +7,6 @@
 
 
 from mpi4py.MPI import Comm
-import numpy as np
 
 import os
 from typing import Dict, Optional

httomo/runner/task_runner.py

@@ -311,19 +311,10 @@ def _get_methods_name_for_snapshot(self, section: Section) -> str:
             "find_center_vo",
             "save_intermediate_data",
         ]
-        for ind_outer in range(1, len(section)):
-            methods_name = section.methods[-ind_outer].method_name
-            if methods_name in irrelevant_method_names_snapshots:
-                for ind_inner in range(1, len(section)):
-                    methods_name = section.methods[-(ind_inner + 1)].method_name
-                    if methods_name in irrelevant_method_names_snapshots:
-                        methods_name = section.methods[-ind_inner + 1].method_name
-                    else:
-                        break
-            else:
-                break
-            break
-        return methods_name
+        for wrapper in list(reversed(section.methods)):
+            if wrapper.method_name not in irrelevant_method_names_snapshots:
+                return wrapper.method_name                    
+        raise ValueError("Unable to find method name in section for snapshot saving")
 
     def _log_pipeline(self, msg: Any, level: int = logging.INFO):
         log_once(msg, level=level)