Skip to content

Commit 87953fe

Browse files
committed
Fixes error to raise
1 parent a7438f3 commit 87953fe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hamilton/plugins/mlflow_extensions.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
try:
88
import mlflow
99
except ImportError:
10-
raise ImportError("Please install mlflow to use this plugin")
10+
raise NotImplementedError("Please install mlflow to use this plugin")
1111

1212
"""
1313
While it can be valid to wrap the entire code within the start_run block,

0 commit comments

Comments
 (0)