From 8eb00798d85cf709ac836ad835bad6c1988db356 Mon Sep 17 00:00:00 2001
From: Yx Jiang <2237303+yxjiang@users.noreply.github.com>
Date: Sat, 25 May 2024 14:24:36 -0700
Subject: [PATCH] update

---
 .github/workflows/extract_extra_deps.py | 4 ++--
 .github/workflows/test.yaml             | 4 +---
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/extract_extra_deps.py b/.github/workflows/extract_extra_deps.py
index dc474bf..e124961 100644
--- a/.github/workflows/extract_extra_deps.py
+++ b/.github/workflows/extract_extra_deps.py
@@ -1,6 +1,6 @@
-# .github/workflows/extract_extras.py
+# .github/workflows/extract_extra_deps.py
 import toml
 
 pyproject = toml.load("pyproject.toml")
-extras = ",".join(pyproject["tool"]["poetry"]["extras"].keys())
+extras = " ".join([f"-E {key}" for key in pyproject["tool"]["poetry"]["extras"].keys()])
 print(extras)
diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
index 7ba96d5..fdda89c 100644
--- a/.github/workflows/test.yaml
+++ b/.github/workflows/test.yaml
@@ -35,9 +35,7 @@ jobs:
         python -m pip install toml
         EXTRAS=$(python .github/workflows/extract_extra_deps.py)
         echo "EXTRAS=$EXTRAS"
-        EXTRAS_FLAGS=$(echo $EXTRAS | sed 's/\([^,]\+\)/--extras \1/g')
-        echo "EXTRAS_FLAGS=$EXTRAS_FLAGS"
-        poetry install $EXTRAS_FLAGS
+        poetry install $EXTRAS
 
     - name: Install pytest
       run: |