fixing bugs

Author: kujaku11

mtpy/processing/aurora/process_aurora.py

@@ -138,9 +138,7 @@ def create_config(
             cc = ConfigCreator()
             config = cc.create_from_kernel_dataset(kernel_dataset, **kwargs)
             if kernel_dataset.sample_rate > 1000:
-                decimation_kwargs.update(
-                    self.default_window_parameters["high"]
-                )
+                decimation_kwargs.update(self.default_window_parameters["high"])
             else:
                 decimation_kwargs.update(self.default_window_parameters["low"])
             self._set_decimation_level_parameters(config, **decimation_kwargs)
@@ -158,6 +156,8 @@ def _set_decimation_level_parameters(self, config, **kwargs):
 
         for decimation in config.decimations:
             for key, value in kwargs.items():
+                if "stft" in key:
+                    continue  # stft is not a decimation attribute
                 decimation.set_attr_from_name(key, value)
 
     def _initialize_kernel_dataset(self, sample_rate=None):
@@ -304,9 +304,7 @@ def process(
         """
 
         if sample_rates is None and processing_dict is None:
-            raise ValueError(
-                "Must set either sample rates or processing_dict."
-            )
+            raise ValueError("Must set either sample rates or processing_dict.")
 
         if processing_dict is None:
             if isinstance(sample_rates, (int, float)):

mtpy/processing/kernel_dataset.py

@@ -810,9 +810,12 @@ def update_survey_metadata(
         :rtype: None
         """
         survey_id = run_ts.survey_metadata.id
-        if i == 0:
-            self.survey_metadata[survey_id] = run_ts.survey_metadata
-        elif i > 0:
+        if survey_id not in self.survey_metadata.keys():
+            self.survey_metadata[survey_id] = mt_metadata.timeseries.Survey(
+                id=survey_id
+            )
+            self.survey_metadata[survey_id].add_station(run_ts.station_metadata)
+        else:
             if row.station in self.survey_metadata[survey_id].stations.keys():
                 self.survey_metadata[survey_id].stations[row.station].add_run(
                     run_ts.run_metadata
@@ -821,8 +824,8 @@ def update_survey_metadata(
                 self.survey_metadata[survey_id].add_station(
                     run_ts.station_metadata
                 )
-        if len(self.survey_metadata.keys()) > 1:
-            raise NotImplementedError
+        # if len(self.survey_metadata.keys()) > 1:
+        #     raise NotImplementedError
 
     @property
     def mth5_objs(self):
@@ -886,7 +889,9 @@ def initialize_dataframe_for_processing(self) -> None:
             self.df["run_hdf5_reference"].at[i] = run_obj.hdf5_group.ref
 
             if row.fc:
-                msg = f"row {row} already has fcs prescribed by processing config"
+                msg = (
+                    f"row {row} already has fcs prescribed by processing config"
+                )
                 msg += "-- skipping time series initialisation"
                 logger.info(msg)
                 # see Note #3