Skip to content

Commit 2f67331

Browse files
committed
fix: add import BaseModel to timeseries models
1 parent 4885446 commit 2f67331

File tree

3 files changed

+4
-2
lines changed

3 files changed

+4
-2
lines changed

src/ydata_synthetic/synthesizers/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,8 @@ def __init__(self,
3636
cat_cols: Optional[List[str]] = None,
3737
**kwargs):
3838
warn(
39-
f"{self.__class__.__name__} is deprecated. Please leverage ydata-sdk RegularSynthesizer or TimeSeriesSynthesizer instead. For more information, "
40-
f"check ydata-sdk documentation: https://docs.fabric.ydata.ai/latest/sdk/examples/synthesize_tabular_data/.",
39+
f"{self.__class__.__name__} is deprecated. Please leverage ydata-sdk **RegularSynthesizer** or **TimeSeriesSynthesizer** instead. For more information, "
40+
f"check ydata-sdk documentation: https://docs.fabric.ydata.ai/latest/sdk/examples/.",
4141
DeprecationWarning,
4242
stacklevel=2
4343
)

src/ydata_synthetic/synthesizers/timeseries/doppelganger/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""
22
Doppelganger implementation file
33
"""
4+
from ydata_synthetic.synthesizers.base import BaseModel
45

56
class DoppelGANgerNetwork(BaseModel):
67
"""

src/ydata_synthetic/synthesizers/timeseries/timegan/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""
22
TimeGAN file definition
33
"""
4+
from ydata_synthetic.synthesizers.base import BaseModel
45

56
class TimeGAN(BaseModel):
67
"""

0 commit comments

Comments
 (0)