Notifications#

MLRun supports configuring notifications on jobs and scheduled jobs. This section describes the SDK for notifications.

The Notification Object#

The notification object’s schema is:

  • kind: str - notification kind (slack, git, etc…)

  • when: list[str] - run states on which to send the notification (completed, error, aborted)

  • name: str - notification name

  • message: str - notification message

  • severity: str - notification severity (info, warning, error, debug)

  • params: dict - notification parameters (See definitions in Notification Kinds)

  • condition: str - jinja template for a condition that determines whether the notification is sent or not (See Notification Conditions)

Local vs Remote#

Notifications can be sent either locally from the SDK, or remotely from the MLRun API. Usually, a local run sends locally, and a remote run sends remotely. However, there are several special cases where the notification is sent locally either way. These cases are:

  • Pipelines: To conserve backwards compatibility, the SDK sends the notifications as it did before adding the run notifications mechanism. This means you need to watch the pipeline in order for its notifications to be sent.

  • Dask: Dask runs are always local (against a remote dask cluster), so the notifications are sent locally as well.

Disclaimer: Local notifications aren’t persisted in mlrun API

Notification Params and Secrets#

The notification parameters might contain sensitive information (slack webhook, git token, etc.). For this reason, when a notification is created its params are masked in a kubernetes secret. The secret is named <run-uid>-<notification-id> (or <schedule-name>-<notification-id>) and is created in the namespace where mlrun is installed. In the notification params the secret reference is stored under the secret key once masked.

Notification Kinds#

Currently, the supported notification kinds and their params are as follows:

  • slack:

    • webhook: The slack webhook to which to send the notification.

  • git:

    • token: The git token to use for the git notification.

    • repo: The git repo to which to send the notification.

    • issue: The git issue to which to send the notification.

    • merge_request: In gitlab (as opposed to github), merge requests and issues are separate entities. If using merge request, the issue will be ignored, and vice versa.

    • server: The git server to which to send the notification.

    • gitlab: (bool) Whether the git server is gitlab or not.

  • console (no params, local only)

  • ipython (no params, local only)

Configuring Notifications For Runs#

In any run method you can configure the notifications via their model. For example:

notification = mlrun.model.Notification(
    kind="slack",
    when=["completed","error"],
    name="notification-1",
    message="completed",
    severity="info",
    params={"webhook": "<slack webhook url>"}
)
function.run(handler=handler, notifications=[notification])

Configuring Notifications For Pipelines#

For pipelines, you configure the notifications on the project notifiers. For example:

project.notifiers.add_notification(notification_type="slack",params={"webhook":"<slack webhook url>"})
project.notifiers.add_notification(notification_type="git", params={"repo": "<repo>", "issue": "<issue>", "token": "<token>"})

Instead of passing the webhook in the notification params, it is also possible in a Jupyter notebook to use the %env magic command:

%env SLACK_WEBHOOK=<slack webhook url>

Editing and removing notifications is done similarly with the following methods:

project.notifiers.edit_notification(notification_type="slack",params={"webhook":"<new slack webhook url>"})
project.notifiers.remove_notification(notification_type="slack")

Setting Notifications on Live Runs#

You can set notifications on live runs via the set_run_notifications method. For example:

import mlrun

mlrun.get_run_db().set_run_notifications("<project-name>", "<run-uid>", [notification1, notification2])

Using the set_run_notifications method overrides any existing notifications on the run. To delete all notifications, pass an empty list.

Setting Notifications on Scheduled Runs#

You can set notifications on scheduled runs via the set_schedule_notifications method. For example:

import mlrun

mlrun.get_run_db().set_schedule_notifications("<project-name>", "<schedule-name>", [notification1, notification2])

Using the set_schedule_notifications method overrides any existing notifications on the schedule. To delete all notifications, pass an empty list.

Notification Conditions#

You can configure the notification to be sent only if the run meets certain conditions. This is done using the condition parameter in the notification object. The condition is a string that is evaluated using a jinja templator with the run object in its context. The jinja template should return a boolean value that determines whether the notification is sent or not. If any other value is returned or if the template is malformed, the condition is ignored and the notification is sent as normal.

Take the case of a run that calculates and outputs model drift. This example code sets a notification to fire only if the drift is above a certain threshold:

notification = mlrun.model.Notification(
    kind="slack",
    when=["completed","error"],
    name="notification-1",
    message="completed",
    severity="info",
    params={"webhook": "<slack webhook url>"},
    condition='{{ run["status"]["results"]["drift"] > 0.1 }}'
)