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:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user