use numpy instead of math

UXARRAY · Nov 18, 2024 · 074a0ea · 074a0ea
1 parent e3dce0a
commit 074a0ea
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/uxarray/grid/coordinates.py b/uxarray/grid/coordinates.py
@@ -1,6 +1,5 @@
 import xarray as xr
 import numpy as np
-import math
 import warnings
 
 from uxarray.conventions import ugrid
@@ -53,8 +52,8 @@ def _xyz_to_lonlat_rad_no_norm(
         Latitude in radians
     """
 
-    lon = math.atan2(y, x)
-    lat = math.asin(z)
+    lon = np.atan2(y, x)
+    lat = np.asin(z)
 
     # set longitude range to [0, pi]
     lon = np.mod(lon, 2 * np.pi)
@@ -102,8 +101,8 @@ def _xyz_to_lonlat_rad_scalar(
         y /= denom
         z /= denom
 
-    lon = math.atan2(y, x)
-    lat = math.asin(z)
+    lon = np.atan2(y, x)
+    lat = np.asin(z)
 
     # set longitude range to [0, pi]
     lon = np.mod(lon, 2 * np.pi)