From b2a2ecea65bca98e25cd6df6a82d7e672dc5d4d2 Mon Sep 17 00:00:00 2001
From: Sasha <srush.research@gmail.com>
Date: Mon, 19 Aug 2024 11:36:39 -0400
Subject: [PATCH] update

---
 setup.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index eb87e43..1128164 100644
--- a/setup.py
+++ b/setup.py
@@ -32,9 +32,12 @@
     install_requires=[
         "toolz",
         "colour",
-        "svgwrite",
-        "Pillow",
-        "loguru",
+        "imageio",
+        "numpy",
+        "matplotlib",
+        "jax",
+        "chex",
+        "jaxtyping",
         "typing-extensions",
         "importlib-metadata",
     ],