heavyball constraint, cleaned commit

TBBristol · TBBristol · commit 1d4ed0e924bc · 2025-10-02T15:39:50.000+01:00
diff --git a/setup.py b/setup.py
@@ -258,7 +258,7 @@ def run(self):
         'rich_argparse',
         'imageio',
         'pyro-ppl',
-        'heavyball',
+        'heavyball<2.0.0',
         'neptune',
         'wandb',
     ]
