Skip to content

Commit

Permalink
fix: update import for TimeGAN (#321)
Browse files Browse the repository at this point in the history
* fix: update import for TimeGAN

* feat: update requirements version

* fix: allow up to python 3.11 as 3.12 is not supported by tf
  • Loading branch information
aquemy authored Dec 28, 2023
1 parent 5a7036f commit 7d1ffb9
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 8 deletions.
8 changes: 4 additions & 4 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -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
Expand Down
6 changes: 3 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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",
],
},
)
2 changes: 1 addition & 1 deletion src/ydata_synthetic/streamlit_app/pages/functions/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}
Expand Down

0 comments on commit 7d1ffb9

Please sign in to comment.