fix refactored packages

infer-actively · Sep 24, 2024 · cc48366 · cc48366
1 parent 57d730d
commit cc48366
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -49,10 +49,10 @@ Repository = "https://github.com/infer-actively/pymdp"
 packages = [
     'pymdp',
     'pymdp.envs',
-    'pymdp.algos',
-    'pymdp.jax',
-    'pymdp.jax.envs',
-    'pymdp.jax.planning',
+    'pymdp.planning',
+    'pymdp.legacy',
+    'pymdp.legacy.algos',
+    'pymdp.legacy.envs',
 ]
 
 [tool.black]