diff --git a/conda/meta.yaml b/conda/meta.yaml index 1a3a97731..7d77035da 100644 --- a/conda/meta.yaml +++ b/conda/meta.yaml @@ -26,7 +26,7 @@ requirements: - scikit-learn >=0.23,<1 - pytorch >=1.4,<2 - torchvision >=0.5.0,<1 - - rdt >=0.4.2,<0.5 + - rdt >=0.4.2,<0.6 run: - packaging - python >=3.6,<3.9 @@ -35,7 +35,7 @@ requirements: - scikit-learn >=0.23,<1 - pytorch >=1.4,<2 - torchvision >=0.5.0,<1 - - rdt >=0.4.2,<0.5 + - rdt >=0.4.2,<0.6 about: home: "https://github.com/sdv-dev/CTGAN" diff --git a/setup.py b/setup.py index 5facdce89..629cfd4c6 100644 --- a/setup.py +++ b/setup.py @@ -18,7 +18,7 @@ 'scikit-learn>=0.23,<1', 'torch>=1.4,<2', 'torchvision>=0.5.0,<1', - 'rdt>=0.4.2,<0.5', + 'rdt>=0.4.2,<0.6', ] setup_requires = [