Merge pull request #24 from zenml-io/bugfix/PRD-651-missing-load-metrics

Missing `load_metric` in `datasets`
zenml-io · Sep 23, 2024 · 21c670b · 21c670b
2 parents cf021ce + dba7852
commit 21c670b
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/.github/actions/nlp_template_test/action.yml b/.github/actions/nlp_template_test/action.yml
@@ -82,7 +82,7 @@ runs:
         UV_HTTP_TIMEOUT: '1500'
       run: |
         uv pip install --system -r ./local_checkout/all-requirements.txt
-        uv pip install --system accelerate torchvision
+        uv pip install --system torchvision
 
     - name: Run pytests
       shell: bash

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -40,7 +40,7 @@ jobs:
       matrix:
         stack-name: [local]
         os: [windows-latest, ubuntu-latest, macos-13]
-        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
     env:
       ZENML_DEBUG: true
       ZENML_ANALYTICS_OPT_IN: false
@@ -56,5 +56,5 @@ jobs:
         with:
           stack-name: ${{ matrix.stack-name }}
           python-version: ${{ matrix.python-version }}
-          ref-zenml: ${{ inputs.ref-zenml || 'feature/PRD-566-dependency-cleanup' }}
+          ref-zenml: ${{ inputs.ref-zenml || 'develop' }}
           ref-template: ${{ inputs.ref-template || github.ref }}
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
 copier
 jinja2-time
 pyyaml-include<2.0
+datasets>=2.12.0,<3.0.0
diff --git a/template/config.yaml b/template/config.yaml
@@ -26,7 +26,6 @@ settings:
       - mlflow
       - discord
     requirements:
-      - accelerate
       - zenml[server]
 
 extra:

diff --git a/template/requirements.txt b/template/requirements.txt
@@ -1,4 +1,4 @@
 torchvision
-accelerate
 gradio
 zenml[server]>=0.56.3
+datasets>=2.12.0,<3.0.0
-Original file line number
+Diff line change
@@ Expand Up / @@ -26,7 +26,6 @@ settings: @@
           - mlflow
           - discord
         requirements:
-          - accelerate
           - zenml[server]
     extra:
@@ Expand Down @@