Skip to content

Commit 471c8ed

Browse files
committed
fix: include lr_features in exo_features filtering for ForwardPassStrategy
1 parent 5a7d88a commit 471c8ed

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

sup3r/pipeline/strategy.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -392,7 +392,11 @@ def _init_features(self, model):
392392
"""Initialize feature attributes."""
393393
self.exo_handler_kwargs = self.exo_handler_kwargs or {}
394394
exo_features = list(self.exo_handler_kwargs)
395-
exo_features = [f for f in exo_features if f in model.hr_exo_features]
395+
exo_features = [
396+
f
397+
for f in exo_features
398+
if f in model.hr_exo_features or f in model.lr_features
399+
]
396400
features = [f for f in model.lr_features if f not in exo_features]
397401
return features, exo_features
398402

0 commit comments

Comments
 (0)