Fix circular import bug

emilyhunt · Jan 31, 2025 · 55d94fe · 55d94fe
1 parent 6df7579
commit 55d94fe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ocelot/model/observation/common.py b/src/ocelot/model/observation/common.py
@@ -4,7 +4,7 @@
 import pandas as pd
 from astropy.coordinates import SkyCoord
 from astropy.units import Quantity
-from ocelot.model.observation import BaseObservation
+import ocelot.model.observation
 import ocelot.simulate.cluster
 
 
@@ -63,7 +63,7 @@ def calculate_separation(primary: pd.DataFrame, secondary: pd.DataFrame) -> Quan
 
 def apply_astrometric_errors_simple_gaussian(
     cluster: ocelot.simulate.cluster.SimulatedCluster,
-    model: BaseObservation,
+    model: ocelot.model.observation.BaseObservation,
     columns: None | list[str] | tuple[str] = None,
 ):
     """Calculates astrometry sampled from a Gaussian error distribution and adds it
@@ -97,7 +97,7 @@ def apply_astrometric_errors_simple_gaussian(
 
 def apply_photometric_errors_simple_gaussian(
     cluster: ocelot.simulate.cluster.SimulatedCluster,
-    model: BaseObservation,
+    model: ocelot.model.observation.BaseObservation,
     bands: None | list[str] | tuple[str] = None,
 ):
     """Calculates photometry sampled from a Gaussian error distribution and adds it