Run without optinoal deps

jinlow · Aug 26, 2023 · ea2400b · ea2400b
1 parent 145f705
commit ea2400b
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 4 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -19,10 +19,12 @@ jobs:
           python-version: ${{ matrix.pyversion }}
           architecture: x64
       - name: Install deps
-        run: pip install numpy pandas seaborn xgboost=='1.6.1' scikit-learn
+        run: pip install numpy pandas seaborn xgboost=='1.6.1' scikit-learn toml
       - run: |
           cp README.md py-forust/README.md
           cp LICENSE py-forust/LICENSE
+      - name: Update TOML
+        run: python scripts/remove-optional-deps.py
       - name: Build test data
         run: python scripts/make_resources.py
       - name: Run Cargo tests
@@ -72,10 +74,12 @@ jobs:
           python-version: ${{ matrix.pyversion }}
           architecture: x64
       - name: Install deps
-        run: pip install numpy pandas seaborn xgboost=='1.6.1' scikit-learn
+        run: pip install numpy pandas seaborn xgboost=='1.6.1' scikit-learn toml
       - run: |
           cp README.md py-forust/README.md
           cp LICENSE py-forust/LICENSE
+      - name: Update TOML
+        run: python scripts/remove-optional-deps.py
       - name: Build test data
         run: python scripts/make_resources.py
       - name: Run Cargo tests
@@ -127,10 +131,12 @@ jobs:
           python-version: ${{ matrix.pyversion }}
           architecture: x64
       - name: Install deps
-        run: pip install numpy pandas seaborn xgboost=='1.6.1' scikit-learn
+        run: pip install numpy pandas seaborn xgboost=='1.6.1' scikit-learn toml
       - run: |
           cp README.md py-forust/README.md
           cp LICENSE py-forust/LICENSE
+      - name: Update TOML
+        run: python scripts/remove-optional-deps.py
       - name: Build test data
         run: python scripts/make_resources.py
       - name: Run Cargo tests
@@ -180,10 +186,12 @@ jobs:
           python-version: "3.10"
           architecture: x64
       - name: Install deps
-        run: pip install numpy pandas seaborn xgboost=='1.6.1' scikit-learn
+        run: pip install numpy pandas seaborn xgboost=='1.6.1' scikit-learn toml
       - run: |
           cp README.md py-forust/README.md
           cp LICENSE py-forust/LICENSE
+      - name: Update TOML
+        run: python scripts/remove-optional-deps.py
       - name: Build test data
         run: python scripts/make_resources.py
       - name: Run Cargo tests

diff --git a/scripts/remove-optional-deps.py b/scripts/remove-optional-deps.py
@@ -0,0 +1,13 @@
+# Building with polars is sooo slow.
+# It's only there for the example, so let's remove it
+# in the regular build process.
+# Requires toml package
+import toml
+
+ct = toml.load("cargo.toml")
+
+del ct["dev-dependencies"]
+del ct["bench"]
+
+with open("cargo.toml", "w") as file:
+    toml.dump(ct, file)