Add module for optimization and losses

.gitignore

@@ -1,3 +1,6 @@
+# temporary ignoring
+src/pymc_elicito/optimization.py
+
 # Notebooks
 *.ipynb
 

pyproject.toml

@@ -8,7 +8,10 @@ authors = [
 license = { text = "placeholder" }
 requires-python = ">=3.9"
 dependencies = [
+    "numpy>=2.0.2",
     "pandas>=2.3.3",
+    "pymc>=5.12.0",
+    "pytensor>=2.19.0",
 ]
 readme = "README.md"
 classifiers = [
@@ -41,7 +44,7 @@ Issues = "https://github.com/pymc-devs/pymc-elicito/issues"
 
 [project.optional-dependencies]
 plots = [
-    "matplotlib>=3.7.1",
+    "matplotlib>=3.9.4",
 ]
 full = [
     "pymc-elicito[plots]",

requirements-docs-locked.txt

@@ -5,34 +5,45 @@ appnope==0.1.4 ; sys_platform == 'darwin'
 argon2-cffi==23.1.0
 argon2-cffi-bindings==21.2.0
 arrow==1.3.0
+arviz==0.16.0 ; python_full_version < '3.10'
+arviz==0.22.0 ; python_full_version >= '3.10'
 asttokens==3.0.0
 async-lru==2.0.4
 attrs==25.3.0
 babel==2.16.0
 backrefs==5.9
 beautifulsoup4==4.12.3
 bleach==6.2.0
+cachetools==6.2.2
 certifi==2024.12.14
 cffi==1.17.1
 charset-normalizer==3.4.1
 click==8.1.8
+cloudpickle==3.1.2
 colorama==0.4.6
 comm==0.2.2
+cons==0.4.7
 contourpy==1.3.0 ; python_full_version < '3.10'
 contourpy==1.3.2 ; python_full_version == '3.10.*'
 contourpy==1.3.3 ; python_full_version >= '3.11'
 cycler==0.12.1
 debugpy==1.8.11
 decorator==5.1.1
 defusedxml==0.7.1
+etuples==0.3.10
 exceptiongroup==1.3.0 ; python_full_version < '3.11'
 executing==2.1.0
 fastjsonschema==2.21.1
+fastprogress==1.0.3 ; python_full_version < '3.10'
+filelock==3.16.1
 fonttools==4.60.1
 fqdn==1.5.1
 ghp-import==2.1.0
 griffe==1.11.0
 h11==0.14.0
+h5netcdf==1.7.3
+h5py==3.14.0 ; python_full_version < '3.10'
+h5py==3.15.1 ; python_full_version >= '3.10'
 httpcore==1.0.7
 httpx==0.28.1
 idna==3.10
@@ -62,6 +73,7 @@ jupyterlab-server==2.27.3
 jupytext==1.17.2
 kiwisolver==1.4.7 ; python_full_version < '3.10'
 kiwisolver==1.4.9 ; python_full_version >= '3.10'
+logical-unification==0.4.7
 markdown==3.7
 markdown-it-py==3.0.0
 markupsafe==3.0.2
@@ -71,6 +83,7 @@ matplotlib-inline==0.1.7
 mdit-py-plugins==0.4.2
 mdurl==0.1.2
 mergedeep==1.3.4
+minikanren==1.0.5
 mistune==3.0.2
 mkdocs==1.6.1
 mkdocs-autorefs==1.4.2
@@ -84,6 +97,7 @@ mkdocs-section-index==0.3.10
 mkdocstrings==0.30.0
 mkdocstrings-python==1.16.12
 mkdocstrings-python-xref==1.16.3
+multipledispatch==1.0.0
 nbclient==0.10.2
 nbconvert==7.16.4
 nbformat==5.10.4
@@ -110,8 +124,14 @@ ptyprocess==0.7.0 ; (python_full_version < '3.10' and sys_platform == 'emscripte
 pure-eval==0.2.3
 pycparser==2.22
 pygments==2.19.1
+pymc==5.12.0 ; python_full_version < '3.10'
+pymc==5.25.1 ; python_full_version == '3.10.*'
+pymc==5.26.1 ; python_full_version >= '3.11'
 pymdown-extensions==10.16.1
 pyparsing==3.2.5
+pytensor==2.19.0 ; python_full_version < '3.10'
+pytensor==2.31.7 ; python_full_version == '3.10.*'
+pytensor==2.35.1 ; python_full_version >= '3.11'
 python-dateutil==2.9.0.post0
 python-json-logger==3.2.1
 pytz==2025.2
@@ -124,21 +144,27 @@ referencing==0.35.1
 requests==2.32.3
 rfc3339-validator==0.1.4
 rfc3986-validator==0.1.1
+rich==13.9.4 ; python_full_version >= '3.10'
 rpds-py==0.22.3
 ruff==0.12.8
+scipy==1.13.1 ; python_full_version < '3.10'
+scipy==1.15.3 ; python_full_version == '3.10.*'
+scipy==1.16.3 ; python_full_version >= '3.11'
 send2trash==1.8.3
 setuptools==75.6.0
 six==1.17.0
 sniffio==1.3.1
 soupsieve==2.6
 stack-data==0.6.3
 terminado==0.18.1
+threadpoolctl==3.6.0 ; python_full_version >= '3.10'
 tinycss2==1.4.0
 tomli==2.2.1 ; python_full_version < '3.11'
+toolz==1.1.0
 tornado==6.4.2
 traitlets==5.14.3
 types-python-dateutil==2.9.0.20241206
-typing-extensions==4.12.2 ; python_full_version < '3.13'
+typing-extensions==4.12.2
 tzdata==2025.2
 uri-template==1.3.0
 urllib3==2.3.0
@@ -147,4 +173,10 @@ wcwidth==0.2.13
 webcolors==24.11.1
 webencodings==0.5.1
 websocket-client==1.8.0
+xarray==2024.7.0 ; python_full_version < '3.10'
+xarray==2025.6.1 ; python_full_version == '3.10.*'
+xarray==2025.11.0 ; python_full_version >= '3.11'
+xarray-einstats==0.7.0 ; python_full_version < '3.10'
+xarray-einstats==0.8.0 ; python_full_version == '3.10.*'
+xarray-einstats==0.9.1 ; python_full_version >= '3.11'
 zipp==3.23.0 ; python_full_version < '3.10'

requirements-incl-optional-locked.txt

@@ -1,25 +1,62 @@
 # This file was autogenerated by uv via the following command:
 #    uv export -o requirements-incl-optional-locked.txt --no-hashes --no-dev --no-emit-project --all-extras
+arviz==0.16.0 ; python_full_version < '3.10'
+arviz==0.22.0 ; python_full_version >= '3.10'
+cachetools==6.2.2
+cloudpickle==3.1.2
+cons==0.4.7
 contourpy==1.3.0 ; python_full_version < '3.10'
 contourpy==1.3.2 ; python_full_version == '3.10.*'
 contourpy==1.3.3 ; python_full_version >= '3.11'
 cycler==0.12.1
+etuples==0.3.10
+fastprogress==1.0.3 ; python_full_version < '3.10'
+filelock==3.16.1
 fonttools==4.60.1
+h5netcdf==1.7.3
+h5py==3.14.0 ; python_full_version < '3.10'
+h5py==3.15.1 ; python_full_version >= '3.10'
 importlib-resources==6.5.2 ; python_full_version < '3.10'
 kiwisolver==1.4.7 ; python_full_version < '3.10'
 kiwisolver==1.4.9 ; python_full_version >= '3.10'
+logical-unification==0.4.7
+markdown-it-py==3.0.0 ; python_full_version >= '3.10'
 matplotlib==3.9.4 ; python_full_version < '3.10'
 matplotlib==3.10.7 ; python_full_version >= '3.10'
+mdurl==0.1.2 ; python_full_version >= '3.10'
+minikanren==1.0.5
+multipledispatch==1.0.0
 numpy==2.0.2 ; python_full_version < '3.10'
 numpy==2.2.6 ; python_full_version == '3.10.*'
 numpy==2.3.5 ; python_full_version >= '3.11'
 packaging==24.2
 pandas==2.3.3
 pillow==11.3.0 ; python_full_version < '3.10'
 pillow==12.0.0 ; python_full_version >= '3.10'
+pygments==2.19.1 ; python_full_version >= '3.10'
+pymc==5.12.0 ; python_full_version < '3.10'
+pymc==5.25.1 ; python_full_version == '3.10.*'
+pymc==5.26.1 ; python_full_version >= '3.11'
 pyparsing==3.2.5
+pytensor==2.19.0 ; python_full_version < '3.10'
+pytensor==2.31.7 ; python_full_version == '3.10.*'
+pytensor==2.35.1 ; python_full_version >= '3.11'
 python-dateutil==2.9.0.post0
 pytz==2025.2
+rich==13.9.4 ; python_full_version >= '3.10'
+scipy==1.13.1 ; python_full_version < '3.10'
+scipy==1.15.3 ; python_full_version == '3.10.*'
+scipy==1.16.3 ; python_full_version >= '3.11'
+setuptools==75.6.0
 six==1.17.0
+threadpoolctl==3.6.0 ; python_full_version >= '3.10'
+toolz==1.1.0
+typing-extensions==4.12.2
 tzdata==2025.2
+xarray==2024.7.0 ; python_full_version < '3.10'
+xarray==2025.6.1 ; python_full_version == '3.10.*'
+xarray==2025.11.0 ; python_full_version >= '3.11'
+xarray-einstats==0.7.0 ; python_full_version < '3.10'
+xarray-einstats==0.8.0 ; python_full_version == '3.10.*'
+xarray-einstats==0.9.1 ; python_full_version >= '3.11'
 zipp==3.23.0 ; python_full_version < '3.10'

requirements-locked.txt

@@ -1,10 +1,62 @@
 # This file was autogenerated by uv via the following command:
 #    uv export -o requirements-locked.txt --no-hashes --no-dev --no-emit-project
+arviz==0.16.0 ; python_full_version < '3.10'
+arviz==0.22.0 ; python_full_version >= '3.10'
+cachetools==6.2.2
+cloudpickle==3.1.2
+cons==0.4.7
+contourpy==1.3.0 ; python_full_version < '3.10'
+contourpy==1.3.2 ; python_full_version == '3.10.*'
+contourpy==1.3.3 ; python_full_version >= '3.11'
+cycler==0.12.1
+etuples==0.3.10
+fastprogress==1.0.3 ; python_full_version < '3.10'
+filelock==3.16.1
+fonttools==4.60.1
+h5netcdf==1.7.3
+h5py==3.14.0 ; python_full_version < '3.10'
+h5py==3.15.1 ; python_full_version >= '3.10'
+importlib-resources==6.5.2 ; python_full_version < '3.10'
+kiwisolver==1.4.7 ; python_full_version < '3.10'
+kiwisolver==1.4.9 ; python_full_version >= '3.10'
+logical-unification==0.4.7
+markdown-it-py==3.0.0 ; python_full_version >= '3.10'
+matplotlib==3.9.4 ; python_full_version < '3.10'
+matplotlib==3.10.7 ; python_full_version >= '3.10'
+mdurl==0.1.2 ; python_full_version >= '3.10'
+minikanren==1.0.5
+multipledispatch==1.0.0
 numpy==2.0.2 ; python_full_version < '3.10'
 numpy==2.2.6 ; python_full_version == '3.10.*'
 numpy==2.3.5 ; python_full_version >= '3.11'
+packaging==24.2
 pandas==2.3.3
+pillow==11.3.0 ; python_full_version < '3.10'
+pillow==12.0.0 ; python_full_version >= '3.10'
+pygments==2.19.1 ; python_full_version >= '3.10'
+pymc==5.12.0 ; python_full_version < '3.10'
+pymc==5.25.1 ; python_full_version == '3.10.*'
+pymc==5.26.1 ; python_full_version >= '3.11'
+pyparsing==3.2.5
+pytensor==2.19.0 ; python_full_version < '3.10'
+pytensor==2.31.7 ; python_full_version == '3.10.*'
+pytensor==2.35.1 ; python_full_version >= '3.11'
 python-dateutil==2.9.0.post0
 pytz==2025.2
+rich==13.9.4 ; python_full_version >= '3.10'
+scipy==1.13.1 ; python_full_version < '3.10'
+scipy==1.15.3 ; python_full_version == '3.10.*'
+scipy==1.16.3 ; python_full_version >= '3.11'
+setuptools==75.6.0
 six==1.17.0
+threadpoolctl==3.6.0 ; python_full_version >= '3.10'
+toolz==1.1.0
+typing-extensions==4.12.2
 tzdata==2025.2
+xarray==2024.7.0 ; python_full_version < '3.10'
+xarray==2025.6.1 ; python_full_version == '3.10.*'
+xarray==2025.11.0 ; python_full_version >= '3.11'
+xarray-einstats==0.7.0 ; python_full_version < '3.10'
+xarray-einstats==0.8.0 ; python_full_version == '3.10.*'
+xarray-einstats==0.9.1 ; python_full_version >= '3.11'
+zipp==3.23.0 ; python_full_version < '3.10'