From e35e27bfb2320049bf1ecbf5eced72b67d814994 Mon Sep 17 00:00:00 2001
From: vitor-aignosi <vitor.santos@aignosi.com.br>
Date: Fri, 20 Mar 2026 16:00:11 -0300
Subject: [PATCH] SIENTIAPDE-1712  SIENTIAPDE-1712 Update type hints in
 PredictionProcess to improve clarity and enforce data structure consistency.
 Changed `data` parameter to a dictionary type and updated the way
 `last_timestamp` is accessed.

---
 laborious/workflows/sub_workflows/prediction_process.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/laborious/workflows/sub_workflows/prediction_process.py b/laborious/workflows/sub_workflows/prediction_process.py
index 3560d5f..2c20ce8 100644
--- a/laborious/workflows/sub_workflows/prediction_process.py
+++ b/laborious/workflows/sub_workflows/prediction_process.py
@@ -110,13 +110,13 @@ class PredictionProcess:
         self,
         input_data: dict[str, Any],
         metadata: dict[str, Any],
-        data: Any,
-        model_id: Any,
+        data: dict[str, Any],
+        model_id: str,
         model_name: str,
         model_config: dict[str, Any],
         save_transform: bool,
     ) -> None:
-        last_timestamp = data.last_timestamp
+        last_timestamp = data['last_timestamp']
 
         # Apply input data quality gates
         gate_input = {