added req for BNN, GP and DeepEnsemble

tloehrlmu · Aug 8, 2024 · 530b496 · 530b496
1 parent 2befc58
commit 530b496
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -8,7 +8,6 @@ mapie
 properscoring
 scipy
 pyparsing
-torch
 tensorflow
 statsmodels
 tqdm
@@ -18,13 +17,13 @@ networkx
 mpltern
 venn-abers
 pycalib
-arviz
-pymc
 xarray
 aesara
 seaborn
 ffmpeg-python
+pymc==5.15.1
+arviz==0.18.0
 pytensor==2.22.1
-sphinx-proof==0.1.3
-pyro-ppl==1.9.1
-gpytorch==1.11
+gpytorch==1.11
+torch==2.3.0
+torchvision==0.18.0