Add pip install step to CI

404-Repo · Feb 4, 2025 · 1bd7fab · 1bd7fab
1 parent 6c79ecf
commit 1bd7fab
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/.github/workflows/spz.yml b/.github/workflows/spz.yml
@@ -25,13 +25,20 @@ jobs:
         if: runner.os == 'Linux'
         run: |
           sudo apt-get update
-          sudo apt-get install -y build-essential cmake ninja-build
+          sudo apt-get install -y build-essential cmake ninja-build python3-pip
 
       - name: Set up C++ environment on macOS
         if: runner.os == 'macOS'
         run: |
           brew update
-          brew install cmake ninja
+          brew install cmake ninja python3
+
+      - name: Set up C++ environment on Windows
+        if: runner.os == 'Windows'
+        run: |
+          choco install python --version 3.10
+          python -m pip install --upgrade pip
+        shell: cmd
 
       - name: Configure CMake with MSBuild (Windows)
         if: runner.os == 'Windows'
@@ -57,3 +64,7 @@ jobs:
       - name: Build with Ninja (Linux/macOS)
         if: runner.os != 'Windows'
         run: cmake --build build --parallel
+
+      - name: Install Python spz package
+        run: |
+          pip3 install .
diff --git a/src/cc/load-spz.cc b/src/cc/load-spz.cc
@@ -558,7 +558,7 @@ bool saveSpz(const GaussianCloud &g, std::vector<uint8_t> *out,
     data = ss.str();
   }
   return compressZstd(reinterpret_cast<const uint8_t *>(data.data()),
-                      data.size(), out, compressionLevel);
+                      data.size(), out, compressionLevel, workers);
 }
 
 PackedGaussians loadSpzPacked(const uint8_t *data, int size) {