ci: download sample for validation

IVproger · Jul 24, 2024 · 8a9433a · 8a9433a
1 parent 903b330
commit 8a9433a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/.github/workflows/validate-model.yaml b/.github/workflows/validate-model.yaml
@@ -54,5 +54,8 @@ jobs:
       #----------------------------------------------
       #   Run Giskard validation for champion model
       #----------------------------------------------
+      - name: Download sample data
+        if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true'
+        run: poetry run python3 -m src.download_sample
       - name: Validate champion model
         run: poetry run mlflow run . --env-manager=local -e validate
diff --git a/src/download_sample.py b/src/download_sample.py
@@ -0,0 +1,8 @@
+import gdown
+
+gdown.download(
+    "https://drive.google.com/uc?id=1OrOPS0GDsFno3FzukmOqPzETFsDYbiiU",
+    "data/samples/sample.csv",
+    quiet=False,
+    use_cookies=False,
+)