[dev] add torchdata dependency into DGL requirements (#6801)

80763d4b · Rhett Ying · GitHub · 2b966da2 · 80763d4b · 80763d4b
Unverified Commit 80763d4b authored Dec 21, 2023 by Rhett Ying Committed by GitHub Dec 21, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

python/setup.py python/setup.py +1 -0

script/dgl_dev.yml.template script/dgl_dev.yml.template +1 -1

No files found.
--- a/python/setup.py
+++ b/python/setup.py
@@ -226,6 +226,7 @@ setup(
        "requests>=2.19.0",
        "tqdm",
        "psutil>=5.8.0",
+        "torchdata>=0.5.0",
    ],
    url="https://github.com/dmlc/dgl",
    distclass=BinaryDistribution,

--- a/script/dgl_dev.yml.template
+++ b/script/dgl_dev.yml.template
@@ -27,7 +27,7 @@ dependencies:
    - scikit-learn
    - scipy
    - torch==__TORCH_VERSION__
-    - torchdata
+    - torchdata>=0.5.0
    - torcheval
    - torchmetrics
    - tqdm