diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 753639535..793c886ad 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -31,7 +31,7 @@ jobs: python -m pip install --upgrade pip pip install jaxlib pip install jax - pip install .[doc,test] + pip install '.[doc,test]' pip install https://github.com/pyro-ppl/funsor/archive/master.zip pip install -r docs/requirements.txt pip freeze @@ -70,7 +70,7 @@ jobs: pip install jaxlib pip install jax pip install https://github.com/pyro-ppl/funsor/archive/master.zip - pip install -e .[dev,test] + pip install -e '.[dev,test]' pip freeze - name: Test with pytest run: | @@ -109,7 +109,7 @@ jobs: pip install jaxlib pip install jax pip install https://github.com/pyro-ppl/funsor/archive/master.zip - pip install -e .[dev,test] + pip install -e '.[dev,test]' pip freeze - name: Test with pytest run: | @@ -160,7 +160,7 @@ jobs: pip install jaxlib pip install jax pip install https://github.com/pyro-ppl/funsor/archive/master.zip - pip install -e .[dev,examples,test] + pip install -e '.[dev,examples,test]' pip freeze - name: Test with pytest run: | diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index fcfad550d..0977dc2bb 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -12,7 +12,7 @@ To set up local development environment, install NumPyro from source: ```sh git clone https://github.com/pyro-ppl/numpyro.git # install jax/jaxlib first for CUDA support -pip install -e .[dev,test,doc,examples] # contains additional dependencies for NumPyro development +pip install -e '.[dev,test,doc,examples]' # contains additional dependencies for NumPyro development ``` For running `make doctest`, [install pandoc](https://pandoc.org/installing.html). diff --git a/Makefile b/Makefile index c5d6d4a99..642351434 100644 --- a/Makefile +++ b/Makefile @@ -14,7 +14,7 @@ format: license FORCE ruff format . install: FORCE - pip install -e .[dev,doc,test,examples] + pip install -e '.[dev,doc,test,examples]' doctest: FORCE JAX_PLATFORM_NAME=cpu $(MAKE) -C docs doctest diff --git a/README.md b/README.md index 713636107..ccdcf4bed 100644 --- a/README.md +++ b/README.md @@ -240,13 +240,13 @@ In case of compatibility issues arise during execution of the above command, you compatible CPU version of JAX with ``` -pip install numpyro[cpu] +pip install 'numpyro[cpu]' ``` To use **NumPyro on the GPU**, you need to install CUDA first and then use the following pip command: ``` -pip install numpyro[cuda] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html +pip install 'numpyro[cuda]' -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html ``` If you need further guidance, please have a look at the [JAX GPU installation instructions](https://github.com/jax-ml/jax#pip-installation-gpu-cuda). @@ -265,7 +265,7 @@ You can also install NumPyro from source: git clone https://github.com/pyro-ppl/numpyro.git cd numpyro # install jax/jaxlib first for CUDA support -pip install -e .[dev] # contains additional dependencies for NumPyro development +pip install -e '.[dev]' # contains additional dependencies for NumPyro development ``` You can also install NumPyro with conda: diff --git a/docker/dev/Dockerfile b/docker/dev/Dockerfile index da336312f..409ff279d 100644 --- a/docker/dev/Dockerfile +++ b/docker/dev/Dockerfile @@ -20,9 +20,9 @@ RUN apt update && \ ENV PATH=/root/.local/bin:$PATH # install python packages via pip -RUN pip3 install jax[cuda] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html +RUN pip3 install 'jax[cuda]' -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html # clone the numpyro git repository and run pip install RUN git clone https://github.com/pyro-ppl/numpyro.git && \ cd numpyro && \ - pip3 install -e .[dev, test] # contains additional dependencies for NumPyro development + pip3 install -e '.[dev, test]' # contains additional dependencies for NumPyro development diff --git a/docker/release/Dockerfile b/docker/release/Dockerfile index 73755e96c..abb87413f 100644 --- a/docker/release/Dockerfile +++ b/docker/release/Dockerfile @@ -20,4 +20,4 @@ ENV PATH=/root/.local/bin:$PATH RUN pip3 install --user \ # we pull wheels from google's api as per https://github.com/jax-ml/jax#installation # the pre-compiled wheels that google provides work for now. This may change in the future (and necessitate building from source) - numpyro[cuda] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html + 'numpyro[cuda]' -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html diff --git a/setup.py b/setup.py index d062fbfb2..26ea1d53c 100644 --- a/setup.py +++ b/setup.py @@ -82,7 +82,7 @@ ], "cpu": f"jax[cpu]{_jax_version_constraints}", # TPU and CUDA installations, currently require to add package repository URL, i.e., - # pip install numpyro[cuda] -f https://storage.googleapis.com/jax-releases/jax_releases.html + # pip install 'numpyro[cuda]' -f https://storage.googleapis.com/jax-releases/jax_releases.html "tpu": f"jax[tpu]{_jax_version_constraints}", "cuda": f"jax[cuda]{_jax_version_constraints}", },