📌 Use flaxmodels from PyPI

Installation from PyPI as PyPI packages can't have direct dependencies. "Invalid value for requires_dist. Error: Can't have direct dependency"

pyproject.toml

@@ -36,9 +36,6 @@ Source = "https://github.com/alvarobartt/safejax"
 [tool.hatch.version]
 path = "src/safejax/__init__.py"
 
-[tool.hatch.metadata]
-allow-direct-references = true
-
 [project.optional-dependencies]
 quality = [
   "black~=22.10.0",
@@ -48,7 +45,7 @@ quality = [
 tests = [
   "pytest~=7.1.2",
   "pytest-lazy-fixture~=0.6.3",
-  "flaxmodels @ git+https://github.com/matthias-wright/flaxmodels.git",
+  "flaxmodels~=0.1.2",
 ]
 
 [tool.hatch.envs.quality]