[do not merge] Demonstrate registering committed models via GitHub Actions

.github/workflows/register_verta_model.yaml

@@ -0,0 +1,41 @@
+name: Register Verta Model Version
+
+on:
+  push:
+    paths:
+      - models/*.pkl
+    branches:
+      - workflow1
+
+jobs:
+  Register-Verta-Model-Version:
+    runs-on: ubuntu-latest
+    defaults:
+      run:
+        working-directory: ./models
+
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0 # fetch full history, in case push is multiple commits
+
+      - name: Set up Python 3.10
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.10"
+          cache: "pip"
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install -r requirements.txt
+
+      - name: Register model version
+        env:
+          VERTA_HOST: dev.verta.ai
+          VERTA_EMAIL: [email protected]
+          VERTA_DEV_KEY: ${{ secrets.TEST_ACCOUNT_DEV_KEY }}
+          REGISTERED_MODEL_NAME: From GitHub Action
+        run: |
+          export MODEL_FILENAMES=$(git diff --diff-filter=AR --relative --name-only ${{ github.event.before }} ${{ github.event.after }} | grep '\.pkl$')
+          python register_model_version.py

models/register_model_version.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+
+import os
+
+import cloudpickle
+from verta import Client
+from verta.environment import Python
+
+
+REGISTERED_MODEL_NAME = os.environ["REGISTERED_MODEL_NAME"]
+MODEL_FILENAMES = os.environ["MODEL_FILENAMES"]
+
+
+if __name__ == "__main__":
+    client = Client()
+    reg_model = client.get_or_create_registered_model(REGISTERED_MODEL_NAME)
+
+    for model_filename in MODEL_FILENAMES.splitlines():
+        with open(model_filename, "rb") as f:
+            model_cls = cloudpickle.load(f)
+
+        requirements = Python.read_pip_file(
+            os.path.join(os.path.dirname(__file__), "requirements.txt"),
+        )
+
+        print(f'Registering model "{model_filename}"')
+        model_ver = reg_model.create_standard_model(
+            model_cls,
+            environment=Python(requirements),
+        )

models/requirements.txt

@@ -0,0 +1 @@
+verta==0.22.2

models/save_model_file.py

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+
+from datetime import datetime
+import os
+
+import cloudpickle
+from verta.registry import VertaModelBase, verify_io
+
+
+class Model(VertaModelBase):
+    def __init__(self, artifacts=None):
+        pass
+
+    @verify_io
+    def predict(self, input):
+        return input
+
+
+if __name__ == "__main__":
+    model_filename = f"{datetime.now().isoformat()}.pkl"
+    model_filepath = os.path.join(os.path.dirname(__file__), model_filename)
+    with open(model_filepath, "wb") as f:
+        cloudpickle.dump(Model, f)