Update cml.yaml

.github/workflows/cml.yaml

@@ -3,26 +3,34 @@ on: [push]
 jobs:
   run:
     runs-on: ubuntu-latest
-    container: ghcr.io/iterative/cml:0-dvc2-base1
+    # optionally use a convenient Ubuntu LTS + DVC + CML image
+    # container: ghcr.io/iterative/cml:0-dvc2-base1
     steps:
-      - uses: actions/checkout@v4
+      - 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
         run: |
           # Your ML workflow goes here
-          python -m pip install --upgrade pip
           pip install -r requirements.txt
           python train.py
-          
       - name: Write CML report
         env:
           REPO_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         run: |
           echo "## Model metrics" > report.md
           cat metrics.txt >> report.md
-
-          echo "## Data viz" >> report.md
-          echo "![](./features.png 'Features')" >> report.md
-          echo "![](./residuals.png 'Residuals')" >> report.md
+          echo "Data viz" >> report.md
+          echo "![](./features.png)" >> report.md
+          # echo "![](./residuals.png 'Residuals')" >> report.md
           
+          # cml-publish feature_importance.png --md >> report.md
+          # cml-publish residuals.png --md >> report.md
           # Post reports as comments in GitHub PRs
           cml comment create report.md