/ tests / projects / backend / test_loader.py
test_loader.py
 1  from mlflow.projects.backend import loader
 2  
 3  
 4  def test_plugin_backend():
 5      backend = loader.load_backend("dummy-backend")
 6      assert backend is not None
 7  
 8  
 9  def test_plugin_does_not_exist():
10      backend = loader.load_backend("my_plugin")
11      assert backend is None