feat(models): use sparse arrays for posterior samples

MODULE.bazel

@@ -4,7 +4,7 @@ pyrovelocity MODULE
 
 module(
     name = "pyrovelocity",
-    version = "0.3.0b4",
+    version = "0.3.0b5",
     compatibility_level = 1,
 )
 

Makefile

@@ -913,8 +913,8 @@ approve-prs: ## Approve github pull requests from bots: PR_ENTRIES="2-5 10 12-18
 		fi; \
 	done
 
-PREVIOUS_VERSION := 0.3.0b3
-NEXT_VERSION := 0.3.0b4
+PREVIOUS_VERSION := 0.3.0b4
+NEXT_VERSION := 0.3.0b5
 VERSION_FILES := \
 	pyproject.toml \
 	conda/colab/construct.yaml \

conda/colab/construct.yaml

@@ -1,5 +1,5 @@
 name: pyrovelocity-colab
-version: 0.3.0b4
+version: 0.3.0b5
 
 channels:
   - pytorch
@@ -22,12 +22,12 @@ specs:
   - adal=1.2.7=pyhd8ed1ab_0
   - adjusttext=1.2.0=pyhd8ed1ab_0
   - adlfs=2024.7.0=pyhd8ed1ab_0
-  - aiobotocore=2.13.1=pyhd8ed1ab_0
+  - aiobotocore=2.13.2=pyhd8ed1ab_0
   - aiohttp=3.9.5=py311h459d7ec_0
   - aioitertools=0.11.0=pyhd8ed1ab_0
   - aiosignal=1.3.1=pyhd8ed1ab_0
   - alembic=1.13.2=pyhd8ed1ab_0
-  - altair=5.3.0=pyhd8ed1ab_0
+  - altair=5.4.0=pyhd8ed1ab_0
   - aniso8601=9.0.1=pyhd8ed1ab_0
   - anndata=0.10.8=pyhd8ed1ab_0
   - annotated-types=0.7.0=pyhd8ed1ab_0
@@ -86,7 +86,7 @@ specs:
   - brotli-bin=1.1.0=hd590300_1
   - brotli-python=1.1.0=py311hb755f60_1
   - bzip2=1.0.8=h4bc722e_7
-  - c-ares=1.32.3=h4bc722e_0
+  - c-ares=1.33.0=ha66036c_0
   - ca-certificates=2024.7.4=hbcca054_0
   - cachecontrol=0.14.0=pyhd8ed1ab_1
   - cachecontrol-with-filecache=0.14.0=pyhd8ed1ab_1
@@ -165,14 +165,14 @@ specs:
   - exceptiongroup=1.2.2=pyhd8ed1ab_0
   - executing=2.0.1=pyhd8ed1ab_0
   - expat=2.6.2=h59595ed_0
-  - expression=5.0.2=pyhd8ed1ab_0
+  - expression=5.1.0=pyhd8ed1ab_0
   - fastapi=0.112.0=pyhd8ed1ab_0
   - fastapi-cli=0.0.5=pyhd8ed1ab_0
   - filelock=3.15.4=pyhd8ed1ab_0
   - flask=3.0.3=pyhd8ed1ab_0
   - flax=0.8.5=pyhd8ed1ab_0
-  - flyteidl=1.13.0=pyhd8ed1ab_0
-  - flytekit=1.13.0=pyhd8ed1ab_0
+  - flyteidl=1.13.1=pyhd8ed1ab_0
+  - flytekit=1.13.3=pyhd8ed1ab_0
   - fmt=10.2.1=h00ab1b0_0
   - font-ttf-dejavu-sans-mono=2.37=hab24e00_0
   - font-ttf-inconsolata=3.000=h77eed37_0
@@ -191,8 +191,8 @@ specs:
   - gcsfs=2024.6.1=pyhd8ed1ab_0
   - gdk-pixbuf=2.42.12=hb9ae30d_0
   - get-annotations=0.1.2=pyhd8ed1ab_0
-  - gettext=0.22.5=h59595ed_2
-  - gettext-tools=0.22.5=h59595ed_2
+  - gettext=0.22.5=he02047a_3
+  - gettext-tools=0.22.5=he02047a_3
   - gflags=2.2.2=he1b5a44_1004
   - gitdb=4.0.11=pyhd8ed1ab_0
   - gitpython=3.1.43=pyhd8ed1ab_0
@@ -206,9 +206,9 @@ specs:
   - google-auth-httplib2=0.2.0=pyhd8ed1ab_0
   - google-auth-oauthlib=1.2.1=pyhd8ed1ab_0
   - google-cloud-core=2.4.1=pyhd8ed1ab_0
-  - google-cloud-storage=2.18.1=pyhff2d567_0
+  - google-cloud-storage=2.18.2=pyhff2d567_0
   - google-crc32c=1.1.2=py311h9b08b9c_5
-  - google-resumable-media=2.7.0=pyhd8ed1ab_0
+  - google-resumable-media=2.7.2=pyhd8ed1ab_1
   - googleapis-common-protos=1.63.2=pyhd8ed1ab_0
   - graphene=3.3=pyhd8ed1ab_0
   - graphite2=1.3.13=h59595ed_1003
@@ -237,7 +237,7 @@ specs:
   - hydra-core=1.3.2=pyhd8ed1ab_0
   - hydra-zen=0.13.0=pyhd8ed1ab_0
   - hyperframe=6.0.1=pyhd8ed1ab_0
-  - hypothesis=6.110.1=pyha770c72_0
+  - hypothesis=6.111.0=pyha770c72_0
   - ibis-framework=9.3.0=hd8ed1ab_0
   - ibis-framework-core=9.3.0=pyhd8ed1ab_0
   - icu=75.1=he02047a_0
@@ -290,8 +290,8 @@ specs:
   - libarrow-flight-sql=15.0.2=h61825be_19_cpu
   - libarrow-gandiva=15.0.2=hfcb4b9d_19_cpu
   - libarrow-substrait=15.0.2=h61825be_19_cpu
-  - libasprintf=0.22.5=h661eb56_2
-  - libasprintf-devel=0.22.5=h661eb56_2
+  - libasprintf=0.22.5=he8f35ee_3
+  - libasprintf-devel=0.22.5=he8f35ee_3
   - libavif16=1.1.1=h9b56c87_0
   - libblas=3.9.0=16_linux64_mkl
   - libbrotlicommon=1.1.0=hd590300_1
@@ -316,12 +316,12 @@ specs:
   - libgcc-ng=14.1.0=h77fa898_0
   - libgcrypt=1.11.0=h4ab18f5_1
   - libgd=2.3.3=hd3e95f3_10
-  - libgettextpo=0.22.5=h59595ed_2
-  - libgettextpo-devel=0.22.5=h59595ed_2
+  - libgettextpo=0.22.5=he02047a_3
+  - libgettextpo-devel=0.22.5=he02047a_3
   - libgfortran-ng=14.1.0=h69a702a_0
   - libgfortran5=14.1.0=hc5f4f2c_0
   - libgirepository=1.80.1=h003a4f0_0
-  - libglib=2.80.3=h8a4344b_1
+  - libglib=2.80.3=h315aac3_2
   - libgomp=14.1.0=h77fa898_0
   - libgoogle-cloud=2.26.0=h26d7fe4_0
   - libgoogle-cloud-storage=2.26.0=ha262f82_0
@@ -401,6 +401,7 @@ specs:
   - multipledispatch=0.6.0=pyhd8ed1ab_1
   - munkres=1.1.4=pyh9f0ad1d_0
   - mypy_extensions=1.0.0=pyha770c72_0
+  - narwhals=1.4.0=pyhd8ed1ab_0
   - natsort=8.4.0=pyhd8ed1ab_0
   - nbclient=0.10.0=pyhd8ed1ab_0
   - nbconvert=7.16.4=hd8ed1ab_1
@@ -418,7 +419,7 @@ specs:
   - oauthlib=3.2.2=pyhd8ed1ab_0
   - omegaconf=2.3.0=pyhd8ed1ab_0
   - openjpeg=2.5.2=h488ebb8_0
-  - openpyxl=3.1.4=py311h459d7ec_0
+  - openpyxl=3.1.5=py311h459d7ec_0
   - openssl=3.3.1=h4bc722e_2
   - opentelemetry-api=1.16.0=pyhd8ed1ab_0
   - opentelemetry-sdk=1.16.0=pyhd8ed1ab_0
@@ -430,19 +431,19 @@ specs:
   - orbax-checkpoint=0.4.4=pyhd8ed1ab_0
   - orc=2.0.1=h17fec99_1
   - ordered-set=4.1.0=pyhd8ed1ab_0
-  - orjson=3.10.6=py311hb3a8bbb_0
+  - orjson=3.10.7=py311hb3a8bbb_0
   - outcome=1.3.0.post0=pyhd8ed1ab_0
   - packaging=24.1=pyhd8ed1ab_0
   - pandas=2.2.2=py311h14de704_1
   - pandoc=3.3=ha770c72_0
   - pandocfilters=1.5.0=pyhd8ed1ab_0
   - pango=1.54.0=h4c5309f_1
   - papermill=2.6.0=pyhd8ed1ab_0
-  - paramiko=3.4.0=pyhd8ed1ab_0
+  - paramiko=3.4.1=pyhd8ed1ab_0
   - parso=0.8.4=pyhd8ed1ab_0
   - parsy=2.1=pyhd8ed1ab_0
   - patsy=0.5.6=pyhd8ed1ab_0
-  - pcre2=10.44=h0f59acf_0
+  - pcre2=10.44=hba22ea6_2
   - pexpect=4.9.0=pyhd8ed1ab_0
   - pickleshare=0.7.5=py_1003
   - pillow=10.4.0=py311h82a398c_0
@@ -521,12 +522,12 @@ specs:
   - pyyaml=6.0.2=py311h61187de_0
   - pyzmq=26.1.0=py311h759c1eb_0
   - qhull=2020.2=h434a139_5
-  - querystring_parser=1.2.4=py_0
+  - querystring_parser=1.2.4=pyhd8ed1ab_1
   - rapidfuzz=3.9.6=py311hf86e51f_0
   - rav1e=0.6.6=he8a937b_2
   - rdma-core=53.0=he02047a_0
   - re2=2023.09.01=h7f4b329_2
-  - readchar=4.1.0=pyhd8ed1ab_0
+  - readchar=4.2.0=pyhd8ed1ab_0
   - readline=8.2=h8228510_1
   - referencing=0.35.1=pyhd8ed1ab_0
   - regex=2024.7.24=py311h61187de_0
@@ -570,9 +571,9 @@ specs:
   - statsd=4.0.1=pyh1a96a4e_0
   - statsmodels=0.14.2=py311h18e1886_0
   - stdlib-list=0.10.0=pyhd8ed1ab_0
-  - stringcase=1.2.0=py_0
+  - stringcase=1.2.0=pyhd8ed1ab_1
   - svt-av1=2.1.2=hac33072_0
-  - sympy=1.13.0=pypyh2585a3b_103
+  - sympy=1.13.2=pypyh2585a3b_103
   - sysroot_linux-64=2.17=h4a8ded7_16
   - tbb=2021.12.0=h434a139_3
   - tenacity=9.0.0=pyhd8ed1ab_0
@@ -586,7 +587,7 @@ specs:
   - tk=8.6.13=noxft_h4845f30_101
   - toml=0.10.2=pyhd8ed1ab_0
   - tomli=2.0.1=pyhd8ed1ab_0
-  - tomlkit=0.13.0=pyha770c72_0
+  - tomlkit=0.13.2=pyha770c72_0
   - toolz=0.12.1=pyhd8ed1ab_0
   - torchmetrics=1.4.0.post0=pyhd8ed1ab_0
   - torchtriton=3.0.0=py311
@@ -645,7 +646,7 @@ specs:
   - yaml=0.2.5=h7f98852_2
   - yarl=1.9.4=py311h459d7ec_0
   - zeromq=4.3.5=h75354e8_4
-  - zipp=3.19.2=pyhd8ed1ab_0
+  - zipp=3.20.0=pyhd8ed1ab_0
   - zlib=1.3.1=h4ab18f5_1
   - zstandard=0.23.0=py311h5cd10c7_0
   - zstd=1.5.6=ha6fb4c9_0