From cd47163184f3e9e1665aba779ed8706b7a015258 Mon Sep 17 00:00:00 2001
From: Jirka <jirka.borovec@seznam.cz>
Date: Tue, 12 Sep 2023 09:00:33 +0200
Subject: [PATCH] all

---
 .github/azure-gpu-tests.yml     | 2 +-
 .github/workflows/cpu-tests.yml | 2 +-
 README.md                       | 2 +-
 howto/tpus.md                   | 2 +-
 pyproject.toml                  | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/.github/azure-gpu-tests.yml b/.github/azure-gpu-tests.yml
index 0d61f917..52bb528e 100644
--- a/.github/azure-gpu-tests.yml
+++ b/.github/azure-gpu-tests.yml
@@ -50,7 +50,7 @@ jobs:
         python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu == 2, f'GPU: {mgpu}'"
       displayName: 'Image info & NVIDIA'
 
-    - script: pip install .[dev] pytest
+    - script: pip install .[all] pytest
       displayName: 'Install dependencies'
 
     - bash: pytest -v --durations=10 --disable-pytest-warnings --strict-markers --color=yes
diff --git a/.github/workflows/cpu-tests.yml b/.github/workflows/cpu-tests.yml
index 65815d72..6aeb3d53 100644
--- a/.github/workflows/cpu-tests.yml
+++ b/.github/workflows/cpu-tests.yml
@@ -40,7 +40,7 @@ jobs:
 
     - name: Run tests without the package installed
       run: |
-        pip install ".[dev]" pytest
+        pip install ".[all]" pytest
         pip list
 
         pytest --disable-pytest-warnings --strict-markers --color=yes
diff --git a/README.md b/README.md
index 5544ad42..48d6faee 100644
--- a/README.md
+++ b/README.md
@@ -67,7 +67,7 @@ cd lit-llama
 install dependencies
 
 ```bash
-pip install -e ".[dev]"
+pip install -e ".[all]"
 ```
 
 You are all set! 🎉
diff --git a/howto/tpus.md b/howto/tpus.md
index 816f1c80..c3fdc365 100644
--- a/howto/tpus.md
+++ b/howto/tpus.md
@@ -14,7 +14,7 @@ Now that you are in the machine, let's clone the repository and install the depe
 ```shell
 git clone https://github.com/Lightning-AI/lit-llama
 cd lit-llama
-pip install -e ".[dev]"
+pip install -e ".[all]"
 ```
 
 By default, computations will run using the new (and experimental) PjRT runtime. Still, it's recommended that you set the following environment variables
diff --git a/pyproject.toml b/pyproject.toml
index 07999908..75ad688b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,7 +24,7 @@ classifiers = [
 ]
 
 [project.optional-dependencies]
-dev = [
+all = [
     "tqdm",  # convert_checkpoint.py
     "numpy",  # train.py dataset memmap
     "jsonargparse[signatures]",  # generate.py, convert_checkpoint.py CLI