fixed get segments

TheCacophonyProject · Nov 6, 2023 · b3304b1 · b3304b1
1 parent d78eee8
commit b3304b1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/ml_tools/dataset.py b/src/ml_tools/dataset.py
@@ -71,7 +71,7 @@ def __init__(
             else:
                 self.use_segments = config.train.hyper_params.get("use_segments", True)
                 if self.use_segments:
-                    elf.segment_length = config.build.segment_length
+                    self.segment_length = config.build.segment_length
                 else:
                     self.segment_length = 1
             # number of seconds segments are spaced apart

diff --git a/src/ml_tools/thermalwriter.py b/src/ml_tools/thermalwriter.py
@@ -210,7 +210,7 @@ def get_data(clip_samples, extra_args):
                     continue
                 # GP All assumes we dont have a track over multiple bins (Whcih we probably never want)
                 if extra_args.get("use_segments", True):
-                    track.calculate_segments(
+                    track.get_segments(
                         extra_args.get("segment_frame_spacing", 9),
                         extra_args.get("segment_width", 25),
                         extra_args.get("segment_type"),