Make pip install calls compatible with zsh

.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: |

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).

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

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:

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

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

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}",
     },