Skip to content

Commit 68e935b

Browse files
committed
avoid where function
1 parent 3f66775 commit 68e935b

File tree

1 file changed

+2
-2
lines changed
  • src/vector/_compute/spatial

1 file changed

+2
-2
lines changed

src/vector/_compute/spatial/eta.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828

2929

3030
def xy_z(lib, x, y, z):
31-
return lib.where(z, lib.arcsinh(z / lib.sqrt(x**2 + y**2)), z)
31+
return lib.arcsinh(z / lib.sqrt(x**2 + y**2))
3232

3333

3434
def xy_theta(lib, x, y, theta):
@@ -40,7 +40,7 @@ def xy_eta(lib, x, y, eta):
4040

4141

4242
def rhophi_z(lib, rho, phi, z):
43-
return lib.where(z, lib.arcsinh(z / rho), z)
43+
return lib.arcsinh(z / rho)
4444

4545

4646
def rhophi_theta(lib, rho, phi, theta):

0 commit comments

Comments
 (0)