Skip to content

Commit 2d545b9

Browse files
Deprecate TensorDictFeatureStore in cuGraph-PyG (#160)
First step in handling #159 . The unified API will store all data in WholeGraph, which has superior performance and lower memory usage. `TensorDictFeatureStore` is no longer a useful alternative. Authors: - Alex Barghi (https://github.com/alexbarghi-nv) Approvers: - Tingyu Wang (https://github.com/tingyu66) URL: #160
1 parent b173063 commit 2d545b9

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

python/cugraph-pyg/cugraph_pyg/data/__init__.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,22 @@
1818
)
1919
from cugraph_pyg.data.graph_store import GraphStore
2020
from cugraph_pyg.data.feature_store import (
21-
TensorDictFeatureStore,
21+
TensorDictFeatureStore as DEPRECATED__TensorDictFeatureStore,
2222
WholeFeatureStore,
2323
)
2424

2525

26+
def TensorDictFeatureStore(*args, **kwargs):
27+
warnings.warn(
28+
"TensorDictFeatureStore is deprecated. Consider changing your "
29+
"workflow to launch using 'torchrun' and store data in "
30+
"the faster and more memory-efficient WholeFeatureStore instead.",
31+
FutureWarning,
32+
)
33+
34+
return DEPRECATED__TensorDictFeatureStore(*args, **kwargs)
35+
36+
2637
def DaskGraphStore(*args, **kwargs):
2738
warnings.warn(
2839
"DaskGraphStore and the Dask API are deprecated."

0 commit comments

Comments
 (0)