Fix requirements (#11)

Author: mathpluscode

docker/environment.yml

@@ -5,7 +5,7 @@ dependencies:
   - python=3.8
   - pip=22.3.1
   - pip:
-      - tensorflow-cpu==2.11.0
+      - tensorflow-cpu==2.12.0
       - jax==0.4.1
       - jaxlib==0.4.1
       - -r requirements.txt

docker/environment_mac_m1.yml

@@ -6,8 +6,8 @@ dependencies:
   - pip=22.3.1
   - grpcio==1.43.0
   - pip:
-      - tensorflow-macos==2.11.0
-      - tensorflow-metal==0.7.0
+      - tensorflow-macos==2.12.0
+      - tensorflow-metal==0.8.0
       - jax==0.4.1
       - jaxlib==0.4.1
       - -r requirements.txt

docker/environment_tpu.yml

@@ -8,7 +8,7 @@ dependencies:
   - pip:
       - --find-links
         https://storage.googleapis.com/jax-releases/libtpu_releases.html
-      - tensorflow-cpu==2.11.0
+      - tensorflow-cpu==2.12.0
       - jax[tpu]==0.4.1
       - jaxlib==0.4.1
       - -r requirements.txt