🐛 Fix units in observe.py

astrojarred · Dec 11, 2023 · 036ae9d · 036ae9d
1 parent 518d57f
commit 036ae9d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/gravitational_wave_toy/observe.py b/gravitational_wave_toy/observe.py
@@ -39,9 +39,9 @@ def __init__(
         self.filepath = Path(filepath).absolute()
         self.min_energy, self.max_energy = min_energy, max_energy
         self.seen = False
-        self.obs_time = -1
-        self.start_time = -1
-        self.end_time = -1
+        self.obs_time = -1 * u.s
+        self.start_time = -1 * u.s
+        self.end_time = -1 * u.s
         self.error_message = ""
         try:
             self.id = int(self.filepath.stem.split("_")[1])
@@ -574,8 +574,8 @@ def observe(
                 self.obs_time = end_time - start_time
                 self.seen = True
             else:
-                self.end_time = -1
-                self.obs_time = -1
+                self.end_time = -1 * u.s
+                self.obs_time = -1 * u.s
                 self.seen = False
 
             # just return dict now