Add zeus-mcmc to docker files

containers/dockerfile-template

@@ -34,7 +34,7 @@ RUN pip install sphinx numpydoc nbsphinx sphinx_rtd_theme sphinx-tabs autodoc
 
 # Install dependencies and samplers
 RUN pip install corner lalsuite theano healpy cython tables
-RUN pip install cpnest dynesty emcee nestle ptemcee pymc3 kombine dnest4 nessai
+RUN pip install cpnest dynesty emcee nestle ptemcee pymc3 kombine dnest4 nessai zeus-mcmc
 RUN conda install -n ${{conda_env}} -c conda-forge pymultinest ultranest
 
 # Install Polychord

containers/v2-dockerfile-test-suite-python35

@@ -36,7 +36,7 @@ RUN pip install sphinx numpydoc nbsphinx sphinx_rtd_theme sphinx-tabs autodoc
 
 # Install dependencies and samplers
 RUN pip install corner lalsuite theano healpy cython tables
-RUN pip install cpnest dynesty emcee nestle ptemcee pymc3 kombine dnest4 nessai
+RUN pip install cpnest dynesty emcee nestle ptemcee pymc3 kombine dnest4 nessai 
 RUN conda install -n ${conda_env} -c conda-forge pymultinest ultranest
 
 # Install Polychord

containers/v2-dockerfile-test-suite-python36

@@ -36,7 +36,7 @@ RUN pip install sphinx numpydoc nbsphinx sphinx_rtd_theme sphinx-tabs autodoc
 
 # Install dependencies and samplers
 RUN pip install corner lalsuite theano healpy cython tables
-RUN pip install cpnest dynesty emcee nestle ptemcee pymc3 kombine dnest4 nessai
+RUN pip install cpnest dynesty emcee nestle ptemcee pymc3 kombine dnest4 nessai zeus-mcmc
 RUN conda install -n ${conda_env} -c conda-forge pymultinest ultranest
 
 # Install Polychord

containers/v2-dockerfile-test-suite-python37

@@ -36,7 +36,7 @@ RUN pip install sphinx numpydoc nbsphinx sphinx_rtd_theme sphinx-tabs autodoc
 
 # Install dependencies and samplers
 RUN pip install corner lalsuite theano healpy cython tables
-RUN pip install cpnest dynesty emcee nestle ptemcee pymc3 kombine dnest4 nessai
+RUN pip install cpnest dynesty emcee nestle ptemcee pymc3 kombine dnest4 nessai zeus-mcmc
 RUN conda install -n ${conda_env} -c conda-forge pymultinest ultranest
 
 # Install Polychord

containers/v2-dockerfile-test-suite-python38

@@ -36,7 +36,7 @@ RUN pip install sphinx numpydoc nbsphinx sphinx_rtd_theme sphinx-tabs autodoc
 
 # Install dependencies and samplers
 RUN pip install corner lalsuite theano healpy cython tables
-RUN pip install cpnest dynesty emcee nestle ptemcee pymc3 kombine dnest4 nessai
+RUN pip install cpnest dynesty emcee nestle ptemcee pymc3 kombine dnest4 nessai zeus-mcmc
 RUN conda install -n ${conda_env} -c conda-forge pymultinest ultranest
 
 # Install Polychord

containers/write_dockerfiles.py

@@ -7,6 +7,12 @@ python_versions = [(3, 5), (3, 6), (3, 7), (3, 8)]
 today = date.today().strftime("%Y%m%d")
 
 for python_major_version, python_minor_version in python_versions:
+    # remove zeus from Python 3.5 version
+    if python_major_version == 3 and python_minor_version == 5:
+        edittemplate = template.replace("zeus-mcmc", "")
+    else:
+        edittemplate = template
+
     with open(
         "v2-dockerfile-test-suite-python"
         f"{python_major_version}{python_minor_version}",
@@ -16,7 +22,7 @@ for python_major_version, python_minor_version in python_versions:
             "# This dockerfile is written automatically and should not be "
             "modified by hand.\n\n"
         )
-        ff.write(template.format(
+        ff.write(edittemplate.format(
             date=today,
             python_major_version=python_major_version,
             python_minor_version=python_minor_version