diff --git a/laborious/utils/models/minio_dataframe_payload.py b/laborious/utils/models/minio_dataframe_payload.py index b480ee1..0521f5e 100644 --- a/laborious/utils/models/minio_dataframe_payload.py +++ b/laborious/utils/models/minio_dataframe_payload.py @@ -98,7 +98,7 @@ class MinioDataFramePayload: """ if logger is None: return - logger.debug(message, metadata) + logger.custom_debug(message, metadata) @classmethod def from_dict(cls, raw: 'dict[str, Any] | MinioDataFramePayload') -> 'MinioDataFramePayload': @@ -152,8 +152,11 @@ class MinioDataFramePayload: except Exception: size = len(pickle.dumps(df)) - if logger is not None: - logger.debug(f'DataFrame size: {size} bytes', metadata) + MinioDataFramePayload._debug( + logger, + f'DataFrame size: {size} bytes', + metadata, + ) return size @staticmethod