Removes remaining references to experimental.datasets.

rapidsai · Nov 18, 2023 · aea91f1 · aea91f1
1 parent ccead3e
commit aea91f1
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 22 deletions.
diff --git a/python/cugraph/cugraph/dask/community/leiden.py b/python/cugraph/cugraph/dask/community/leiden.py
@@ -125,7 +125,7 @@ def leiden(
 
     Examples
     --------
-    >>> from cugraph.experimental.datasets import karate
+    >>> from cugraph.datasets import karate
     >>> G = karate.get_graph(fetch=True)
     >>> parts, modularity_score = cugraph.leiden(G)
 

diff --git a/python/cugraph/cugraph/dask/community/louvain.py b/python/cugraph/cugraph/dask/community/louvain.py
@@ -129,7 +129,7 @@ def louvain(
 
     Examples
     --------
-    >>> from cugraph.experimental.datasets import karate
+    >>> from cugraph.datasets import karate
     >>> G = karate.get_graph(fetch=True)
     >>> parts = cugraph.louvain(G)
 

diff --git a/python/cugraph/cugraph/tests/centrality/test_edge_betweenness_centrality_mg.py b/python/cugraph/cugraph/tests/centrality/test_edge_betweenness_centrality_mg.py
@@ -16,7 +16,7 @@
 
 import dask_cudf
 from pylibcugraph.testing.utils import gen_fixture_params_product
-from cugraph.experimental.datasets import DATASETS_UNDIRECTED
+from cugraph.datasets import DATASETS_UNDIRECTED
 
 import cugraph
 import cugraph.dask as dcg

diff --git a/python/cugraph/cugraph/tests/nx/test_compat_pr.py b/python/cugraph/cugraph/tests/nx/test_compat_pr.py
@@ -24,7 +24,7 @@
 import numpy as np
 
 from cugraph.testing import utils
-from cugraph.experimental.datasets import karate
+from cugraph.datasets import karate
 from pylibcugraph.testing.utils import gen_fixture_params_product
 
 

diff --git a/python/cugraph/cugraph/tests/utils/test_dataset.py b/python/cugraph/cugraph/tests/utils/test_dataset.py
@@ -13,8 +13,6 @@
 
 import os
 import gc
-import sys
-import warnings
 from pathlib import Path
 from tempfile import TemporaryDirectory
 
@@ -75,22 +73,6 @@ def setup(tmpdir):
     gc.collect()
 
 
-@pytest.fixture()
-def setup_deprecation_warning_tests():
-    """
-    Fixture used to set warning filters to 'default' and reload
-    experimental.datasets module if it has been previously
-    imported. Tests that import this fixture are expected to
-    import cugraph.experimental.datasets
-    """
-    warnings.filterwarnings("default")
-
-    if "cugraph.experimental.datasets" in sys.modules:
-        del sys.modules["cugraph.experimental.datasets"]
-
-    yield
-
-
 ###############################################################################
 # Helpers