[FIX] Add mkl version, limit tensordict to 0.6.2 (#161)

Adds a constraint on the `mkl` and `tensordict` versions to prevent
installing incompatible versions of PyTorch, as well as incompatible
versions of `tensordict`. We were previously installing the cpu-only
version of PyTorch in nighly tests and tensordict 0.7.2, which appears
to be incompatible with *some* versions of PyTorch 2.6 (version 0.6.2
works without issue).

Author: alexbarghi-nv

dependencies.yaml

@@ -78,6 +78,7 @@ files:
       - depends_on_pytorch
       - py_version
       - test_python_common
+      - depends_on_mkl
       - depends_on_pylibwholegraph
       - depends_on_cugraph_dgl
   test_cugraph_pyg:
@@ -90,6 +91,7 @@ files:
       - depends_on_pytorch
       - py_version
       - test_python_common
+      - depends_on_mkl
       - depends_on_pylibwholegraph
       - depends_on_cugraph_pyg
   test_pylibwholegraph:
@@ -436,7 +438,7 @@ dependencies:
       - output_types: [conda]
         packages:
           - dgl>=2.4.0.cu*
-          - &tensordict tensordict>=0.1.2
+          - &tensordict tensordict>=0.1.2,<=0.6.2
   cugraph_pyg_dev:
     common:
       - output_types: [conda]
@@ -452,7 +454,7 @@ dependencies:
       - output_types: [requirements, pyproject]
         packages:
           - torch>=2.3
-          - tensordict
+          - tensordict>=0.1.2,<=0.6.2
     specific:
       - output_types: [requirements]
         matrices:

python/cugraph-dgl/conda/cugraph_dgl_dev_cuda-118.yaml

@@ -17,6 +17,6 @@ dependencies:
 - pytest-cov
 - pytest-xdist
 - pytorch>=2.3,<=2.5.1
-- tensordict>=0.1.2
+- tensordict>=0.1.2,<=0.6.2
 - torchdata
 name: cugraph_dgl_dev_cuda-118

python/cugraph-dgl/pyproject.toml

@@ -36,7 +36,7 @@ test = [
     "pytest-benchmark",
     "pytest-cov",
     "pytest-xdist",
-    "tensordict",
+    "tensordict>=0.1.2,<=0.6.2",
     "torch>=2.3",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 

python/cugraph-pyg/conda/cugraph_pyg_dev_cuda-118.yaml

@@ -16,6 +16,6 @@ dependencies:
 - pytest-xdist
 - pytorch>=2.3,<=2.5.1
 - pytorch_geometric>=2.5,<2.7
-- tensordict>=0.1.2
+- tensordict>=0.1.2,<=0.6.2
 - torchdata
 name: cugraph_pyg_dev_cuda-118

python/cugraph-pyg/pyproject.toml

@@ -49,7 +49,7 @@ test = [
     "pytest-benchmark",
     "pytest-cov",
     "pytest-xdist",
-    "tensordict",
+    "tensordict>=0.1.2,<=0.6.2",
     "torch-geometric>=2.5,<2.7",
     "torch>=2.3",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.