From 2f673316d185dfc55341e0bf1fd8b43cff7a822a Mon Sep 17 00:00:00 2001
From: Fabiana <30911746+fabclmnt@users.noreply.github.com>
Date: Thu, 29 Aug 2024 15:57:19 +0100
Subject: [PATCH] fix: add import BaseModel to timeseries models

---
 src/ydata_synthetic/synthesizers/base.py                      | 4 ++--
 .../synthesizers/timeseries/doppelganger/model.py             | 1 +
 src/ydata_synthetic/synthesizers/timeseries/timegan/model.py  | 1 +
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/ydata_synthetic/synthesizers/base.py b/src/ydata_synthetic/synthesizers/base.py
index 5a17fb71..bc564afe 100644
--- a/src/ydata_synthetic/synthesizers/base.py
+++ b/src/ydata_synthetic/synthesizers/base.py
@@ -36,8 +36,8 @@ def __init__(self,
                   cat_cols: Optional[List[str]] = None,
                   **kwargs):
         warn(
-            f"{self.__class__.__name__} is deprecated. Please leverage ydata-sdk RegularSynthesizer or TimeSeriesSynthesizer instead. For more information, "
-            f"check ydata-sdk documentation:  https://docs.fabric.ydata.ai/latest/sdk/examples/synthesize_tabular_data/.",
+            f"{self.__class__.__name__} is deprecated. Please leverage ydata-sdk **RegularSynthesizer** or **TimeSeriesSynthesizer** instead. For more information, "
+            f"check ydata-sdk documentation:  https://docs.fabric.ydata.ai/latest/sdk/examples/.",
             DeprecationWarning,
             stacklevel=2
         )
\ No newline at end of file
diff --git a/src/ydata_synthetic/synthesizers/timeseries/doppelganger/model.py b/src/ydata_synthetic/synthesizers/timeseries/doppelganger/model.py
index 4fc22f6b..b5331a5a 100644
--- a/src/ydata_synthetic/synthesizers/timeseries/doppelganger/model.py
+++ b/src/ydata_synthetic/synthesizers/timeseries/doppelganger/model.py
@@ -1,6 +1,7 @@
 """
     Doppelganger implementation file
 """
+from ydata_synthetic.synthesizers.base import BaseModel
 
 class DoppelGANgerNetwork(BaseModel):
     """
diff --git a/src/ydata_synthetic/synthesizers/timeseries/timegan/model.py b/src/ydata_synthetic/synthesizers/timeseries/timegan/model.py
index cf5adead..75461e27 100644
--- a/src/ydata_synthetic/synthesizers/timeseries/timegan/model.py
+++ b/src/ydata_synthetic/synthesizers/timeseries/timegan/model.py
@@ -1,6 +1,7 @@
 """
     TimeGAN file definition
 """
+from ydata_synthetic.synthesizers.base import BaseModel
 
 class TimeGAN(BaseModel):
     """