Merge pull request #93 from peijin94/traceBline

Minor fix for B-field tracing
predsci · Nov 6, 2022 · 0f2c3f3 · 0f2c3f3
2 parents 8018242 + 20d0e0b
commit 0f2c3f3
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"]