rdt 1.12.0.dev1__tar.gz → 1.12.1.dev0__tar.gz

{rdt-1.12.0.dev1/rdt.egg-info → rdt-1.12.1.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rdt
-Version: 1.12.0.dev1
+Version: 1.12.1.dev0
 Summary: Reversible Data Transforms
 Author-email: "DataCebo, Inc." <info@sdv.dev>
 License: BSL-1.1
@@ -24,17 +24,16 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: <3.13,>=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: numpy>=1.20.0; python_version < "3.10"
+Requires-Dist: numpy>=1.21.0; python_version < "3.10"
 Requires-Dist: numpy>=1.23.3; python_version >= "3.10" and python_version < "3.12"
 Requires-Dist: numpy>=1.26.0; python_version >= "3.12"
-Requires-Dist: pandas>=1.1.3; python_version < "3.10"
-Requires-Dist: pandas>=1.3.4; python_version >= "3.10" and python_version < "3.11"
+Requires-Dist: pandas>=1.4.0; python_version < "3.11"
 Requires-Dist: pandas>=1.5.0; python_version >= "3.11" and python_version < "3.12"
 Requires-Dist: pandas>=2.1.1; python_version >= "3.12"
-Requires-Dist: scipy>=1.5.4; python_version < "3.10"
+Requires-Dist: scipy>=1.7.3; python_version < "3.10"
 Requires-Dist: scipy>=1.9.2; python_version >= "3.10" and python_version < "3.12"
 Requires-Dist: scipy>=1.12.0; python_version >= "3.12"
-Requires-Dist: scikit-learn>=0.24; python_version < "3.10"
+Requires-Dist: scikit-learn>=1.0.2; python_version < "3.10"
 Requires-Dist: scikit-learn>=1.1.0; python_version >= "3.10" and python_version < "3.11"
 Requires-Dist: scikit-learn>=1.1.3; python_version >= "3.11" and python_version < "3.12"
 Requires-Dist: scikit-learn>=1.3.1; python_version >= "3.12"
@@ -56,33 +55,7 @@ Requires-Dist: build<2,>=1.0.0; extra == "dev"
 Requires-Dist: bump-my-version<1,>=0.18.3; extra == "dev"
 Requires-Dist: pip>=9.0.1; extra == "dev"
 Requires-Dist: watchdog<5,>=1.0.1; extra == "dev"
-Requires-Dist: pycodestyle<2.12,>=2.7.0; extra == "dev"
-Requires-Dist: pyflakes<3.3,>=2.3.0; extra == "dev"
-Requires-Dist: flake8<8,>=3.7.7; extra == "dev"
-Requires-Dist: flake8-absolute-import<2,>=1.0; extra == "dev"
-Requires-Dist: flake8-builtins<3,>=1.5.3; extra == "dev"
-Requires-Dist: flake8-comprehensions<4,>=3.6.1; extra == "dev"
-Requires-Dist: flake8-debugger<5,>=4.0.0; extra == "dev"
-Requires-Dist: flake8-docstrings<2,>=1.5.0; extra == "dev"
-Requires-Dist: flake8-eradicate<2,>=1.1.0; extra == "dev"
-Requires-Dist: flake8-fixme<1.2,>=1.1.1; extra == "dev"
-Requires-Dist: flake8-mock<1,>=0.3; extra == "dev"
-Requires-Dist: flake8-multiline-containers<0.1,>=0.0.18; extra == "dev"
-Requires-Dist: flake8-mutable<1.3,>=1.2.0; extra == "dev"
-Requires-Dist: flake8-expression-complexity<0.1,>=0.0.9; extra == "dev"
-Requires-Dist: flake8-print<4.1,>=4.0.0; extra == "dev"
-Requires-Dist: flake8-pytest-style<3,>=2.0.0; extra == "dev"
-Requires-Dist: flake8-quotes<4,>=3.3.0; extra == "dev"
-Requires-Dist: flake8-sfs<2,>=0.0.3; extra == "dev"
-Requires-Dist: flake8-variables-names<0.1,>=0.0.4; extra == "dev"
-Requires-Dist: dlint<1,>=0.11.0; extra == "dev"
-Requires-Dist: isort<6,>=5.13.2; extra == "dev"
-Requires-Dist: pandas-vet<2024,>=0.2.3; extra == "dev"
-Requires-Dist: pep8-naming<1,>=0.12.1; extra == "dev"
-Requires-Dist: pydocstyle<7,>=6.1.1; extra == "dev"
-Requires-Dist: pylint<4,>=2.5.3; extra == "dev"
-Requires-Dist: autoflake<3,>=1.1; extra == "dev"
-Requires-Dist: autopep8<3,>=1.4.3; extra == "dev"
+Requires-Dist: ruff<1,>=0.3.2; extra == "dev"
 Requires-Dist: twine<6,>=1.10.0; extra == "dev"
 Requires-Dist: wheel>=0.30.0; extra == "dev"
 Requires-Dist: coverage<8,>=4.5.12; extra == "dev"

{rdt-1.12.0.dev1 → rdt-1.12.1.dev0}/pyproject.toml

@@ -21,17 +21,16 @@ license = { text = 'BSL-1.1' }
 requires-python = '>=3.8,<3.13'
 readme = 'README.md'
 dependencies = [
-    "numpy>=1.20.0;python_version<'3.10'",
+    "numpy>=1.21.0;python_version<'3.10'",
     "numpy>=1.23.3;python_version>='3.10' and python_version<'3.12'",
     "numpy>=1.26.0;python_version>='3.12'",
-    "pandas>=1.1.3;python_version<'3.10'",
-    "pandas>=1.3.4;python_version>='3.10' and python_version<'3.11'",
+    "pandas>=1.4.0;python_version<'3.11'",
     "pandas>=1.5.0;python_version>='3.11' and python_version<'3.12'",
     "pandas>=2.1.1;python_version>='3.12'",
-    "scipy>=1.5.4;python_version<'3.10'",
+    "scipy>=1.7.3;python_version<'3.10'",
     "scipy>=1.9.2;python_version>='3.10' and python_version<'3.12'",
     "scipy>=1.12.0;python_version>='3.12'",
-    "scikit-learn>=0.24;python_version<'3.10'",
+    "scikit-learn>=1.0.2;python_version<'3.10'",
     "scikit-learn>=1.1.0;python_version>='3.10' and python_version<'3.11'",
     "scikit-learn>=1.1.3;python_version>='3.11' and python_version<'3.12'",
     "scikit-learn>=1.3.1;python_version>='3.12'",
@@ -71,35 +70,7 @@ dev = [
     'watchdog>=1.0.1,<5',
 
     # style check
-    'pycodestyle>=2.7.0,<2.12',
-    'pyflakes>=2.3.0,<3.3',
-    'flake8>=3.7.7,<8',
-    'flake8-absolute-import>=1.0,<2',
-    'flake8-builtins>=1.5.3,<3',
-    'flake8-comprehensions>=3.6.1,<4',
-    'flake8-debugger>=4.0.0,<5',
-    'flake8-docstrings>=1.5.0,<2',
-    'flake8-eradicate>=1.1.0,<2',
-    'flake8-fixme>=1.1.1,<1.2',
-    'flake8-mock>=0.3,<1',
-    'flake8-multiline-containers>=0.0.18,<0.1',
-    'flake8-mutable>=1.2.0,<1.3',
-    'flake8-expression-complexity>=0.0.9,<0.1',
-    'flake8-print>=4.0.0,<4.1',
-    'flake8-pytest-style>=2.0.0,<3',
-    'flake8-quotes>=3.3.0,<4',
-    'flake8-sfs>=0.0.3,<2',
-    'flake8-variables-names>=0.0.4,<0.1',
-    'dlint>=0.11.0,<1',
-    'isort>=5.13.2,<6',
-    'pandas-vet>=0.2.3,<2024',
-    'pep8-naming>=0.12.1,<1',
-    'pydocstyle>=6.1.1,<7',
-    'pylint>=2.5.3,<4',
-
-    # fix style issues
-    'autoflake>=1.1,<3',
-    'autopep8>=1.4.3,<3',
+    'ruff>=0.3.2,<1',
 
     # distribute on PyPI
     'twine>=1.10.0,<6',
@@ -166,7 +137,7 @@ collect_ignore = ['pyproject.toml']
 exclude_lines = ['NotImplementedError()']
 
 [tool.bumpversion]
-current_version = "1.12.0.dev1"
+current_version = "1.12.1.dev0"
 parse = '(?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(\.(?P<release>[a-z]+)(?P<candidate>\d+))?'
 serialize = [
     '{major}.{minor}.{patch}.{release}{candidate}',
@@ -201,3 +172,52 @@ replace = "__version__ = '{new_version}'"
 [build-system]
 requires = ['setuptools', 'wheel']
 build-backend = 'setuptools.build_meta'
+
+[tool.ruff]
+preview = true
+line-length = 100
+indent-width = 4
+src = ["rdt"]
+target-version = "py312"
+exclude = [
+    "docs",
+    ".tox",
+    ".git",
+    "__pycache__",
+    ".ipynb_checkpoints"
+]
+
+[tool.ruff.lint]
+select = [
+    # Pyflakes
+    "F",
+    # Pycodestyle
+    "E",
+    "W",
+    # isort
+    "I001"
+]
+ignore = [
+    "E501",
+    "D107",  # Missing docstring in __init__
+    "D417",   # Missing argument descriptions in the docstring, this is a bug from pydocstyle: https://github.com/PyCQA/pydocstyle/issues/449
+]
+
+[tool.ruff.format]
+quote-style = "single"
+indent-style = "space"
+preview = true
+docstring-code-format = true
+docstring-code-line-length = "dynamic"
+
+[tool.ruff.lint.pep8-naming]
+extend-ignore-names = ["X", "C", "X_padded", "Y", "Y_padded"]
+
+[tool.ruff.lint.isort]
+known-first-party = ["rdt"]
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401", "E402", "F403", "F405", "E501", "I001"]
+
+[tool.ruff.lint.pydocstyle]
+convention = "google"

{rdt-1.12.0.dev1 → rdt-1.12.1.dev0}/rdt/__init__.py

@@ -2,10 +2,9 @@
 
 """Top-level package for RDT."""
 
-
 __author__ = 'DataCebo, Inc.'
 __email__ = 'info@sdv.dev'
-__version__ = '1.12.0.dev1'
+__version__ = '1.12.1.dev0'
 
 
 import sys
@@ -20,10 +19,7 @@ import pandas as pd
 from rdt import transformers
 from rdt.hyper_transformer import HyperTransformer
 
-__all__ = [
-    'HyperTransformer',
-    'transformers'
-]
+__all__ = ['HyperTransformer', 'transformers']
 
 RANDOM_SEED = 42
 
@@ -42,9 +38,10 @@ def get_demo(num_rows=5):
         pd.DataFrame
     """
     # Hard code first five rows
-    login_dates = pd.Series([
-        '2021-06-26', '2021-02-10', 'NAT', '2020-09-26', '2020-12-22'
-    ], dtype='datetime64[ns]')
+    login_dates = pd.Series(
+        ['2021-06-26', '2021-02-10', 'NAT', '2020-09-26', '2020-12-22'],
+        dtype='datetime64[ns]',
+    )
     email_optin = pd.Series([False, False, False, True, np.nan], dtype='object')
     credit_card = ['VISA', 'VISA', 'AMEX', np.nan, 'DISCOVER']
     age = [29, 18, 21, 45, 32]
@@ -55,7 +52,7 @@ def get_demo(num_rows=5):
         'email_optin': email_optin,
         'credit_card': credit_card,
         'age': age,
-        'dollars_spent': dollars_spent
+        'dollars_spent': dollars_spent,
     })
 
     if num_rows <= 5:
@@ -67,14 +64,18 @@ def get_demo(num_rows=5):
     try:
         num_rows -= 5
 
-        login_dates = np.array([
-            np.datetime64('2000-01-01') + np.timedelta64(np.random.randint(0, 10000), 'D')
-            for _ in range(num_rows)
-        ], dtype='datetime64[ns]')
+        login_dates = np.array(
+            [
+                np.datetime64('2000-01-01') + np.timedelta64(np.random.randint(0, 10000), 'D')
+                for _ in range(num_rows)
+            ],
+            dtype='datetime64[ns]',
+        )
         login_dates[np.random.random(size=num_rows) > 0.8] = np.datetime64('NaT')
 
         email_optin = pd.Series([True, False, np.nan], dtype='object').sample(
-            num_rows, replace=True)
+            num_rows, replace=True
+        )
         credit_card = np.random.choice(['VISA', 'AMEX', np.nan, 'DISCOVER'], size=num_rows)
         age = np.random.randint(18, 100, size=num_rows)
 
@@ -84,16 +85,19 @@ def get_demo(num_rows=5):
     finally:
         np.random.set_state(random_state)
 
-    return pd.concat([
-        data,
-        pd.DataFrame({
-            'last_login': login_dates,
-            'email_optin': email_optin,
-            'credit_card': credit_card,
-            'age': age,
-            'dollars_spent': dollars_spent
-        })
-    ], ignore_index=True)
+    return pd.concat(
+        [
+            data,
+            pd.DataFrame({
+                'last_login': login_dates,
+                'email_optin': email_optin,
+                'credit_card': credit_card,
+                'age': age,
+                'dollars_spent': dollars_spent,
+            }),
+        ],
+        ignore_index=True,
+    )
 
 
 def _get_addon_target(addon_path_name):
@@ -159,7 +163,7 @@ def _find_addons():
         try:
             addon = entry_point.load()
         except Exception:  # pylint: disable=broad-exception-caught
-            msg = f'Failed to load "{entry_point.name}" from "{entry_point.version}".'
+            msg = f'Failed to load "{entry_point.name}" from "{entry_point.value}".'
             warnings.warn(msg)
             continue
 

{rdt-1.12.0.dev1 → rdt-1.12.1.dev0}/rdt/hyper_transformer.py

@@ -9,11 +9,20 @@ from copy import deepcopy
 import pandas as pd
 
 from rdt.errors import (
-    ConfigNotSetError, InvalidConfigError, InvalidDataError, NotFittedError, TransformerInputError,
-    TransformerProcessingError)
+    ConfigNotSetError,
+    InvalidConfigError,
+    InvalidDataError,
+    NotFittedError,
+    TransformerInputError,
+    TransformerProcessingError,
+)
 from rdt.transformers import (
-    BaseMultiColumnTransformer, BaseTransformer, get_class_by_transformer_name,
-    get_default_transformer, get_transformers_by_type)
+    BaseMultiColumnTransformer,
+    BaseTransformer,
+    get_class_by_transformer_name,
+    get_default_transformer,
+    get_transformers_by_type,
+)
 from rdt.transformers.utils import flatten_column_list
 
 LOGGER = logging.getLogger(__name__)
@@ -31,7 +40,7 @@ class Config(dict):
 
         config = {
             'sdtypes': self['sdtypes'],
-            'transformers': {str(k): repr(v) for k, v in self['transformers'].items()}
+            'transformers': {str(k): repr(v) for k, v in self['transformers'].items()},
         }
 
         printed = json.dumps(config, indent=4)
@@ -59,11 +68,7 @@ class HyperTransformer:
         'b': 'boolean',
         'M': 'datetime',
     }
-    _DEFAULT_OUTPUT_SDTYPES = [
-        'numerical',
-        'float',
-        'integer'
-    ]
+    _DEFAULT_OUTPUT_SDTYPES = ['numerical', 'float', 'integer']
     _REFIT_MESSAGE = (
         "For this change to take effect, please refit your data using 'fit' or 'fit_transform'."
     )
@@ -92,11 +97,7 @@ class HyperTransformer:
 
     @staticmethod
     def _subset(input_list, other_list, not_in=False):
-        return [
-            element
-            for element in input_list
-            if (element in other_list) ^ not_in
-        ]
+        return [element for element in input_list if (element in other_list) ^ not_in]
 
     def _create_multi_column_fields(self):
         multi_column_fields = {}
@@ -109,9 +110,11 @@ class HyperTransformer:
     def _validate_field_transformers(self):
         for field in self.field_transformers:
             if self._field_in_set(field, self._specified_fields):
-                raise ValueError(f'Multiple transformers specified for the field {field}. '
-                                 'Each field can have at most one transformer defined in '
-                                 'field_transformers.')
+                raise ValueError(
+                    f'Multiple transformers specified for the field {field}. '
+                    'Each field can have at most one transformer defined in '
+                    'field_transformers.'
+                )
 
             self._add_field_to_set(field, self._specified_fields)
 
@@ -150,7 +153,7 @@ class HyperTransformer:
         """
         return Config({
             'sdtypes': self.field_sdtypes,
-            'transformers': self.field_transformers
+            'transformers': self.field_transformers,
         })
 
     @staticmethod
@@ -250,7 +253,9 @@ class HyperTransformer:
 
     def _validate_update_columns(self, update_columns):
         unknown_columns = self._subset(
-            flatten_column_list(update_columns), self.field_sdtypes.keys(), not_in=True
+            flatten_column_list(update_columns),
+            self.field_sdtypes.keys(),
+            not_in=True,
         )
         if unknown_columns:
             raise InvalidConfigError(
@@ -279,7 +284,8 @@ class HyperTransformer:
             warnings.warn(self._REFIT_MESSAGE)
 
     def _validate_update_transformers_by_sdtype(
-            self, sdtype, transformer, transformer_name, transformer_parameters):
+        self, sdtype, transformer, transformer_name, transformer_parameters
+    ):
         if not self.field_sdtypes:
             raise ConfigNotSetError(
                 'Nothing to update. Use the `detect_initial_config` method to '
@@ -301,10 +307,14 @@ class HyperTransformer:
                 )
 
         else:
-            if transformer_name not in get_class_by_transformer_name() or sdtype not in \
-                    get_class_by_transformer_name()[transformer_name].get_supported_sdtypes():
+            if (
+                transformer_name not in get_class_by_transformer_name()
+                or sdtype
+                not in get_class_by_transformer_name()[transformer_name].get_supported_sdtypes()
+            ):
                 raise InvalidConfigError(
-                    f"Invalid transformer name '{transformer_name}' for the '{sdtype}' sdtype.")
+                    f"Invalid transformer name '{transformer_name}' for the '{sdtype}' sdtype."
+                )
 
             if transformer_parameters is not None:
                 transformer = get_class_by_transformer_name()[transformer_name]
@@ -325,14 +335,15 @@ class HyperTransformer:
                 warnings.warn(
                     "The 'transformer' parameter will no longer be supported in future versions "
                     "of the RDT. Using the 'transformer_name' parameter instead.",
-                    FutureWarning
+                    FutureWarning,
                 )
 
         else:
             warnings.warn(
                 "The 'transformer' parameter will no longer be supported in future versions "
                 "of the RDT. Please use the 'transformer_name' and 'transformer_parameters' "
-                'parameters instead.', FutureWarning
+                'parameters instead.',
+                FutureWarning,
             )
 
     def _remove_column_in_multi_column_fields(self, column):
@@ -349,7 +360,7 @@ class HyperTransformer:
         new_tuple = tuple(item for item in old_tuple if item != column)
 
         if len(new_tuple) == 1:
-            new_tuple, = new_tuple
+            (new_tuple,) = new_tuple
             self._multi_column_fields.pop(new_tuple, None)
         else:
             for col in new_tuple:
@@ -369,7 +380,7 @@ class HyperTransformer:
 
             columns_to_sdtypes = self._get_columns_to_sdtypes(field)
             try:
-                transformer._validate_sdtypes(   # pylint: disable=protected-access
+                transformer._validate_sdtypes(  # pylint: disable=protected-access
                     columns_to_sdtypes
                 )
             except TransformerInputError:
@@ -384,7 +395,12 @@ class HyperTransformer:
         self._multi_column_fields = self._create_multi_column_fields()
 
     def update_transformers_by_sdtype(
-            self, sdtype, transformer=None, transformer_name=None, transformer_parameters=None):
+        self,
+        sdtype,
+        transformer=None,
+        transformer_name=None,
+        transformer_parameters=None,
+    ):
         """Update the transformers for the specified ``sdtype``.
 
         Given an ``sdtype`` and a ``transformer``, change all the fields of the ``sdtype``
@@ -403,15 +419,17 @@ class HyperTransformer:
                 A dict of the kwargs of the transformer.
         """
         self._validate_update_transformers_by_sdtype(
-            sdtype, transformer, transformer_name, transformer_parameters)
+            sdtype, transformer, transformer_name, transformer_parameters
+        )
         self._warn_update_transformers_by_sdtype(transformer, transformer_name)
 
         transformer_instance = transformer
 
         if transformer_name is not None:
             if transformer_parameters is not None:
-                transformer_instance = \
-                    get_class_by_transformer_name()[transformer_name](**transformer_parameters)
+                transformer_instance = get_class_by_transformer_name()[transformer_name](
+                    **transformer_parameters
+                )
 
             else:
                 transformer_instance = get_class_by_transformer_name()[transformer_name]()
@@ -620,7 +638,7 @@ class HyperTransformer:
 
         config = Config({
             'sdtypes': self.field_sdtypes,
-            'transformers': self.field_transformers
+            'transformers': self.field_transformers,
         })
 
         LOGGER.info('Config:')
@@ -675,7 +693,6 @@ class HyperTransformer:
 
             next_transformers = transformer.get_next_transformers()
             for column_name, next_transformer in next_transformers.items():
-
                 # If the column is part of a multi-column field, and at least one column
                 # isn't present in the data, then it should not fit the next transformer
                 if self._field_in_data(column_name, data):

{rdt-1.12.0.dev1 → rdt-1.12.1.dev0}/rdt/performance/datasets/__init__.py

@@ -2,7 +2,14 @@
 
 from collections import defaultdict
 
-from rdt.performance.datasets import boolean, categorical, datetime, numerical, pii, text
+from rdt.performance.datasets import (
+    boolean,
+    categorical,
+    datetime,
+    numerical,
+    pii,
+    text,
+)
 from rdt.performance.datasets.base import BaseDatasetGenerator
 
 __all__ = [

{rdt-1.12.0.dev1 → rdt-1.12.1.dev0}/rdt/performance/datasets/boolean.py

@@ -36,18 +36,12 @@ class RandomBooleanGenerator(BooleanGenerator):
     def get_performance_thresholds():
         """Return the expected threseholds."""
         return {
-            'fit': {
-                'time': 2e-5,
-                'memory': 400.0
-            },
-            'transform': {
-                'time': 1e-5,
-                'memory': 400.0
-            },
+            'fit': {'time': 2e-5, 'memory': 400.0},
+            'transform': {'time': 1e-5, 'memory': 400.0},
             'reverse_transform': {
                 'time': 5e-5,
                 'memory': 500.0,
-            }
+            },
         }
 
 
@@ -71,18 +65,12 @@ class RandomBooleanNaNsGenerator(BooleanGenerator):
     def get_performance_thresholds():
         """Return the expected threseholds."""
         return {
-            'fit': {
-                'time': 2e-5,
-                'memory': 400.0
-            },
-            'transform': {
-                'time': 1e-5,
-                'memory': 1000.0
-            },
+            'fit': {'time': 2e-5, 'memory': 400.0},
+            'transform': {'time': 1e-5, 'memory': 1000.0},
             'reverse_transform': {
                 'time': 5e-5,
                 'memory': 1000.0,
-            }
+            },
         }
 
 
@@ -104,18 +92,12 @@ class RandomSkewedBooleanGenerator(BooleanGenerator):
     def get_performance_thresholds():
         """Return the expected threseholds."""
         return {
-            'fit': {
-                'time': 1e-5,
-                'memory': 400.0
-            },
-            'transform': {
-                'time': 1e-5,
-                'memory': 400.0
-            },
+            'fit': {'time': 1e-5, 'memory': 400.0},
+            'transform': {'time': 1e-5, 'memory': 400.0},
             'reverse_transform': {
                 'time': 5e-5,
                 'memory': 500.0,
-            }
+            },
         }
 
 
@@ -139,18 +121,12 @@ class RandomSkewedBooleanNaNsGenerator(BooleanGenerator):
     def get_performance_thresholds():
         """Return the expected threseholds."""
         return {
-            'fit': {
-                'time': 1e-5,
-                'memory': 400.0
-            },
-            'transform': {
-                'time': 1e-5,
-                'memory': 1000.0
-            },
+            'fit': {'time': 1e-5, 'memory': 400.0},
+            'transform': {'time': 1e-5, 'memory': 1000.0},
             'reverse_transform': {
                 'time': 5e-5,
                 'memory': 1000.0,
-            }
+            },
         }
 
 
@@ -167,18 +143,12 @@ class ConstantBooleanGenerator(BooleanGenerator):
     def get_performance_thresholds():
         """Return the expected threseholds."""
         return {
-            'fit': {
-                'time': 1e-5,
-                'memory': 400.0
-            },
-            'transform': {
-                'time': 1e-5,
-                'memory': 400.0
-            },
+            'fit': {'time': 1e-5, 'memory': 400.0},
+            'transform': {'time': 1e-5, 'memory': 400.0},
             'reverse_transform': {
                 'time': 5e-5,
                 'memory': 500.0,
-            }
+            },
         }
 
 
@@ -201,16 +171,10 @@ class ConstantBooleanNaNsGenerator(BooleanGenerator):
     def get_performance_thresholds():
         """Return the expected threseholds."""
         return {
-            'fit': {
-                'time': 1e-5,
-                'memory': 400.0
-            },
-            'transform': {
-                'time': 1e-5,
-                'memory': 1000.0
-            },
+            'fit': {'time': 1e-5, 'memory': 400.0},
+            'transform': {'time': 1e-5, 'memory': 1000.0},
             'reverse_transform': {
                 'time': 5e-5,
                 'memory': 1000.0,
-            }
+            },
         }