theta range

predsci · Nov 3, 2022 · 20d0e0b · 20d0e0b
1 parent f291894
commit 20d0e0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/psipy/tracing/tracing.py b/psipy/tracing/tracing.py
@@ -55,7 +55,7 @@ def _vector_grid_from_bs(self, bs: xr.DataArray):
         from streamtracer import VectorGrid
 
         # Account for tracing in spherical coordinates
-        bs.loc[..., "bp"] /= np.cos(bs.coords["theta"])
+        bs.loc[..., "bp"] /= np.abs(np.cos(bs.coords["theta"]))
         bs.loc[..., "bp"] /= bs.coords["r"]
         bs.loc[..., "bt"] /= bs.coords["r"]