Merge pull request #83 from giotherobot/patch-1

Actually loosen jax and pytorch dependencies
conda-forge · Mar 21, 2024 · 03d2cd3 · 03d2cd3
2 parents 201c987 + b88e5e6
commit 03d2cd3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/.azure-pipelines/azure-pipelines-linux.yml b/.azure-pipelines/azure-pipelines-linux.yml
diff --git a/.scripts/build_steps.sh b/.scripts/build_steps.sh
diff --git a/recipe/meta.yaml b/recipe/meta.yaml
@@ -15,7 +15,7 @@ source:
 build:
   script: {{ PYTHON }} -m pip install . -vv
   noarch: python
-  number: 0
+  number: 1
 
 requirements:
   host:
@@ -33,17 +33,17 @@ requirements:
     - ml_dtypes
   run_constrained:
     - tensorflow >=2.15.0,<2.17.0a
-    - jax 0.4
-    - torch 2.1
+    - jax 0.4.*
+    - pytorch 2.1.*
 
 test:
   requires:
     - pip
+    - jax
   commands:
     - pip check
-    # even the numpy backend requires jax....
-    # https://github.com/keras-team/keras/issues/18840
-    # - KERAS_BACKEND=numpy python -c "import keras"
+    - KERAS_BACKEND=numpy python -c "import keras"
+    - KERAS_BACKEND=jax python -c "import keras"
 
 about:
   home: https://github.com/keras-team/keras