From 1adacac4ef2bf3e9e41fecf6fb6e2400e4ad6b46 Mon Sep 17 00:00:00 2001
From: Roman Bredehoft <roman.bredehoft@zama.ai>
Date: Tue, 13 Aug 2024 12:09:45 +0200
Subject: [PATCH] chore: fix wheel steps in weekly tests

---
 script/make_utils/pytest_pypi_cml.sh | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/script/make_utils/pytest_pypi_cml.sh b/script/make_utils/pytest_pypi_cml.sh
index 49c192cc4..028ec96b5 100755
--- a/script/make_utils/pytest_pypi_cml.sh
+++ b/script/make_utils/pytest_pypi_cml.sh
@@ -52,6 +52,9 @@ python -m pip install --upgrade pip
 if ${USE_PIP_WHEEL}; then
     # Delete the directory where the pypi wheel file will be created (if it already exists)
     rm -rf dist
+    
+    # Install dev dependencies for testing
+    poetry install --only dev
 
     # Build the wheel file
     poetry build -f wheel
@@ -61,8 +64,6 @@ if ${USE_PIP_WHEEL}; then
     PYPI_WHEEL=$(find dist -type f -name "*.whl")
     python -m pip install --extra-index-url https://pypi.zama.ai/cpu "${PYPI_WHEEL}"
 
-    # Install dev dependencies for testing
-    poetry install --only dev
 else
     if [ -z "${VERSION}" ]; then
         python -m pip install concrete-ml[dev]