handle key check correctly
This commit is contained in:
parent
c5a16e91fb
commit
0bf915054d
@ -225,7 +225,7 @@ class IFreqaiModel(ABC):
|
|||||||
def check_if_feature_list_matches_strategy(self, dataframe: DataFrame,
|
def check_if_feature_list_matches_strategy(self, dataframe: DataFrame,
|
||||||
dh: FreqaiDataKitchen) -> None:
|
dh: FreqaiDataKitchen) -> None:
|
||||||
strategy_provided_features = dh.find_features(dataframe)
|
strategy_provided_features = dh.find_features(dataframe)
|
||||||
if dh.data['training_features_list_raw']:
|
if 'training_features_list_raw' in dh.data:
|
||||||
feature_list = dh.data['training_features_list_raw']
|
feature_list = dh.data['training_features_list_raw']
|
||||||
else:
|
else:
|
||||||
feature_list = dh.training_features_list
|
feature_list = dh.training_features_list
|
||||||
|
Loading…
Reference in New Issue
Block a user