Merge pull request #83 from alan-turing-institute/IFenton-patch-1

Temporary pinning of jaxtyping
alan-turing-institute · Nov 28, 2024 · 40b6c0a · 40b6c0a
2 parents ce127af + 7eb1ffa
commit 40b6c0a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -39,10 +39,10 @@ dependencies = [
     "typer",
     "lightning",
     "torch>=2.3.0",  # needed for numpy 2.0
-    "jaxtyping",
+    "jaxtyping<=0.2.34", # currently >0.2.34 causing errors
     "wandb",
     "tqdm",
-    "moviepy>=1.0.3",
+    "moviepy==1.0.3", # currently >1.0.3 not working with wandb
     "imageio>=2.35.1",
     "numpy <2.1.0",  # https://github.com/wandb/wandb/issues/8166
     "chex",