rdt 1.17.1.dev0__tar.gz → 1.17.2.dev0__tar.gz

{rdt-1.17.1.dev0/rdt.egg-info → rdt-1.17.2.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rdt
-Version: 1.17.1.dev0
+Version: 1.17.2.dev0
 Summary: Reversible Data Transforms
 Author-email: "DataCebo, Inc." <info@sdv.dev>
 License: BSL-1.1

{rdt-1.17.1.dev0 → rdt-1.17.2.dev0}/pyproject.toml

@@ -144,7 +144,7 @@ collect_ignore = ['pyproject.toml']
 exclude_lines = ['NotImplementedError()']
 
 [tool.bumpversion]
-current_version = "1.17.1.dev0"
+current_version = "1.17.2.dev0"
 parse = '(?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(\.(?P<release>[a-z]+)(?P<candidate>\d+))?'
 serialize = [
     '{major}.{minor}.{patch}.{release}{candidate}',

{rdt-1.17.1.dev0 → rdt-1.17.2.dev0}/rdt/__init__.py

@@ -4,7 +4,7 @@
 
 __author__ = 'DataCebo, Inc.'
 __email__ = 'info@sdv.dev'
-__version__ = '1.17.1.dev0'
+__version__ = '1.17.2.dev0'
 
 
 import sys

{rdt-1.17.1.dev0 → rdt-1.17.2.dev0}/rdt/transformers/numerical.py

@@ -1,6 +1,7 @@
 """Transformers for numerical data."""
 
 import copy
+import logging
 import warnings
 from importlib import import_module
 
@@ -13,6 +14,8 @@ from rdt.transformers.base import BaseTransformer
 from rdt.transformers.null import NullTransformer
 from rdt.transformers.utils import learn_rounding_digits, logit, sigmoid
 
+LOGGER = logging.getLogger(__name__)
+
 EPSILON = np.finfo(np.float32).eps
 INTEGER_BOUNDS = {
     'Int8': (-(2**7), 2**7 - 1),
@@ -300,6 +303,8 @@ class GaussianNormalizer(FloatFormatter):
     """
 
     _univariate = None
+    _fallback_distribution = 'norm'
+    _learned_distribution_name = None
     _DEPRECATED_DISTRIBUTIONS_MAPPING = {
         'gaussian': 'norm',
         'student_t': 't',
@@ -359,8 +364,15 @@ class GaussianNormalizer(FloatFormatter):
                 )
                 distribution = self._DEPRECATED_DISTRIBUTIONS_MAPPING[distribution]
 
+            self._learned_distribution_name = distribution
             distribution = self._distributions[distribution]
 
+        else:
+            distribution_name_map = {v: k for k, v in self._distributions.items()}
+            self._learned_distribution_name = distribution_name_map.get(
+                distribution if isinstance(distribution, type) else type(distribution), 'unknown'
+            )
+
         self._distribution = distribution
 
     def _get_univariate(self):
@@ -390,7 +402,16 @@ class GaussianNormalizer(FloatFormatter):
 
         with warnings.catch_warnings():
             warnings.simplefilter('ignore')
-            self._univariate.fit(data)
+            try:
+                self._univariate.fit(data)
+            except Exception:
+                LOGGER.info(
+                    f"Unable to fit the distribution '{self._learned_distribution_name}'. "
+                    f"Falling back to '{self._fallback_distribution}'."
+                )
+                self._learned_distribution_name = self._fallback_distribution
+                self._univariate = self._distributions[self._fallback_distribution]()
+                self._univariate.fit(data)
 
     def _copula_transform(self, data):
         cdf = self._univariate.cdf(data)
@@ -434,6 +455,24 @@ class GaussianNormalizer(FloatFormatter):
 
         return super()._reverse_transform(data)
 
+    @property
+    def learned_distribution(self):
+        """Get the learned distribution name and parameters.
+
+        Returns:
+            dict:
+                A dictionary containing the distribution name and parameters.
+        """
+        if self._univariate is None:
+            raise ValueError('The transformer has not been fitted yet.')
+
+        parameters = {}
+        if hasattr(self._univariate, 'to_dict') and self._univariate.to_dict is not None:
+            parameters = self._univariate.to_dict()
+            parameters.pop('type')
+
+        return {'distribution': self._learned_distribution_name, 'parameters': parameters}
+
 
 class ClusterBasedNormalizer(FloatFormatter):
     """Transformer for numerical data using a Bayesian Gaussian Mixture Model.

{rdt-1.17.1.dev0 → rdt-1.17.2.dev0/rdt.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rdt
-Version: 1.17.1.dev0
+Version: 1.17.2.dev0
 Summary: Reversible Data Transforms
 Author-email: "DataCebo, Inc." <info@sdv.dev>
 License: BSL-1.1