CI: try using cotengra@althash

jcmgray · Dec 5, 2023 · ba21dab · ba21dab
1 parent 884a466
commit ba21dab
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 6 deletions.
diff --git a/ci/requirements/py-base.yml b/ci/requirements/py-base.yml
@@ -3,7 +3,7 @@ channels:
   - conda-forge
 dependencies:
   - autoray>=0.6.7
-  - cotengra>=0.5.3
+  # - cotengra>=0.5.3
   - coverage
   - cytoolz
   - joblib
@@ -16,3 +16,5 @@ dependencies:
   - pytest-cov
   - scipy
   - tqdm
+  - pip:
+    - git+https://github.com/jcmgray/cotengra.git@althash
diff --git a/ci/requirements/py-jax.yml b/ci/requirements/py-jax.yml
@@ -3,7 +3,7 @@ channels:
   - conda-forge
 dependencies:
   - autoray>=0.6.7
-  - cotengra>=0.5.3
+  # - cotengra>=0.5.3
   - coverage
   - cytoolz
   - joblib
@@ -19,3 +19,4 @@ dependencies:
   - pip
   - pip:
     - jax[cpu]
+    - git+https://github.com/jcmgray/cotengra.git@althash
diff --git a/ci/requirements/py-slepc.yml b/ci/requirements/py-slepc.yml
@@ -3,7 +3,7 @@ channels:
   - defaults
 dependencies:
   - autoray>=0.6.7
-  - cotengra>=0.5.3
+  # - cotengra>=0.5.3
   - coverage
   - cytoolz
   - joblib
@@ -22,3 +22,5 @@ dependencies:
   - slepc=*=*complex*
   - slepc4py
   - tqdm
+  - pip:
+    - git+https://github.com/jcmgray/cotengra.git@althash
diff --git a/ci/requirements/py-tensorflow.yml b/ci/requirements/py-tensorflow.yml
@@ -3,7 +3,7 @@ channels:
   - conda-forge
 dependencies:
   - autoray>=0.6.7
-  - cotengra>=0.5.3
+  # - cotengra>=0.5.3
   - coverage
   - cytoolz
   - joblib
@@ -19,3 +19,4 @@ dependencies:
   - pip
   - pip:
     - tensorflow
+    - git+https://github.com/username/repo.git@branch_name
diff --git a/ci/requirements/py-torch.yml b/ci/requirements/py-torch.yml
@@ -4,7 +4,7 @@ channels:
   - conda-forge
 dependencies:
   - autoray>=0.6.7
-  - cotengra>=0.5.3
+  # - cotengra>=0.5.3
   - coverage
   - cpuonly
   - cytoolz
@@ -19,3 +19,5 @@ dependencies:
   - pytorch
   - scipy
   - tqdm
+  - pip:
+    - git+https://github.com/username/repo.git@branch_name
diff --git a/tests/test_tensor/test_tensor_2d.py b/tests/test_tensor/test_tensor_2d.py
@@ -273,7 +273,7 @@ def test_contract_hotrg_two_layer_rand_peps(self):
             bond_dim=2,
         )
         norm = psi.make_norm()
-        xe = norm.contract(all, optimize="auto-hq", cache_expression=False)
+        xe = norm.contract(all, optimize="auto-hq")
         xt = norm.contract_hotrg(max_bond=5)
         assert xt == pytest.approx(xe, rel=1e-4)