fix: wrong import

HERA-Team · Dec 6, 2023 · 481dd28 · 481dd28
1 parent 806e1db
commit 481dd28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hera_sim/tests/test_vis.py b/hera_sim/tests/test_vis.py
@@ -7,6 +7,7 @@
 from astropy import units
 from astropy.coordinates.angles import Latitude, Longitude
 from astropy.units import rad, sday
+from matvis import HAVE_GPU
 from pathlib import Path
 from pyradiosky import SkyModel
 from pyuvsim.analyticbeam import AnalyticBeam
@@ -22,7 +23,6 @@
     VisibilitySimulation,
     load_simulator_from_yaml,
 )
-from vis_cpu import HAVE_GPU
 
 SIMULATORS = (MatVis, UVSim)