Demos/201118

.github/workflows/modelcard.yml

@@ -1,8 +1,12 @@
 name: modelcard-test
 
 on:
+  push:
+    branches:
+      - demos/*
   pull_request:
-    branches: [main]
+    branches:
+      - main
   workflow_dispatch:
 
 jobs:
@@ -13,8 +17,9 @@ jobs:
         uses: actions/checkout@v3
 
       - name: Check for model card file
-        uses: CompliancePal/modelcard-action@mlflow-action-experiments
+        uses: CompliancePal/modelcard-action@main
         env:
           MLFLOW_TRACKING_URI: ${{ secrets.MLFLOW_TRACKING_URI }}
         with:
           modelcard: modelcard.yaml
+          rules: .rules

.rules/rules.yaml

@@ -0,0 +1,8 @@
+rules:
+  model-documentation:
+    message: '{{error}}'
+    given: '$.model_details'
+    severity: error
+    then:
+      - function: truthy
+        field: documentation

Dockerfile

@@ -3,7 +3,7 @@ FROM continuumio/anaconda3
 RUN apt-get update
 RUN apt-get install -y git
 # Install python packages
-RUN pip install mlflow[extras]
+RUN pip install mlflow[extras]==1.29.0
 RUN pip install pymysql
 
 WORKDIR /mlflow

conda.yaml

@@ -7,4 +7,4 @@ dependencies:
   - scikit-learn>0.19.1
   - pip
   - pip:
-      - mlflow
+      - mlflow==1.29.0

modelcard.yaml

@@ -1,8 +1,7 @@
 model_details:
   name: Wine prediction
+  documentation: |
+    Now we have documentation! :tada:
   version:
     name: 1.0.0
     date: 2022-11-03
-  run:
-    id: 2a84d204c6794965bc36641b46b77255
-    type: mlflow