update setup and requirements

containers/cleanair/gpjax_models/requirements.txt

@@ -1,26 +1,22 @@
-typer
-loguru
-slurmjobs
-sacred
-seml
-scikit-learn
-tabulate
-jupyter
-PyYAML
-black
-pandas 
-matplotlib
-geopandas
-rtree
-dvc
-convertbng
-rich
+git+https://github.com/defaultobject/stdata@update_polygon_patch
 git+https://github.com/defaultobject/batchjax
 git+https://github.com/defaultobject/stgp
-tensorflow-probability
-jaxlib
-bibtexparser
-flake8
-flake8-docstrings
-bng_latlon
+typer==0.12.3
+loguru==0.7.2
+slurmjobs==1.1.3
+sacred==0.8.5
+seml==0.5.1
+tabulate==0.9.0
+jupyter==1.0.0
+PyYAML==6.0.1
+black==24.4.2
+rtree==1.2.0
+dvc==3.51.2
+convertbng==0.7.2
+rich==13.7.1
+bibtexparser==1.4.1
+flake8==7.1.0
+flake8-docstrings==1.7.0
+bng-latlon==1.1
+colorlog==6.7.0
 

containers/cleanair/gpjax_models/setup.py

@@ -20,17 +20,18 @@
 
 # Dependencies
 INSTALL_REQUIRES = [
-    "jax",
-    "numpy",
-    "pandas==2.1.3",
+    "jax==0.4.25",
+    "jaxlib==0.4.27",
+    "numpy==1.26.4",
+    "pandas==2.2.2",
     "pydantic==1.10.12",
-    "scikit-learn",
+    "scikit-learn==1.5.1",
     "scipy==1.11.2",
-    "typer==0.9.0",
     "tinygp==0.3.0",
-    "tensorflow-probability",
-    "colorlog==6.7.0",
+    "tensorflow-probability==0.23.0",
     "nptyping==2.5.0",
+    "matplotlib==3.9.1",
+    "geopandas==1.0.1",
 ]
 SETUP_REQUIRES = ["setuptools_scm"]