From 1efcd4c29ec7bcef1bd751a5cb42bb780e93fdce Mon Sep 17 00:00:00 2001
From: Daniel Williams <daniel.williams@ligo.org>
Date: Mon, 26 Aug 2024 13:59:44 +0100
Subject: [PATCH] Add MSUN_SI

---
 heron/models/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/heron/models/__init__.py b/heron/models/__init__.py
index c311280..171c0f6 100644
--- a/heron/models/__init__.py
+++ b/heron/models/__init__.py
@@ -1,5 +1,6 @@
 import torch
-from lal import antenna
+import numpy as np
+from lal import antenna, MSUN_SI
 from astropy import units as u