[pre-commit.ci] pre-commit autoupdate

.pre-commit-config.yaml

@@ -15,7 +15,7 @@ repos:
       - id: trailing-whitespace
 
   - repo: https://github.com/psf/black
-    rev: 23.12.1
+    rev: 24.1.1
     hooks:
       - id: black
         args: ["--line-length", "100"]

example/1_ctgan_example.py

@@ -1,6 +1,7 @@
 """
 Example for CTGAN
 """
+
 from sdgx.data_connectors.csv_connector import CsvConnector
 from sdgx.models.ml.single_table.ctgan import CTGANSynthesizerModel
 from sdgx.synthesizer import Synthesizer

example/extension/dummycache/dummycache/dummycache.py

@@ -3,8 +3,7 @@
 from sdgx.cachers.base import Cacher
 
 
-class MyOwnCache(Cacher):
-    ...
+class MyOwnCache(Cacher): ...
 
 
 from sdgx.cachers.extension import hookimpl

example/extension/dummydataconnector/dummydataconnector/dataconnector.py

@@ -3,8 +3,7 @@
 from sdgx.data_connectors.base import DataConnector
 
 
-class MyOwnDataConnector(DataConnector):
-    ...
+class MyOwnDataConnector(DataConnector): ...
 
 
 from sdgx.data_connectors.extension import hookimpl

example/extension/dummydataprocessor/dummydataprocessor/dataprocessor.py

@@ -3,8 +3,7 @@
 from sdgx.data_processors.base import DataProcessor
 
 
-class MyOwnDataProcessor(DataProcessor):
-    ...
+class MyOwnDataProcessor(DataProcessor): ...
 
 
 from sdgx.data_processors.extension import hookimpl

example/extension/dummyexporter/dummyexporter/dummyexporter.py

@@ -3,8 +3,7 @@
 from sdgx.data_exporters.base import DataExporter
 
 
-class MyOwnExporter(DataExporter):
-    ...
+class MyOwnExporter(DataExporter): ...
 
 
 from sdgx.data_exporters.extension import hookimpl

example/extension/dummymetadatainspector/dummymetadatainspector/inspector.py

@@ -4,8 +4,7 @@
 from sdgx.data_models.inspectors.extension import hookimpl
 
 
-class MyOwnInspector(Inspector):
-    ...
+class MyOwnInspector(Inspector): ...
 
 
 @hookimpl

example/extension/dummymodel/dummymodel/model.py

@@ -3,8 +3,7 @@
 from sdgx.models.base import SynthesizerModel
 
 
-class MyOwnModel(SynthesizerModel):
-    ...
+class MyOwnModel(SynthesizerModel): ...
 
 
 from sdgx.models.extension import hookimpl

sdgx/models/components/optimize/sdv_ctgan/data_sampler.py

@@ -1,4 +1,5 @@
 """DataSampler module."""
+
 from __future__ import annotations
 
 import numpy as np

sdgx/models/components/optimize/sdv_ctgan/data_transformer.py

@@ -1,4 +1,5 @@
 """DataTransformer module."""
+
 from __future__ import annotations
 
 from collections import namedtuple

sdgx/models/components/sdv_copulas/bivariate/clayton.py

@@ -82,11 +82,14 @@ def cumulative_distribution(self, X):
 
         else:
             cdfs = [
-                np.power(
-                    np.power(U[i], -self.theta) + np.power(V[i], -self.theta) - 1, -1.0 / self.theta
+                (
+                    np.power(
+                        np.power(U[i], -self.theta) + np.power(V[i], -self.theta) - 1,
+                        -1.0 / self.theta,
+                    )
+                    if (U[i] > 0 and V[i] > 0)
+                    else 0
                 )
-                if (U[i] > 0 and V[i] > 0)
-                else 0
                 for i in range(len(U))
             ]
 

sdgx/models/components/sdv_rdt/transformers/base.py

@@ -1,4 +1,5 @@
 """BaseTransformer module."""
+
 import abc
 import inspect
 

sdgx/models/components/sdv_rdt/transformers/datetime.py

@@ -1,4 +1,5 @@
 """Transformer for datetime data."""
+
 import numpy as np
 import pandas as pd
 from pandas.api.types import is_datetime64_dtype

sdgx/models/components/sdv_rdt/transformers/numerical.py

@@ -1,4 +1,5 @@
 """Transformers for numerical data."""
+
 import copy
 import sys
 import warnings

sdgx/models/components/sdv_rdt/transformers/text.py

@@ -1,4 +1,5 @@
 """Transformers for text data."""
+
 import warnings
 
 import numpy as np

sdgx/models/statistics/single_table/copula.py

@@ -2,6 +2,7 @@
     Wrappers around copulas models.
     需要修改: fit接口以适应性能优化措施
 """
+
 import logging
 import warnings
 from copy import deepcopy

sdgx/synthesizer.py

@@ -103,10 +103,12 @@ def __init__(
             data_processors = []
         self.data_processors_manager = DataProcessorManager()
         self.data_processors = [
-            d
-            if isinstance(d, DataProcessor)
-            else self.data_processors_manager.init_data_processor(
-                d, **(data_processors_kwargs or {})
+            (
+                d
+                if isinstance(d, DataProcessor)
+                else self.data_processors_manager.init_data_processor(
+                    d, **(data_processors_kwargs or {})
+                )
             )
             for d in data_processors
         ]