fix bad values for high (abs) eta

Author: bfis

src/vector/_compute/spatial/eta.py

@@ -28,9 +28,7 @@
 
 
 def xy_z(lib, x, y, z):
-    return lib.nan_to_num(
-        lib.arctanh(z / lib.sqrt(x**2 + y**2 + z**2)), nan=0.0
-    ) * lib.absolute(lib.sign(z))
+    return lib.where(z, lib.arcsinh(z / lib.sqrt(x**2 + y**2)), z)
 
 
 def xy_theta(lib, x, y, theta):
@@ -42,9 +40,7 @@ def xy_eta(lib, x, y, eta):
 
 
 def rhophi_z(lib, rho, phi, z):
-    return lib.nan_to_num(
-        lib.arctanh(z / lib.sqrt(rho**2 + z**2)), nan=0.0
-    ) * lib.absolute(lib.sign(z))
+    return lib.where(z, lib.arcsinh(z / rho), z)
 
 
 def rhophi_theta(lib, rho, phi, theta):