diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index d38c1834..4fb1b419 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -23,4 +23,4 @@ jobs:
         pip install pytest
     - name: Run pytest
       run: |
-        pytest --ignore=tests/test_embeddings.py
\ No newline at end of file
+        pytest tests/
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index fbed156e..c991dd85 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,6 +38,7 @@ datasets = ">=2.9.0"
 huggingface_hub = ">=0.12.0"
 tensorboard = ">=2.10.0"
 scikit-learn = ">=1.0.0"
+numpy = {version="1.24.*", optional=true}
 gensim = {version="4.3.2", optional=true}
 seqeval = {version=">=1.2.0", optional=true}
 jiwer = {version=">=3.*", optional=true}
@@ -51,10 +52,9 @@ rouge_score = {version="*", optional=true}
 nlp = ["seqeval", "jiwer", "nltk", "rouge_score"]
 audio = ["soundfile", "librosa", "jiwer"]
 vision = ["pillow"]
-embeddings = ["gensim"]
-docs = ["sphinx", "myst-parser", "furo", "sphinx-copybutton"]
-linting = ["ruff", "black"]
-all = ["gensim", "seqeval", "jiwer", "soundfile", "librosa", "pillow", "nltk", "rouge_score"]
+embeddings = ["gensim", "numpy"]
+dev = ["pytest", "ruff", "sphinx", "myst-parser", "furo", "sphinx-copybutton"]
+all = ["numpy", "gensim", "seqeval", "jiwer", "soundfile", "librosa", "pillow", "nltk", "rouge_score"]
 
 [tool.black]
 line-length = 120