Readd PyTorch and relax upperbound

rapidsai · Aug 27, 2024 · 3384bc6 · 3384bc6
1 parent d54b7d8
commit 3384bc6
Show file tree

Hide file tree

Showing 12 changed files with 40 additions and 20 deletions.
diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -57,6 +57,7 @@ dependencies:
 - pytest-mpl
 - pytest-xdist
 - python-louvain
+- pytorch>=2.0
 - raft-dask==24.10.*,>=0.0.0a0
 - rapids-build-backend>=0.3.1,<0.4.0.dev0
 - rapids-dask-dependency==24.10.*,>=0.0.0a0

diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -62,6 +62,7 @@ dependencies:
 - pytest-mpl
 - pytest-xdist
 - python-louvain
+- pytorch>=2.0
 - raft-dask==24.10.*,>=0.0.0a0
 - rapids-build-backend>=0.3.1,<0.4.0.dev0
 - rapids-dask-dependency==24.10.*,>=0.0.0a0

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -24,6 +24,7 @@ files:
       - depends_on_pylibcugraphops
       - depends_on_pylibwholegraph
       - depends_on_cupy
+      - depends_on_pytorch
       - python_run_cugraph
       - python_run_nx_cugraph
       - python_run_cugraph_dgl
@@ -65,6 +66,7 @@ files:
       - cuda_version
       - depends_on_cudf
       - depends_on_pylibwholegraph
+      - depends_on_pytorch
       - py_version
       - test_python_common
       - test_python_cugraph
@@ -200,6 +202,7 @@ files:
     includes:
       - test_python_common
       - depends_on_pylibwholegraph
+      - depends_on_pytorch
   py_build_cugraph_pyg:
     output: pyproject
     pyproject_dir: python/cugraph-pyg
@@ -226,6 +229,7 @@ files:
     includes:
       - test_python_common
       - depends_on_pylibwholegraph
+      - depends_on_pytorch
   py_build_cugraph_equivariant:
     output: pyproject
     pyproject_dir: python/cugraph-equivariant
@@ -658,12 +662,10 @@ dependencies:
       - output_types: [conda, pyproject]
         packages:
           - pandas
-          - pydantic
           - pytest
           - pytest-benchmark
           - pytest-cov
           - pytest-xdist
-          - torchdata
           - scipy
   test_python_cugraph:
     common:
@@ -708,6 +710,36 @@ dependencies:
           - *tensordict
           - pyg>=2.5,<2.6
 
+  depends_on_pytorch:
+    common:
+      - output_types: [conda]
+        packages:
+          - &pytorch_unsuffixed pytorch>=2.0
+          - torchdata
+          - pydantic
+
+    specific:
+      - output_types: [requirements]
+        matrices:
+          - matrix: {cuda: "12.*"}
+            packages:
+              - --extra-index-url=https://download.pytorch.org/whl/cu121
+          - matrix: {cuda: "11.*"}
+            packages:
+              - --extra-index-url=https://download.pytorch.org/whl/cu118
+          - {matrix: null, packages: null}
+      - output_types: [requirements, pyproject]
+        matrices:
+          - matrix: {cuda: "12.*"}
+            packages:
+              - &pytorch_pip torch>=2.0
+              - *tensordict
+          - matrix: {cuda: "11.*"}
+            packages:
+              - *pytorch_pip
+              - *tensordict
+          - {matrix: null, packages: [*pytorch_pip, *tensordict]}
+
   depends_on_pylibwholegraph:
     common:
       - output_types: conda

diff --git a/python/cugraph-dgl/conda/cugraph_dgl_dev_cuda-118.yaml b/python/cugraph-dgl/conda/cugraph_dgl_dev_cuda-118.yaml
@@ -13,7 +13,6 @@ dependencies:
 - dgl>=1.1.0.cu*
 - pandas
 - pre-commit
-- pydantic
 - pylibcugraphops==24.10.*,>=0.0.0a0
 - pytest
 - pytest-benchmark
@@ -23,5 +22,4 @@ dependencies:
 - pytorch>=2.0
 - scipy
 - tensordict>=0.1.2
-- torchdata
 name: cugraph_dgl_dev_cuda-118
diff --git a/python/cugraph-dgl/pyproject.toml b/python/cugraph-dgl/pyproject.toml
@@ -33,14 +33,14 @@ dependencies = [
 [project.optional-dependencies]
 test = [
     "pandas",
-    "pydantic",
     "pylibwholegraph==24.10.*,>=0.0.0a0",
     "pytest",
     "pytest-benchmark",
     "pytest-cov",
     "pytest-xdist",
     "scipy",
-    "torchdata",
+    "tensordict>=0.1.2",
+    "torch>=2.0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]

diff --git a/python/cugraph-equivariant/pyproject.toml b/python/cugraph-equivariant/pyproject.toml
@@ -47,13 +47,11 @@ Documentation = "https://docs.rapids.ai/api/cugraph/stable/api_docs/cugraph-ops/
 [project.optional-dependencies]
 test = [
     "pandas",
-    "pydantic",
     "pytest",
     "pytest-benchmark",
     "pytest-cov",
     "pytest-xdist",
     "scipy",
-    "torchdata",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [tool.setuptools]

diff --git a/python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-118.yaml b/python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-118.yaml
@@ -12,7 +12,6 @@ dependencies:
 - cugraph==24.10.*,>=0.0.0a0
 - pandas
 - pre-commit
-- pydantic
 - pyg>=2.5,<2.6
 - pylibcugraphops==24.10.*,>=0.0.0a0
 - pytest
@@ -23,5 +22,4 @@ dependencies:
 - pytorch>=2.0
 - scipy
 - tensordict>=0.1.2
-- torchdata
 name: cugraph_pyg_dev_cuda-118
diff --git a/python/cugraph-pyg/pyproject.toml b/python/cugraph-pyg/pyproject.toml
@@ -42,14 +42,14 @@ Documentation = "https://docs.rapids.ai/api/cugraph/stable/"
 [project.optional-dependencies]
 test = [
     "pandas",
-    "pydantic",
     "pylibwholegraph==24.10.*,>=0.0.0a0",
     "pytest",
     "pytest-benchmark",
     "pytest-cov",
     "pytest-xdist",
     "scipy",
-    "torchdata",
+    "tensordict>=0.1.2",
+    "torch>=2.0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [tool.setuptools]

diff --git a/python/cugraph-service/server/pyproject.toml b/python/cugraph-service/server/pyproject.toml
@@ -49,15 +49,13 @@ test = [
     "networkx>=2.5.1",
     "numpy>=1.23,<3.0a0",
     "pandas",
-    "pydantic",
     "pytest",
     "pytest-benchmark",
     "pytest-cov",
     "pytest-xdist",
     "python-louvain",
     "scikit-learn>=0.23.1",
     "scipy",
-    "torchdata",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]

diff --git a/python/cugraph/pyproject.toml b/python/cugraph/pyproject.toml
@@ -49,7 +49,6 @@ test = [
     "networkx>=2.5.1",
     "numpy>=1.23,<3.0a0",
     "pandas",
-    "pydantic",
     "pylibwholegraph==24.10.*,>=0.0.0a0",
     "pytest",
     "pytest-benchmark",
@@ -58,7 +57,6 @@ test = [
     "python-louvain",
     "scikit-learn>=0.23.1",
     "scipy",
-    "torchdata",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]

diff --git a/python/nx-cugraph/pyproject.toml b/python/nx-cugraph/pyproject.toml
@@ -42,14 +42,12 @@ dependencies = [
 test = [
     "packaging>=21",
     "pandas",
-    "pydantic",
     "pytest",
     "pytest-benchmark",
     "pytest-cov",
     "pytest-mpl",
     "pytest-xdist",
     "scipy",
-    "torchdata",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]

diff --git a/python/pylibcugraph/pyproject.toml b/python/pylibcugraph/pyproject.toml
@@ -43,13 +43,11 @@ test = [
     "cudf==24.10.*,>=0.0.0a0",
     "numpy>=1.23,<3.0a0",
     "pandas",
-    "pydantic",
     "pytest",
     "pytest-benchmark",
     "pytest-cov",
     "pytest-xdist",
     "scipy",
-    "torchdata",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]