Add optax to environment

StevePny · Jun 2, 2024 · f4a9e5a · f4a9e5a
1 parent d1e2597
commit f4a9e5a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/environment.yml b/environment.yml
@@ -164,6 +164,7 @@ dependencies:
   - openjpeg=2.5.0=hfec8fc6_2
   - openssl=3.0.7=h0b41bf4_2
   - opt_einsum=3.3.0=pyhd8ed1ab_1
+  - optax=0.1.4=py310h06a4308_0
   - packaging=23.0=pyhd8ed1ab_0
   - pandas=1.5.3=py310h9b08913_0
   - partd=1.3.0=pyhd8ed1ab_0