diff --git a/pyproject.toml b/pyproject.toml index e53de42b..aea4751a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -48,7 +48,7 @@ dependencies = [ rdt = { main = 'rdt.cli.__main__:main' } [project.optional-dependencies] -copulas = ['copulas>=0.10.0,<0.11',] +copulas = ['copulas@git+ssh://git@github.com/sdv-dev/copulas@main#egg=copulas',] test = [ 'rdt[copulas]', diff --git a/rdt/transformers/categorical.py b/rdt/transformers/categorical.py index 16ae6ff0..e077f5be 100644 --- a/rdt/transformers/categorical.py +++ b/rdt/transformers/categorical.py @@ -6,7 +6,6 @@ import numpy as np import pandas as pd from scipy.stats import norm -from pandas.api.types import is_numeric_dtype from rdt.errors import TransformerInputError from rdt.transformers.base import BaseTransformer diff --git a/tests/integration/transformers/test_categorical.py b/tests/integration/transformers/test_categorical.py index 5ae02b66..907df485 100644 --- a/tests/integration/transformers/test_categorical.py +++ b/tests/integration/transformers/test_categorical.py @@ -474,6 +474,7 @@ def test_one_hot_categoricals(): # Run transformed_data = transformer.fit_transform(test_data, column='A') + # Assert pd.testing.assert_frame_equal( transformed_data,