gto 0.1.11

Version and deploy your models following GitOps principles

GTO

Git Tag Ops. Turn your Git repository into an Artifact Registry:

• Registry: Track new artifacts and their versions for releases and significant changes.
• Lifecycle Management: Promote or roll back versions among a structured set of stages.
• GitOps: Signal CI/CD automation or other downstream systems to act upon these lifecycle updates.
• Enrichments: Annotate and query artifact metadata with additional information.

GTO works by creating annotated Git tags in a standard format.

Installation

GTO requires Python 3. It works on any OS.

$ python -m pip install gto

This package will install the gto command-line interface (CLI) and make the Python API available for use in code.

There's no need to set up any services or databases :)

Getting started

Note: We will cover CLI usage, but every command has a corresponding Python API counterpart in the gto.api module.

In this document we'll use this example repo: https://github.com/iterative/example-gto. Let's clone it first:

$ git clone https://github.com/iterative/example-gto.git
$ cd example-gto

Versioning

To register a new artifact or a new version, use gto register. This is usually done to mark significant changes to the artifact (such as a release or a deprecation).

$ gto register awesome-model
Created git tag 'awesome-model@v0.0.1' that registers a new version

GTO creates a special Git tag for the artifact version, in a standard format: {artifact_name}@{version_number}.

The version is now associated to the current Git commit (HEAD). You can have several versions in a given commit, ordered by their automatic version numbers.

Promoting

Promote a specific artifact version to a lifecycle stage with gto promote. Stages can be seen as the status of your artifact, signaling readiness for usage by downstream systems, e.g. via CI/CD or web hooks. For example: redeploy an ML model.

$ gto promote awesome-model prod
Created git tag 'awesome-model#prod#1' that promotes 'v0.0.1'

GTO creates a special Git tag for the artifact promotion, in a standard format: {artifact_name}#{stage}#{e}.

The event is now associated to the latest version of the artifact. There can be multiple events for a given version, ordered by an automatic incremental event number ({e}). This will keep the history of your promotions.

Note: if you prefer, you can use simple promotion tag format without the incremental {e}, but this will disable the gto history command. This is because promoting an artifact where a promotion tag already existed will require deleting the existing tag.

Annotating

So far we've seen how to register and promote artifact versions, but we still don't have much information about them. What about the type of artifact (dataset, model, etc.) or the file path to find it in the working tree?

For simple projects (e.g. single artifact) we can assume the details in a downstream system. But for more advanced cases, we should codify them in the registry itself.

To annotate an artifact, use gto annotate. GTO writes to an artifacts.yaml file to save this metadata. Feel free to modify the file directly!

$ gto annotate awesome-model --type model --path s3://awesome/model.pkl

# artifacts.yaml
awesome-model:
    type: model
    path: "s3://awesome/model.pkl"

Don't forget to commit artifacts.yaml with Git to associate it with the latest artifact version and stage in any copy of the repo.

By default GTO saves artifact as virtual. Use the --must_exist flag to tell GTO the artifact file is committed to Git.

• Physical files/directories are committed to the repo. When you register a new version or promote it, Git guarantees that it's immutable -- you can return a year later and get the same artifact by providing a version.

• Virtual artifacts could be an external path (e.g. s3://mybucket/myfile) or a local path to a metafile representing an externally stored artifact file (as with DVC). In this case, GTO can't pin versions to a physical state of the artifact and guarantee it's immutability later, e.g. if s3://mybucket/myfile changes the registry won't know it, nor have a way to recover the original file.

In future versions, we will support additional enrichments: useful information that other tools like DVC and MLEM can provide about the artifacts. This will allow treating DVC repo outputs as usual artifacts instead of virtual ones.

Using the registry locally

Let's look at the usage of the gto show and gto history.

Show the current state

This is the entire state of the registry: all artifacts, their latest versions, and what is promoted to stages right now.

$ gto show
╒═══════════════╤══════════╤════════╤═════════╤════════════╕
│ name          │ latest   │ #dev   │ #prod   │ #staging   │
╞═══════════════╪══════════╪════════╪═════════╪════════════╡
│ churn         │ v3.1.0   │ -      │ v3.0.0  │ v3.1.0     │
│ segment       │ v0.4.1   │ v0.4.1 │ -       │ -          │
│ cv-class      │ v0.1.13  │ -      │ -       │ -          │
│ awesome-model │ v0.0.1   │ -      │ v0.0.1  │ -          │
╘═══════════════╧══════════╧════════╧═════════╧════════════╛

Here we'll see both artifacts that have Git tags only and those annotated in artifacts.yaml. Use --all-branches or --all-commits to read artifacts.yaml from more commits than just HEAD.

Add an artifact name to print all og its versions instead:

$ gto show churn
╒════════════╤═══════════╤═════════╤═════════════════════╤═══════════════════╤══════════════╕
│ artifact   │ version   │ stage   │ created_at          │ author            │ ref          │
╞════════════╪═══════════╪═════════╪═════════════════════╪═══════════════════╪══════════════╡
│ churn      │ v3.0.0    │ prod    │ 2022-04-08 23:46:58 │ Alexander Guschin │ churn@v3.0.0 │
│ churn      │ v3.1.0    │ staging │ 2022-04-13 14:53:38 │ Alexander Guschin │ churn@v3.1.0 │
╘════════════╧═══════════╧═════════╧═════════════════════╧═══════════════════╧══════════════╛

See the history of an artifact

gto history will print a journal of the events that happened to an artifact. This allows you to audit the changes.

$ gto history churn
╒═════════════════════╤════════════╤══════════════╤═══════════╤═════════╤══════════╤═══════════════════╕
│ timestamp           │ artifact   │ event        │ version   │ stage   │ commit   │ author            │
╞═════════════════════╪════════════╪══════════════╪═══════════╪═════════╪══════════╪═══════════════════╡
│ 2022-04-07 20:00:18 │ churn      │ commit       │ -         │ -       │ 54d6d39  │ Alexander Guschin │
│ 2022-04-08 23:46:58 │ churn      │ registration │ v3.0.0    │ -       │ 54d6d39  │ Alexander Guschin │
│ 2022-04-12 11:06:58 │ churn      │ commit       │ -         │ -       │ 26cafe9  │ Alexander Guschin │
│ 2022-04-13 14:53:38 │ churn      │ registration │ v3.1.0    │ -       │ 26cafe9  │ Alexander Guschin │
│ 2022-04-14 18:40:18 │ churn      │ promotion    │ v3.1.0    │ staging │ 26cafe9  │ Alexander Guschin │
│ 2022-04-15 22:26:58 │ churn      │ promotion    │ v3.0.0    │ prod    │ 54d6d39  │ Alexander Guschin │
╘═════════════════════╧════════════╧══════════════╧═══════════╧═════════╧══════════╧═══════════════════╛

Consuming the registry downstream

Let's look at integrating with GTO via Git as well as using the gto check-ref, gto latest, gto which, and gto describe utility commands downstream.

Act on new versions and promotions in CI

To act upon annotations (Git tags), you can create simple CI workflow. With GitHub Actions for example, it can look like this:

name: Act on versions or promotions of the "churn" actifact
on:
  push:
    tags:
      - "churn*"

When CI is triggered, you can use the Git reference to determine the version of the artifact. In GH Actions, you can use the GITHUB_REF environment variable (check out our example workflow). You can parse tags manually or use gto check-ref:

$ gto check-ref awesome-model@v0.0.1
{
    "version": {
        "awesome-model": {
            "artifact": "awesome-model",
            "name": "v0.0.1",
            "created_at": "2022-04-21T17:39:14",
            "author": "Alexander Guschin",
            "commit_hexsha": "26cafe958dca65d726b3c9023fbae71ed259b566",
            "discovered": false,
            "tag": "awesome-model@v0.0.1",
        }
    },
    "stage": {}
}

Getting the right version

To get the latest artifact version, its path, and Git reference, use gto latest:

$ gto latest churn
v3.1.0

$ gto latest churn --ref
churn@v3.1.0

To get the version that is currently promoted to an environment (stage), use gto which:

$ gto which churn prod
v3.0.0

$ gto which churn prod --ref
churn#prod#2

To get details about an artifact (from artifacts.yaml) use gto describe:

$ gto describe churn

{
    "type": "model",
    "path": "models/churn.pkl",
    "virtual": false
}

The output is in JSON format for ease of parsing programatically.

Configuration

To configure GTO, use file .gto in the root of your repo or use environment variables (note the GTO_ prefix):

# .gto config file
types: [model, dataset]  # list of allowed types
stages: [dev, stage, prod]  # list of allowed Stages

$ GTO_EMOJIS=false gto show

Setup GTO development environment

1. Clone this repository

$ git clone [e-mail]:iterative/gto.git
$ cd gto

2. Create virtual environment named venv

$ python3 -m venv venv
$ source venv/bin/activate

Install python libraries

$ pip install --upgrade pip setuptools wheel ".[tests]"

3. Run

$ pytest --basetemp=pytest-cache

This will create pytest-cache/ directory with some fixtures that can serve as examples.

Notably, check out this dir:

$ cd pytest-cache/test_api0/
$ gto show -v

The code that generates this folder could be found in this fixture.

To continue experimenting, call gto --help