"""Tests for MLflow Assistant."""