diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index 93a435e..0672242 100644 --- a/laborious/activities/mlflow.py +++ b/laborious/activities/mlflow.py @@ -53,7 +53,7 @@ class MLFlow(SientiaMonitoring): _MAX_DEBUG_DATAFRAME_ROWS = 100 _DEFAULT_MODEL_ALIAS = 'production' - _REFERENCE_ARTIFACT_CANDIDATES = ('retrain_input.csv', 'test_data.csv') + _REFERENCE_ARTIFACT_CANDIDATES = ('evaluation_data.csv', 'test_data.csv') def __init__( self,