gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-15 03:48:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(Types): added nested types for Reporting (#162)

Browse Source
This commit is contained in:
Mark Aron Szulyovszky
2022-01-13 09:21:07 +01:00
committed by GitHub
parent 1856fcad22
commit 3c2a0d4247
8 changed files with 73 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/inference.py
+4 -4
View File
@@ -4,9 +4,9 @@ from data_loader.load_data import load_data
from typing import Optional, Union
import warnings
from utils.encapsulation import Asset, Single_Model, Training_Step
from reporting.types import Reporting
def run_inference_pipeline(data_config:dict, training_config:dict, all_models_all_assets:list[Asset]):
def run_inference_pipeline(data_config:dict, training_config:dict, all_models_all_assets:list[Reporting.Asset]):
data_params = data_config.copy()
data_params['target_asset'] = data_params['assets'][0]
@@ -20,7 +20,7 @@ def run_inference_pipeline(data_config:dict, training_config:dict, all_models_al
return result
def __inference(data:pd.DataFrame, primary_step:Union[Training_Step,None], secondary_step:Union[Training_Step,None]) -> pd.DataFrame:
def __inference(data:pd.DataFrame, primary_step:Union[Reporting.Training_Step,None], secondary_step:Union[Reporting.Training_Step,None]) -> pd.DataFrame:
assert primary_step is not None, "No primary models found. Cancelling Inference."
data = __primary_models(data, primary_step)
@@ -32,7 +32,7 @@ def __inference(data:pd.DataFrame, primary_step:Union[Training_Step,None], secon
return data
def __select_models( data_params:dict, all_models_all_assets:list[Asset])-> tuple[Union[Training_Step,None], Union[Training_Step, None]]:
def __select_models( data_params:dict, all_models_all_assets:list[Reporting.Asset])-> tuple[Union[Reporting.Training_Step,None], Union[Reporting.Training_Step, None]]:
target_asset_name = data_params['target_asset'][1]
primary_step, secondary_step, = None, None
target_asset_models = next((x for x in all_models_all_assets if x.name == target_asset_name), None)