remove type

gradsflow · Dec 25, 2021 · d9ae8c4 · d9ae8c4
1 parent 0692b74
commit d9ae8c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/autotasks/test_core_automodel.py b/tests/autotasks/test_core_automodel.py
@@ -49,7 +49,7 @@ def test_create_search_space():
 
 
 @patch.multiple(AutoModel, __abstractmethods__=set())
-@patch("gradsflow.autotasks.engine.backend_type.pl")
+@patch("gradsflow.autotasks.engine.backend.pl")
 def test_objective(mock_pl):
     optimization_metric = "val_accuracy"
     model = AutoModel(