18 lines
458 B
Python
18 lines
458 B
Python
from flask import request
|
|
from prometheus_flask_exporter.multiprocess import GunicornInternalPrometheusMetrics
|
|
|
|
from mlflow.version import VERSION
|
|
|
|
|
|
def activate_prometheus_exporter(app):
|
|
def mlflow_version(_: request):
|
|
return VERSION
|
|
|
|
return GunicornInternalPrometheusMetrics(
|
|
app,
|
|
export_defaults=True,
|
|
defaults_prefix="mlflow",
|
|
excluded_paths=["/health", "/version"],
|
|
group_by=mlflow_version,
|
|
)
|