Skip to content

Commit f63d86b

Browse files
Fix missing import for load_numpy_array_data in MLflow section
1 parent bbdfd17 commit f63d86b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

train_with_components.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -220,6 +220,9 @@ def __init__(self, train_path, test_path):
220220
mlflow.set_tracking_uri(mlflow_tracking_uri)
221221
mlflow.set_experiment("network-security-classification")
222222

223+
# Import the load_numpy_array_data function in this scope
224+
from networksecurity.utils.main_utils import load_numpy_array_data
225+
223226
# Start a new MLflow run
224227
with mlflow.start_run():
225228
# Log parameters

0 commit comments

Comments
 (0)