From 7d1ffb914eaec6d440ea4340500d765fd05dbfef Mon Sep 17 00:00:00 2001
From: Alexandre Quemy <alexandre.quemy@ydata.ai>
Date: Thu, 28 Dec 2023 14:20:48 +0100
Subject: [PATCH] fix: update import for TimeGAN (#321)

* fix: update import for TimeGAN

* feat: update requirements version

* fix: allow up to python 3.11 as 3.12 is not supported by tf
---
 requirements.txt                                          | 8 ++++----
 setup.py                                                  | 6 +++---
 .../streamlit_app/pages/functions/train.py                | 2 +-
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 3da1b656..8ed4264b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,8 +1,8 @@
 requests>=2.30, <2.31
-pandas==2.0.*
-numpy==1.23.*
-scikit-learn==1.3.*
-matplotlib==3.7.*
+pandas<3
+numpy<2
+scikit-learn<2
+matplotlib<4
 tensorflow==2.12.0
 tensorflow-probability==0.19.0
 easydict==1.10
diff --git a/setup.py b/setup.py
index 9185edca..05c2b92b 100644
--- a/setup.py
+++ b/setup.py
@@ -42,7 +42,7 @@
       keywords='data science ydata',
       url='https://github.com/ydataai/ydata-synthetic',
       license="https://github.com/ydataai/ydata-synthetic/blob/master/LICENSE",
-      python_requires=">=3.9, <3.11",
+      python_requires=">=3.9, <3.12",
       packages=find_namespace_packages('src'),
       package_dir={'':'src'},
       include_package_data=True,
@@ -53,8 +53,8 @@
               "streamlit==1.29.0",
               "typing-extensions>=3.10.0",
               "streamlit_pandas_profiling==0.1.3",
-              "ydata-profiling==4.6.3",
-              "ydata-sdk>=0.2.1"
+              "ydata-profiling<5",
+              "ydata-sdk>=0.2.1",
           ],
       },
       )
diff --git a/src/ydata_synthetic/streamlit_app/pages/functions/train.py b/src/ydata_synthetic/streamlit_app/pages/functions/train.py
index c2ede838..e6dd35da 100644
--- a/src/ydata_synthetic/streamlit_app/pages/functions/train.py
+++ b/src/ydata_synthetic/streamlit_app/pages/functions/train.py
@@ -5,7 +5,7 @@
 import streamlit as st
 
 from ydata_synthetic.synthesizers.regular import RegularSynthesizer
-from ydata_synthetic.synthesizers.timeseries import TimeGAN
+from ydata_synthetic.synthesizers.timeseries.timegan.model import TimeGAN
 from ydata_synthetic.synthesizers import ModelParameters
 
 __MODEL_MAPPING = {'tabular': RegularSynthesizer, 'timeseries': TimeGAN}