Remove theano from 310

containers/dockerfile-template

@@ -30,12 +30,7 @@ RUN conda install -n ${{conda_env}} -c conda-forge scikit-image celerite george
 
 # Install dependencies and samplers
 RUN pip install corner healpy cython tables
-RUN conda install -n ${{conda_env}} -c conda-forge dynesty emcee nestle ptemcee
-RUN conda install -n ${{conda_env}} -c conda-forge pymultinest ultranest
-RUN conda install -n ${{conda_env}} -c conda-forge cpnest kombine dnest4 zeus-mcmc
-RUN conda install -n ${{conda_env}} -c conda-forge pytorch
-RUN conda install -n ${{conda_env}} -c conda-forge theano-pymc
-RUN conda install -n ${{conda_env}} -c conda-forge pymc3
+RUN conda install -n ${{conda_env}} -c conda-forge {conda_samplers}
 RUN pip install nessai
 
 # Install Polychord

containers/v3-dockerfile-test-suite-python310

@@ -32,12 +32,7 @@ RUN conda install -n ${conda_env} -c conda-forge scikit-image celerite george
 
 # Install dependencies and samplers
 RUN pip install corner healpy cython tables
-RUN conda install -n ${conda_env} -c conda-forge dynesty emcee nestle ptemcee
-RUN conda install -n ${conda_env} -c conda-forge pymultinest ultranest
-RUN conda install -n ${conda_env} -c conda-forge cpnest kombine dnest4 zeus-mcmc
-RUN conda install -n ${conda_env} -c conda-forge pytorch
-RUN conda install -n ${conda_env} -c conda-forge theano-pymc
-RUN conda install -n ${conda_env} -c conda-forge pymc3
+RUN conda install -n ${conda_env} -c conda-forge dynesty emcee nestle ptemcee pymultinest ultranest cpnest kombine dnest4 zues-mcmc
 RUN pip install nessai
 
 # Install Polychord

containers/v3-dockerfile-test-suite-python38

@@ -32,12 +32,7 @@ RUN conda install -n ${conda_env} -c conda-forge scikit-image celerite george
 
 # Install dependencies and samplers
 RUN pip install corner healpy cython tables
-RUN conda install -n ${conda_env} -c conda-forge dynesty emcee nestle ptemcee
-RUN conda install -n ${conda_env} -c conda-forge pymultinest ultranest
-RUN conda install -n ${conda_env} -c conda-forge cpnest kombine dnest4 zeus-mcmc
-RUN conda install -n ${conda_env} -c conda-forge pytorch
-RUN conda install -n ${conda_env} -c conda-forge theano-pymc
-RUN conda install -n ${conda_env} -c conda-forge pymc3
+RUN conda install -n ${conda_env} -c conda-forge dynesty emcee nestle ptemcee pymultinest ultranest cpnest kombine dnest4 zues-mcmc pytorch theano-pymc pymc3
 RUN pip install nessai
 
 # Install Polychord

containers/v3-dockerfile-test-suite-python39

@@ -32,12 +32,7 @@ RUN conda install -n ${conda_env} -c conda-forge scikit-image celerite george
 
 # Install dependencies and samplers
 RUN pip install corner healpy cython tables
-RUN conda install -n ${conda_env} -c conda-forge dynesty emcee nestle ptemcee
-RUN conda install -n ${conda_env} -c conda-forge pymultinest ultranest
-RUN conda install -n ${conda_env} -c conda-forge cpnest kombine dnest4 zeus-mcmc
-RUN conda install -n ${conda_env} -c conda-forge pytorch
-RUN conda install -n ${conda_env} -c conda-forge theano-pymc
-RUN conda install -n ${conda_env} -c conda-forge pymc3
+RUN conda install -n ${conda_env} -c conda-forge dynesty emcee nestle ptemcee pymultinest ultranest cpnest kombine dnest4 zues-mcmc pytorch theano-pymc pymc3
 RUN pip install nessai
 
 # Install Polychord

containers/write_dockerfiles.py

@@ -6,10 +6,27 @@ with open("dockerfile-template", "r") as ff:
 python_versions = [(3, 8), (3, 9), (3, 10)]
 today = date.today().strftime("%Y%m%d")
 
+conda_sampler_dict = dict(
+    python38=[
+        "dynesty", "emcee", "nestle", "ptemcee", "pymultinest", "ultranest",
+        "cpnest", "kombine", "dnest4", "zues-mcmc",
+        "pytorch", "theano-pymc", "pymc3"
+    ],
+    python39=[
+        "dynesty", "emcee", "nestle", "ptemcee", "pymultinest", "ultranest",
+        "cpnest", "kombine", "dnest4", "zues-mcmc",
+        "pytorch", "theano-pymc", "pymc3"
+    ],
+    python310=[
+        "dynesty", "emcee", "nestle", "ptemcee", "pymultinest", "ultranest",
+        "cpnest", "kombine", "dnest4", "zues-mcmc",
+    ]
+)
+
 for python_major_version, python_minor_version in python_versions:
+    key = f"python{python_major_version}{python_minor_version}"
     with open(
-        "v3-dockerfile-test-suite-python"
-        f"{python_major_version}{python_minor_version}",
+        f"v3-dockerfile-test-suite-{key}",
         "w"
     ) as ff:
         ff.write(
@@ -19,5 +36,6 @@ for python_major_version, python_minor_version in python_versions:
         ff.write(template.format(
             date=today,
             python_major_version=python_major_version,
-            python_minor_version=python_minor_version
+            python_minor_version=python_minor_version,
+            conda_samplers=" ".join(conda_sampler_dict[key])
         ))