From d9adf61a22bff42fe8a0e50878db82223ded0032 Mon Sep 17 00:00:00 2001
From: mpc <matthewcoole@gmail.com>
Date: Fri, 25 Oct 2024 09:57:54 +0100
Subject: [PATCH] Removed cml container and added cml setup step

---
 .github/workflows/cml.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml
index e1496ba..a60d00d 100644
--- a/.github/workflows/cml.yaml
+++ b/.github/workflows/cml.yaml
@@ -3,12 +3,12 @@ on: [push]
 jobs:
   train-and-report:
     runs-on: ubuntu-latest
-    container: docker://ghcr.io/iterative/cml:0-dvc2-base1
     steps:
       - uses: actions/checkout@v3
       - uses: actions/setup-python@v5
         with:
           python-version: '3.12'
+      - uses: iterative/setup-cml@v2
       - name: Run pipeline
         env:
           REPO_TOKEN: ${{ secrets.GITHUB_TOKEN }}