From 2fa75060754681502c1f9fec1d36c4e19966d09c Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Mon, 23 Mar 2026 08:35:28 -0300 Subject: [PATCH] SIENTIAPDE-1712 SIENTIAPDE-1712 Add timestamp column to processed data in MLFlow and improve dataframe validation in MinioDataFramePayload - Added a 'timestamp' column to the processed data in the MLFlow class for better tracking of data entries. - Updated the validation check in MinioDataFramePayload to handle None values for the dataframe more explicitly. --- laborious/activities/mlflow.py | 2 ++ laborious/utils/models/minio_dataframe_payload.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index e6c75e1..e6522f0 100644 --- a/laborious/activities/mlflow.py +++ b/laborious/activities/mlflow.py @@ -149,6 +149,8 @@ class MLFlow(MinioManager): data.columns.name = None data.index.name = None + data['timestamp'] = data.index + self.debug(f'Processed input data: \n {data.to_csv()}', metadata) # Request transformation from MLFlow model diff --git a/laborious/utils/models/minio_dataframe_payload.py b/laborious/utils/models/minio_dataframe_payload.py index 9d2fe48..c49218d 100644 --- a/laborious/utils/models/minio_dataframe_payload.py +++ b/laborious/utils/models/minio_dataframe_payload.py @@ -194,7 +194,7 @@ class MinioDataFramePayload: MinioDataFramePayload: Instance with data and/or MinIO fields set. """ - if not dataframe or dataframe.empty: + if dataframe is None or dataframe.empty: return cls( data=None, last_timestamp=now().strftime(DATETIME_FORMAT_WITH_TZ), status=status )