Update dockerfiles

containers/dockerfile-template

@@ -30,8 +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 {conda_samplers}
-RUN pip install nessai
+RUN conda install -n ${{conda_env}} -c conda-forge -c pytorch {conda_samplers}
 
 # Install Polychord
 RUN apt-get update --allow-releaseinfo-change

containers/v3-dockerfile-test-suite-python310

@@ -32,8 +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 pymultinest ultranest cpnest kombine dnest4 zeus-mcmc
-RUN pip install nessai
+RUN conda install -n ${conda_env} -c conda-forge dynesty emcee nestle ptemcee pymultinest ultranest cpnest kombine dnest4 zeus-mcmc pytorch theano-pymc3 pymc nessai
 
 # Install Polychord
 RUN apt-get update --allow-releaseinfo-change

containers/v3-dockerfile-test-suite-python38

@@ -32,8 +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 pymultinest ultranest cpnest kombine dnest4 zeus-mcmc pytorch theano-pymc pymc3
-RUN pip install nessai
+RUN conda install -n ${conda_env} -c conda-forge dynesty emcee nestle ptemcee pymultinest ultranest cpnest kombine dnest4 zeus-mcmc pytorch theano-pymc pymc nessai
 
 # Install Polychord
 RUN apt-get update --allow-releaseinfo-change

containers/v3-dockerfile-test-suite-python39

@@ -32,8 +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 pymultinest ultranest cpnest kombine dnest4 zeus-mcmc pytorch theano-pymc pymc3
-RUN pip install nessai
+RUN conda install -n ${conda_env} -c conda-forge dynesty emcee nestle ptemcee pymultinest ultranest cpnest kombine dnest4 zeus-mcmc pytorch theano-pymc pymc nessai
 
 # Install Polychord
 RUN apt-get update --allow-releaseinfo-change

containers/write_dockerfiles.py

@@ -10,16 +10,17 @@ conda_sampler_dict = dict(
     python38=[
         "dynesty", "emcee", "nestle", "ptemcee", "pymultinest", "ultranest",
         "cpnest", "kombine", "dnest4", "zeus-mcmc",
-        "pytorch", "theano-pymc", "pymc3"
+        "pytorch", "theano-pymc", "pymc", "nessai",
     ],
     python39=[
         "dynesty", "emcee", "nestle", "ptemcee", "pymultinest", "ultranest",
         "cpnest", "kombine", "dnest4", "zeus-mcmc",
-        "pytorch", "theano-pymc", "pymc3"
+        "pytorch", "theano-pymc", "pymc", "nessai",
     ],
     python310=[
         "dynesty", "emcee", "nestle", "ptemcee", "pymultinest", "ultranest",
         "cpnest", "kombine", "dnest4", "zeus-mcmc",
+        "pytorch", "theano-pymc3", "pymc", "nessai",
     ]
 )