From 369656a3ba72072075dd4d918aa990f5ced40c1d Mon Sep 17 00:00:00 2001
From: Daniel Williams <daniel.williams@ligo.org>
Date: Wed, 15 Jan 2025 17:13:49 +0000
Subject: [PATCH 1/2] Added arguments to calculate start and end of data load.

---
 heron/inference.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/heron/inference.py b/heron/inference.py
index f680c16..1f4fe14 100644
--- a/heron/inference.py
+++ b/heron/inference.py
@@ -83,6 +83,10 @@ def heron_inference(settings):
 
     if "data files" in settings.get("data", {}):
         # Load frame files from disk
+
+        start = settings['event time'] - settings['segment length'] + settings['after merger']
+        end = settings['event time'] + settings['after merger']
+        
         for ifo in settings["interferometers"]:
             print(f"Loading {ifo} data")
             logger.info(
@@ -93,6 +97,8 @@ def heron_inference(settings):
                 source=settings["data"]["data files"][ifo],
                 channel=settings["data"]["channels"][ifo],
                 format="gwf",
+                start=start,
+                end=end,
             )
     #elif "injection" in other_settings:
     #    pass

From f0344ee33a362b5482e88a805df47518d216e8ac Mon Sep 17 00:00:00 2001
From: Daniel Williams <daniel.williams@ligo.org>
Date: Wed, 15 Jan 2025 17:39:51 +0000
Subject: [PATCH 2/2] Added resampling.

---
 heron/inference.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/heron/inference.py b/heron/inference.py
index 1f4fe14..57e6a77 100644
--- a/heron/inference.py
+++ b/heron/inference.py
@@ -100,6 +100,9 @@ def heron_inference(settings):
                 start=start,
                 end=end,
             )
+            if data[ifo].sample_rate != settings['likelihood']['sampling rate']:
+                logger.info("Resampling the data to the likelihood sampling rate")
+                data[ifo] = data[ifo].resample(settings['likelihood']['sampling rate'])
     #elif "injection" in other_settings:
     #    pass