From f1ce3e15a5917824ef183b3e74b4855440cc8249 Mon Sep 17 00:00:00 2001 From: Alexandria Barghi Date: Fri, 29 Dec 2023 11:08:22 -0800 Subject: [PATCH] Fix mixed experimental import --- python/cugraph/cugraph/experimental/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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