From 11ec15512ac628e82605004e721de33bbe0d4804 Mon Sep 17 00:00:00 2001
From: Philipp Millet <60098436+pppmlt@users.noreply.github.com>
Date: Fri, 4 Nov 2022 16:34:12 +0100
Subject: [PATCH] Update cml.yaml

---
 .github/workflows/cml.yaml | 17 +++++------------
 1 file changed, 5 insertions(+), 12 deletions(-)

diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml
index 84f8b3557..36b35bcc5 100644
--- a/.github/workflows/cml.yaml
+++ b/.github/workflows/cml.yaml
@@ -3,19 +3,12 @@ on: push
 jobs:
   run:
     runs-on: ubuntu-latest
-    # optionally use a convenient Ubuntu LTS + DVC + CML image
-    # container: ghcr.io/iterative/cml:0-dvc2-base1
+    container: docker://dvcorg/cml-py3:latest
     steps:
-      - uses: actions/checkout@v3
-      # may need to setup NodeJS & Python3 on e.g. self-hosted
-      # - uses: actions/setup-node@v3
-      #   with:
-      #     node-version: '16'
-      # - uses: actions/setup-python@v4
-      #   with:
-      #     python-version: '3.x'
-      - uses: iterative/setup-cml@v1
-      - name: Train model
+      - uses: actions/checkout@v2
+      - name: cml_run
+        env:
+          repo_token: ${{ secrets.GITHUB_TOKEN }}
         run: |
           # Your ML workflow goes here
           pip install -r requirements.txt