diff --git a/psycop/common/feature_generation/sequences/cohort_definer_to_prediction_times.py b/psycop/common/feature_generation/sequences/cohort_definer_to_prediction_times.py index bdeb08e8c..ee5f11051 100644 --- a/psycop/common/feature_generation/sequences/cohort_definer_to_prediction_times.py +++ b/psycop/common/feature_generation/sequences/cohort_definer_to_prediction_times.py @@ -52,6 +52,8 @@ def create_prediction_times( if pt_outcome_timestamps is not None: outcome_timestamp = pt_outcome_timestamps[0] + else: + outcome_timestamp = None prediction_times += patient.to_prediction_times( lookbehind=lookbehind,