Correct time nchunks

AlanLoh · Dec 9, 2024 · 1ff388c · 1ff388c
1 parent 010ca2c
commit 1ff388c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/nenupy/__init__.py b/nenupy/__init__.py
@@ -5,7 +5,7 @@
 __copyright__ = "Copyright 2023, nenupy"
 __credits__ = ["Alan Loh"]
 __license__ = "MIT"
-__version__ = "2.7.12"
+__version__ = "2.7.13"
 __maintainer__ = "Alan Loh"
 __email__ = "alan.loh@obspm.fr"
 

diff --git a/nenupy/io/tf.py b/nenupy/io/tf.py
@@ -349,7 +349,7 @@ def wrapper_task(
             return time_unix, frequency_hz, data
 
         return cls(
-            "Polarizartion corection with DreamBeam",
+            "Polarization corection with DreamBeam",
             wrapper_task,
             [
                 "channels",

diff --git a/nenupy/io/tf_utils.py b/nenupy/io/tf_utils.py
@@ -164,7 +164,8 @@ def apply_dreambeam_corrections(
         (n_time_groups, time_group_size)
     )[:, 0]
     time_start_idx = np.argmax(db_time >= group_start_time[0])
-    time_stop_idx = db_time.size - np.argmax(db_time[::-1] < group_start_time[-1])
+    # time_stop_idx = db_time.size - np.argmax(db_time[::-1] < group_start_time[-1])
+    time_stop_idx = time_start_idx + n_time_groups - 1
 
     jones = db_jones[
         time_start_idx : time_stop_idx + 1, freq_start_idx : freq_stop_idx + 1, :, :