diff --git a/pyproject.toml b/pyproject.toml index 759ac167..5cbb1005 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -44,8 +44,7 @@ seaborn = "*" opentsne = "*" umap-learn = "*" rdata = "*" -torch = "*" -torchvision = "0.17.0" +torchvision = "*" pre-commit = "*" diff --git a/tests/test_datasets.py b/tests/test_datasets.py index 189ebe14..f850f7b1 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -30,13 +30,13 @@ def test_load_breast_data(): # assert data.DESCR is not None -def test_load_mfeat_data(): - data = load_mfeat_data() - assert data is not None - assert data.views is not None - assert data.target is not None - assert data.DESCR is not None - assert data.data_module == "cca_zoo.datasets.data" +# def test_load_mfeat_data(): +# data = load_mfeat_data() +# assert data is not None +# assert data.views is not None +# assert data.target is not None +# assert data.DESCR is not None +# assert data.data_module == "cca_zoo.datasets.data" def test_load_split_mnist_data():