Merge pull request #20 from MTgeophysics/updates

- bumpversion

Author: kujaku11

.bumpversion.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 2.0.5
+current_version = 2.0.6
 files = setup.py mtpy/__init__.py README.md docs/source/conf.py
 commit = True
 tag = True

README.md

@@ -7,7 +7,7 @@
 [![Documentation Status](https://readthedocs.org/projects/mtpy-v2/badge/?version=latest)](https://mtpy-v2.readthedocs.io/en/latest/?badge=latest)
 [![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/MTgeophysics/mtpy-v2/main)
 
-## Version 2.0.5
+## Version 2.0.6
 
 # Description
 

docs/source/conf.py

@@ -67,9 +67,9 @@
 #
 
 # The short X.Y version.
-version = "2.0.5"
+version = "2.0.6"
 # The full version, including alpha/beta/rc tags.
-release = "2.0.5"
+release = "2.0.6"
 
 # The language for content autogenerated by Sphinx. Refer to documentation
 # for a list of supported languages.

mtpy/__init__.py

@@ -18,7 +18,7 @@
 from mtpy.imaging.mtcolors import MT_CMAP_DICT, register_cmaps
 
 
-__version__ = "2.0.5"
+__version__ = "2.0.6"
 __all__ = ["MT", "MTData", "MTCollection"]
 
 # =============================================================================

mtpy/core/mt.py

@@ -110,6 +110,7 @@ def __deepcopy__(self, memo):
                 continue
 
             setattr(result, k, deepcopy(v, memo))
+        result.logger = self.logger
         return result
 
     def copy(self):
@@ -987,9 +988,7 @@ def add_white_noise(self, value, inplace=True):
             ] = self._transfer_function.transfer_function.real * (
                 noise_real
             ) + (
-                1j
-                * self._transfer_function.transfer_function.imag
-                * noise_imag
+                1j * self._transfer_function.transfer_function.imag * noise_imag
             )
 
             self._transfer_function["transfer_function_error"] = (
@@ -1003,9 +1002,7 @@ def add_white_noise(self, value, inplace=True):
             ] = self._transfer_function.transfer_function.real * (
                 noise_real
             ) + (
-                1j
-                * self._transfer_function.transfer_function.imag
-                * noise_imag
+                1j * self._transfer_function.transfer_function.imag * noise_imag
             )
 
             self._transfer_function["transfer_function_error"] = (

mtpy/core/mt_data.py

@@ -227,6 +227,8 @@ def mt_list(self, value):
         """
         At this point not implemented, mainly here for inheritance of MTStations
         """
+        if len(self.values()) != 0:
+            self.logger.warning("mt_list cannot be set.")
         pass
 
     @property
@@ -262,6 +264,7 @@ def add_station(
         survey=None,
         compute_relative_location=True,
         interpolate_periods=None,
+        compute_model_error=False,
     ):
         """
         Add a MT object
@@ -295,6 +298,18 @@ def add_station(
 
                 m = m.interpolate(interpolate_periods, bounds_error=False)
 
+            if compute_model_error:
+                m.compute_model_z_errors(
+                    error_value=self.z_model_error.error_value,
+                    error_type=self.z_model_error.error_type,
+                    floor=self.z_model_error.floor,
+                )
+                m.compute_model_t_errors(
+                    error_value=self.t_model_error.error_value,
+                    error_type=self.t_model_error.error_type,
+                    floor=self.t_model_error.floor,
+                )
+
             self.__setitem__(f"{validate_name(m.survey)}.{m.station}", m)
 
         if compute_relative_location:
@@ -531,9 +546,7 @@ def interpolate(self, new_periods, f_type="period", inplace=True):
                 )
 
             else:
-                mt_data.add_station(
-                    new_mt_obj, compute_relative_location=False
-                )
+                mt_data.add_station(new_mt_obj, compute_relative_location=False)
 
         if not inplace:
             return mt_data
@@ -557,9 +570,7 @@ def rotate(self, rotation_angle, inplace=True):
             if not inplace:
                 rot_mt_obj = mt_obj.copy()
                 rot_mt_obj.rotation_angle = rotation_angle
-                mt_data.add_station(
-                    rot_mt_obj, compute_relative_location=False
-                )
+                mt_data.add_station(rot_mt_obj, compute_relative_location=False)
             else:
                 mt_obj.rotation_angle = rotation_angle
 
@@ -657,12 +668,8 @@ def compute_model_errors(
             self.t_model_error.floor = t_floor
 
         for mt_obj in self.values():
-            mt_obj.compute_model_z_errors(
-                **self.z_model_error.error_parameters
-            )
-            mt_obj.compute_model_t_errors(
-                **self.t_model_error.error_parameters
-            )
+            mt_obj.compute_model_z_errors(**self.z_model_error.error_parameters)
+            mt_obj.compute_model_t_errors(**self.t_model_error.error_parameters)
 
     def get_nearby_stations(self, station_key, radius, radius_units="m"):
         """

setup.py

@@ -65,7 +65,7 @@
     test_suite="tests",
     tests_require=test_requirements,
     url="https://github.com/MTgeophysics/mtpy-v2",
-    version="2.0.5",
+    version="2.0.6",
     zip_safe=False,
     package_data={"": []},
 )