From 7512fdb2a97546b16db6ebe7c3874e98c4216e65 Mon Sep 17 00:00:00 2001 From: golmschenk Date: Mon, 26 Aug 2024 23:16:04 -0400 Subject: [PATCH] Add infinite removal to the light curve only post injection transform --- src/qusi/internal/light_curve_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/qusi/internal/light_curve_dataset.py b/src/qusi/internal/light_curve_dataset.py index 61a6edd..34ee515 100644 --- a/src/qusi/internal/light_curve_dataset.py +++ b/src/qusi/internal/light_curve_dataset.py @@ -25,7 +25,7 @@ from qusi.internal.light_curve import ( LightCurve, randomly_roll_light_curve, - remove_nan_flux_data_points_from_light_curve, + remove_nan_flux_data_points_from_light_curve, remove_infinite_flux_data_points_from_light_curve, ) from qusi.internal.light_curve_observation import ( LightCurveObservation, @@ -369,6 +369,7 @@ def default_light_curve_post_injection_transform( :param randomize: Whether to have randomization in the transforms. :return: The transformed light curve. """ + x = remove_infinite_flux_data_points_from_light_curve(x) x = remove_nan_flux_data_points_from_light_curve(x) if randomize: x = randomly_roll_light_curve(x)