Make the default TESS SPOC light curve processed length be 3500

golmschenk · May 8, 2024 · aa298a8 · aa298a8
1 parent 4dc9e1f
commit aa298a8
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/src/qusi/internal/finite_standard_light_curve_dataset.py b/src/qusi/internal/finite_standard_light_curve_dataset.py
@@ -37,7 +37,7 @@ def new(
         :return: The dataset.
         """
         if post_injection_transform is None:
-            post_injection_transform = partial(default_light_curve_post_injection_transform, length=2500,
+            post_injection_transform = partial(default_light_curve_post_injection_transform, length=3500,
                                                randomize=False)
         length = 0
         collection_start_indexes: list[int] = []

diff --git a/src/qusi/internal/finite_standard_light_curve_observation_dataset.py b/src/qusi/internal/finite_standard_light_curve_observation_dataset.py
@@ -37,7 +37,7 @@ def new(
         :return: The dataset.
         """
         if post_injection_transform is None:
-            post_injection_transform = partial(default_light_curve_observation_post_injection_transform, length=2500,
+            post_injection_transform = partial(default_light_curve_observation_post_injection_transform, length=3500,
                                                randomize=False)
         length = 0
         collection_start_indexes: list[int] = []

diff --git a/src/qusi/internal/hadryss_model.py b/src/qusi/internal/hadryss_model.py
@@ -127,7 +127,7 @@ def forward(self, x: Tensor) -> Tensor:
         return x
 
     @classmethod
-    def new(cls, input_length: int = 2500) -> Self:
+    def new(cls, input_length: int = 3500) -> Self:
         """
         Creates a new Hadryss model.
 

diff --git a/src/qusi/internal/light_curve_dataset.py b/src/qusi/internal/light_curve_dataset.py
@@ -168,7 +168,7 @@ def new(
             injectable_light_curve_collections = []
         if post_injection_transform is None:
             post_injection_transform = partial(
-                default_light_curve_observation_post_injection_transform, length=2500
+                default_light_curve_observation_post_injection_transform, length=3500
             )
         instance = cls(
             standard_light_curve_collections=standard_light_curve_collections,