diff --git a/python/cugraph/cugraph/experimental/__init__.py b/python/cugraph/cugraph/experimental/__init__.py index 2309a529047..d809e28c92e 100644 --- a/python/cugraph/cugraph/experimental/__init__.py +++ b/python/cugraph/cugraph/experimental/__init__.py @@ -48,9 +48,9 @@ experimental_warning_wrapper(EXPERIMENTAL__find_bicliques) ) -from cugraph.gnn.data_loading import EXPERIMENTAL__BulkSampler +from cugraph.gnn.data_loading import BulkSampler -BulkSampler = experimental_warning_wrapper(EXPERIMENTAL__BulkSampler) +BulkSampler = promoted_experimental_warning_wrapper(BulkSampler) from cugraph.link_prediction.jaccard import jaccard, jaccard_coefficient