gitea_admin/sientia-dataops-laborious_temporal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1273

Browse Source 
Update sientia-mlops-library dependency to version 0.40.6 and refactor get_prediction_data method calls for improved readability in model_repository.py and test_model_repository.py.
This commit is contained in:
vitor-aignosi
2025-11-19 14:54:50 -03:00
parent 684bc9fb31
commit 3311409885
3 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
laborious/utils/repository/model_repository.py
View File

@@ -705,7 +705,11 @@ class MLFlowRepository(SientiaMonitoring):
"""
def get_prediction_data(
self, prediction_model: Any, retrain_dataset: pd.DataFrame, target_name: str, predict_flavor: str
self,
prediction_model: Any,
retrain_dataset: pd.DataFrame,
target_name: str,
predict_flavor: str,
) -> pd.DataFrame:
"""
Get prediction data from prediction model.
@@ -861,7 +865,9 @@ class MLFlowRepository(SientiaMonitoring):
prediction_model.fit(retrain_dataset)
# get prediction data
prediction_data = self.get_prediction_data(prediction_model, retrain_dataset, target_name, predict_flavor)
prediction_data = self.get_prediction_data(
prediction_model, retrain_dataset, target_name, predict_flavor
)
self.info(f'Model experiment creation completed successfully for {model_name}', metadata)