diff --git a/src/modelplane/runways/annotator.py b/src/modelplane/runways/annotator.py index f03eb8f..22ff5b6 100644 --- a/src/modelplane/runways/annotator.py +++ b/src/modelplane/runways/annotator.py @@ -7,6 +7,7 @@ from typing import Any, Dict, List import mlflow +import modelgauge.annotators.cheval.registration # noqa: F401 import numpy as np from matplotlib import pyplot as plt from modelgauge.annotator import Annotator