Merge branch 'fix_save_energy_offset' of github.com:OpenCOMPES/sed in…

…to fix_save_energy_offset
OpenCOMPES · Nov 9, 2023 · d48897a · d48897a
2 parents ee0da20 + f8a19ca
commit d48897a
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/sed/core/processor.py b/sed/core/processor.py
@@ -65,6 +65,7 @@ def __init__(
         folder: str = None,
         runs: Sequence[str] = None,
         collect_metadata: bool = False,
+        verbose: bool = False,
         **kwds,
     ):
         """Processor class of sed. Contains wrapper functions defining a work flow
@@ -97,6 +98,8 @@ def __init__(
             num_cores = N_CPU - 1
         self._config["binning"]["num_cores"] = num_cores
 
+        self.verbose = verbose
+
         self._dataframe: Union[pd.DataFrame, ddf.DataFrame] = None
         self._timed_dataframe: Union[pd.DataFrame, ddf.DataFrame] = None
         self._files: List[str] = []
@@ -666,7 +669,8 @@ def apply_momentum_correction(
             if preview:
                 print(self._dataframe.head(10))
             else:
-                print(self._dataframe)
+                if self.verbose:
+                    print(self._dataframe)
 
     # Momentum calibration work flow
     # 1. Calculate momentum calibration
@@ -800,7 +804,8 @@ def apply_momentum_calibration(
             if preview:
                 print(self._dataframe.head(10))
             else:
-                print(self._dataframe)
+                if self.verbose:
+                    print(self._dataframe)
 
     # Energy correction workflow
     # 1. Adjust the energy correction parameters
@@ -925,7 +930,8 @@ def apply_energy_correction(
             if preview:
                 print(self._dataframe.head(10))
             else:
-                print(self._dataframe)
+                if self.verbose:
+                    print(self._dataframe)
 
     # Energy calibrator workflow
     # 1. Load and normalize data
@@ -1258,7 +1264,8 @@ def append_energy_axis(
             if preview:
                 print(self._dataframe.head(10))
             else:
-                print(self._dataframe)
+                if self.verbose:
+                    print(self._dataframe)
 
     def add_energy_offset(
         self,
@@ -1473,7 +1480,8 @@ def calibrate_delay_axis(
             if preview:
                 print(self._dataframe.head(10))
             else:
-                print(self._dataframe)
+                if self.verbose:
+                    print(self._dataframe)
 
     def save_workflow_params(
         self,