SIENTIAPDE-1273

Refactor return types in ModelMetrics class methods to enhance type clarity. Updated calculate_drift and calculate_simple_metrics methods to return lists of dictionaries instead of generic lists, improving type specificity for better data handling.
This commit is contained in:
vitor-aignosi
2025-11-17 10:14:47 -03:00
parent a7ab8ffe49
commit 46fe88f69e

View File

@@ -1,7 +1,7 @@
from temporalio import activity, workflow from temporalio import activity, workflow
with workflow.unsafe.imports_passed_through(): with workflow.unsafe.imports_passed_through():
from typing import Any, Hashable from typing import Any
from pandas import DataFrame, Index, to_datetime from pandas import DataFrame, Index, to_datetime
@@ -138,7 +138,7 @@ class ModelMetrics(SientiaMonitoring):
@activity.defn(name='calculate_drift') @activity.defn(name='calculate_drift')
async def calculate_drift(self, input_data: dict[str, Any]) -> list: async def calculate_drift(self, input_data: dict[str, Any]) -> list[dict]:
""" """
Calculate drift metrics for a model. Calculate drift metrics for a model.
@@ -266,7 +266,7 @@ class ModelMetrics(SientiaMonitoring):
return drift_df.to_dict(orient='records') return drift_df.to_dict(orient='records')
@activity.defn(name='calculate_simple_metrics') @activity.defn(name='calculate_simple_metrics')
async def calculate_simple_metrics(self, input_data: dict[str, Any]) -> list[dict[Hashable, Any]]: async def calculate_simple_metrics(self, input_data: dict[str, Any]) -> list[dict]:
""" """
Calculate simple metrics for a model. Metrics available are: Calculate simple metrics for a model. Metrics available are:
- rmse - rmse