diff --git a/docs/source/tutorials/crafting_standard_datasets.md b/docs/source/tutorials/crafting_standard_datasets.md index 47358c23..38cfcd8e 100644 --- a/docs/source/tutorials/crafting_standard_datasets.md +++ b/docs/source/tutorials/crafting_standard_datasets.md @@ -12,7 +12,7 @@ However, the uniform length is set to a specific default value. A good choice fo ```python from functools import partial -from qusi.data import default_light_curve_post_injection_transform +from qusi.transform import default_light_curve_post_injection_transform ``` Then, were we specify the construction of our dataset, we'll add an additional input parameter. So taking what we had in the previous tutorial, we can now change the dataset creation statement to: diff --git a/src/qusi/data.py b/src/qusi/data.py index d56669d9..bf0dd4ec 100644 --- a/src/qusi/data.py +++ b/src/qusi/data.py @@ -1,12 +1,11 @@ """ -Data related public interface. +Data structure related public interface. """ from qusi.internal.finite_standard_light_curve_dataset import FiniteStandardLightCurveDataset from qusi.internal.finite_standard_light_curve_observation_dataset import FiniteStandardLightCurveObservationDataset from qusi.internal.light_curve import LightCurve -from qusi.internal.light_curve_dataset import LightCurveDataset, default_light_curve_post_injection_transform, \ - default_light_curve_observation_post_injection_transform from qusi.internal.light_curve_collection import LightCurveObservationCollection, LightCurveCollection +from qusi.internal.light_curve_dataset import LightCurveDataset __all__ = [ 'FiniteStandardLightCurveDataset', @@ -15,6 +14,4 @@ 'LightCurveCollection', 'LightCurveDataset', 'LightCurveObservationCollection', - 'default_light_curve_post_injection_transform', - 'default_light_curve_observation_post_injection_transform', ] diff --git a/src/qusi/transform.py b/src/qusi/transform.py new file mode 100644 index 00000000..ff89742b --- /dev/null +++ b/src/qusi/transform.py @@ -0,0 +1,19 @@ +""" +Data transform related public interface. +""" +from qusi.internal.light_curve_dataset import default_light_curve_post_injection_transform, \ + default_light_curve_observation_post_injection_transform, make_fluxes_and_label_array_uniform_length +from qusi.internal.light_curve_observation import remove_nan_flux_data_points_from_light_curve_observation, \ + randomly_roll_light_curve_observation +from qusi.internal.light_curve_transforms import from_light_curve_observation_to_fluxes_array_and_label_array, \ + pair_array_to_tensor + +__all__ = [ + 'default_light_curve_post_injection_transform', + 'default_light_curve_observation_post_injection_transform', + 'from_light_curve_observation_to_fluxes_array_and_label_array', + 'make_fluxes_and_label_array_uniform_length', + 'pair_array_to_tensor', + 'randomly_roll_light_curve_observation', + 'remove_nan_flux_data_points_from_light_curve_observation', +]