Update astropy import (#116)

robjmcgibbon · web-flow · commit 3904a4cd1cc1 · 2025-06-16T13:05:07.000+01:00
diff --git a/velociraptor/observations/objects.py b/velociraptor/observations/objects.py
@@ -13,7 +13,7 @@
 from matplotlib import rcParams
 
 from astropy.units import Quantity
-from astropy.cosmology.core import Cosmology
+from astropy.cosmology import Cosmology
 from astropy.cosmology import wCDM, FlatLambdaCDM
 
 import h5py
diff --git a/velociraptor/units.py b/velociraptor/units.py
@@ -7,7 +7,7 @@
 import unyt
 import h5py
 
-from astropy.cosmology.core import Cosmology
+from astropy.cosmology import Cosmology
 from astropy.cosmology import wCDM, FlatLambdaCDM
 
 
