Fix pip installation

thebuckleylab · Jun 12, 2024 · 8924006 · 8924006
1 parent bf774cb
commit 8924006
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 9 deletions.
diff --git a/jpc/__init__.py b/jpc/__init__.py
@@ -17,4 +17,4 @@
 )
 
 
-__version__ = importlib.metadata.version("jpc_local")
+__version__ = importlib.metadata.version("jpc")
diff --git a/jpc/_grads.py b/jpc/_grads.py
@@ -4,7 +4,7 @@
 from equinox import filter_grad
 from jaxtyping import PyTree, ArrayLike, Array
 from typing import Union, Tuple, Callable, Optional
-from _energies import _energy_fn, pc_energy_fn, _lateral_energy_fn
+from ._energies import _energy_fn, pc_energy_fn, _lateral_energy_fn
 
 
 def _neg_activity_grad(

diff --git a/jpc/_infer.py b/jpc/_infer.py
@@ -2,7 +2,7 @@
 
 from jaxtyping import PyTree, ArrayLike, Array
 from typing import Callable, Optional, Union
-from _grads import _neg_activity_grad, _neg_lateral_activity_grad
+from ._grads import _neg_activity_grad, _neg_lateral_activity_grad
 from diffrax import (
     AbstractSolver,
     AbstractStepSizeController,

diff --git a/pyproject.toml b/pyproject.toml
@@ -3,8 +3,8 @@ name = "jpc"
 version = "0.0.1"
 description = "Predictive coding networks in JAX."
 readme = "README.md"
-requires-python ="~=3.8"
-#license = {file = "LICENSE"}
+requires-python ="~=3.9"
+license = {file = "LICENSE"}
 authors = [
   {name = "Francesco Innocenti", email = "F.Innocenti@sussex.ac.uk"},
 ]
@@ -33,10 +33,10 @@ classifiers = [
 ]
 urls = {repository = "https://github.com/thebuckleylab/jpc"}
 dependencies = [
-    "jax>=0.4.13",
-    "equinox>=0.10.4",
-    "diffrax>=0.3.1",
-    "jaxtyping>=0.2.19"
+    "jax>=0.4.23",
+    "equinox>=0.11.2",
+    "diffrax>=0.5.1",
+    "jaxtyping>=0.2.24"
 ]
 
 [build-system]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -17,4 +17,4 @@
		)


		__version__ = importlib.metadata.version("jpc_local")
		__version__ = importlib.metadata.version("jpc")