SIENTIAPDE-1193

Update Gates activity and format_and_export_prediction workflow to use DATETIME_FORMAT_WITH_TZ for consistent timestamp handling

- Modified Gates activity to correctly handle the maximum timestamp without formatting it prematurely.
- Updated format_and_export_prediction workflow to utilize DATETIME_FORMAT_WITH_TZ for timestamp conversion.
- Enhanced tests to ensure timestamp conversion is applied consistently across workflows.
This commit is contained in:
vitor-aignosi
2025-08-22 14:17:54 -03:00
parent 2301dd63c7
commit ae6d28df54
3 changed files with 14 additions and 5 deletions

View File

@@ -5,7 +5,7 @@ with workflow.unsafe.imports_passed_through():
from typing import Any
from datetime import timedelta
from sientia_do.temporal.policies import retry_policy
from sientia_do.temporal.constants import DATETIME_FORMAT
from sientia_do.temporal.constants import DATETIME_FORMAT_WITH_TZ
@workflow.defn(name="format_and_export_prediction")
@@ -94,7 +94,7 @@ class FormatAndExportPrediction():
'data': prediction,
'timestamp_conversion': {
'column': 'timestamp',
'format': DATETIME_FORMAT
'format': DATETIME_FORMAT_WITH_TZ
}
},
retry_policy=retry_policy,