[pre-commit.ci] pre-commit autoupdate (#169)

* [pre-commit.ci] pre-commit autoupdate

updates:
- [github.com/psf/black: 21.12b0 → 22.1.0](psf/black@21.12b0...22.1.0)

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

.pre-commit-config.yaml

@@ -6,7 +6,7 @@ repos:
     args: [--py36-plus]
 
 - repo: https://github.com/psf/black
-  rev: 21.12b0
+  rev: 22.1.0
   hooks:
   - id: black
 

src/vector/_backends/awkward_.py

@@ -865,37 +865,37 @@ class MomentumRecord4D(MomentumAwkward4D, ak.Record):  # type: ignore[misc]
 behavior[numpy.square, "Momentum3D"] = lambda v: v.mag2
 behavior[numpy.square, "Momentum4D"] = lambda v: v.tau2
 
-behavior[numpy.sqrt, "Vector2D"] = lambda v: v.rho2 ** 0.25
-behavior[numpy.sqrt, "Vector3D"] = lambda v: v.mag2 ** 0.25
-behavior[numpy.sqrt, "Vector4D"] = lambda v: v.tau2 ** 0.25
-behavior[numpy.sqrt, "Momentum2D"] = lambda v: v.rho2 ** 0.25
-behavior[numpy.sqrt, "Momentum3D"] = lambda v: v.mag2 ** 0.25
-behavior[numpy.sqrt, "Momentum4D"] = lambda v: v.tau2 ** 0.25
-
-behavior[numpy.cbrt, "Vector2D"] = lambda v: v.rho2 ** 0.16666666666666666
-behavior[numpy.cbrt, "Vector3D"] = lambda v: v.mag2 ** 0.16666666666666666
-behavior[numpy.cbrt, "Vector4D"] = lambda v: v.tau2 ** 0.16666666666666666
-behavior[numpy.cbrt, "Momentum2D"] = lambda v: v.rho2 ** 0.16666666666666666
-behavior[numpy.cbrt, "Momentum3D"] = lambda v: v.mag2 ** 0.16666666666666666
-behavior[numpy.cbrt, "Momentum4D"] = lambda v: v.tau2 ** 0.16666666666666666
+behavior[numpy.sqrt, "Vector2D"] = lambda v: v.rho2**0.25
+behavior[numpy.sqrt, "Vector3D"] = lambda v: v.mag2**0.25
+behavior[numpy.sqrt, "Vector4D"] = lambda v: v.tau2**0.25
+behavior[numpy.sqrt, "Momentum2D"] = lambda v: v.rho2**0.25
+behavior[numpy.sqrt, "Momentum3D"] = lambda v: v.mag2**0.25
+behavior[numpy.sqrt, "Momentum4D"] = lambda v: v.tau2**0.25
+
+behavior[numpy.cbrt, "Vector2D"] = lambda v: v.rho2**0.16666666666666666
+behavior[numpy.cbrt, "Vector3D"] = lambda v: v.mag2**0.16666666666666666
+behavior[numpy.cbrt, "Vector4D"] = lambda v: v.tau2**0.16666666666666666
+behavior[numpy.cbrt, "Momentum2D"] = lambda v: v.rho2**0.16666666666666666
+behavior[numpy.cbrt, "Momentum3D"] = lambda v: v.mag2**0.16666666666666666
+behavior[numpy.cbrt, "Momentum4D"] = lambda v: v.tau2**0.16666666666666666
 
 behavior[numpy.power, "Vector2D", numbers.Real] = (
-    lambda v, expo: v.rho2 if expo == 2 else v.rho ** expo
+    lambda v, expo: v.rho2 if expo == 2 else v.rho**expo
 )
 behavior[numpy.power, "Vector3D", numbers.Real] = (
-    lambda v, expo: v.mag2 if expo == 2 else v.mag ** expo
+    lambda v, expo: v.mag2 if expo == 2 else v.mag**expo
 )
 behavior[numpy.power, "Vector4D", numbers.Real] = (
-    lambda v, expo: v.tau2 if expo == 2 else v.tau ** expo
+    lambda v, expo: v.tau2 if expo == 2 else v.tau**expo
 )
 behavior[numpy.power, "Momentum2D", numbers.Real] = (
-    lambda v, expo: v.rho2 if expo == 2 else v.rho ** expo
+    lambda v, expo: v.rho2 if expo == 2 else v.rho**expo
 )
 behavior[numpy.power, "Momentum3D", numbers.Real] = (
-    lambda v, expo: v.mag2 if expo == 2 else v.mag ** expo
+    lambda v, expo: v.mag2 if expo == 2 else v.mag**expo
 )
 behavior[numpy.power, "Momentum4D", numbers.Real] = (
-    lambda v, expo: v.tau2 if expo == 2 else v.tau ** expo
+    lambda v, expo: v.tau2 if expo == 2 else v.tau**expo
 )
 
 behavior["__cast__", VectorNumpy2D] = lambda v: vector.Array(v)

src/vector/_backends/numba_object.py

@@ -3170,21 +3170,21 @@ def numpy_sqrt(v):
     if isinstance(v, VectorObject2DType):
 
         def numpy_sqrt_impl(v):
-            return v.rho2 ** 0.25
+            return v.rho2**0.25
 
         return numpy_sqrt_impl
 
     elif isinstance(v, VectorObject3DType):
 
         def numpy_sqrt_impl(v):
-            return v.mag2 ** 0.25
+            return v.mag2**0.25
 
         return numpy_sqrt_impl
 
     elif isinstance(v, VectorObject4DType):
 
         def numpy_sqrt_impl(v):
-            return v.tau2 ** 0.25
+            return v.tau2**0.25
 
         return numpy_sqrt_impl
 
@@ -3194,21 +3194,21 @@ def numpy_cbrt(v):
     if isinstance(v, VectorObject2DType):
 
         def numpy_cbrt_impl(v):
-            return v.rho2 ** 0.16666666666666666
+            return v.rho2**0.16666666666666666
 
         return numpy_cbrt_impl
 
     elif isinstance(v, VectorObject3DType):
 
         def numpy_cbrt_impl(v):
-            return v.mag2 ** 0.16666666666666666
+            return v.mag2**0.16666666666666666
 
         return numpy_cbrt_impl
 
     elif isinstance(v, VectorObject4DType):
 
         def numpy_cbrt_impl(v):
-            return v.tau2 ** 0.16666666666666666
+            return v.tau2**0.16666666666666666
 
         return numpy_cbrt_impl
 
@@ -3223,7 +3223,7 @@ def numpy_power_impl(v, expo):
             if expo == 2:
                 return v.rho2
             else:
-                return v.rho ** expo
+                return v.rho**expo
 
         return numpy_power_impl
 
@@ -3235,7 +3235,7 @@ def numpy_power_impl(v, expo):
             if expo == 2:
                 return v.mag2
             else:
-                return v.mag ** expo
+                return v.mag**expo
 
         return numpy_power_impl
 
@@ -3247,7 +3247,7 @@ def numpy_power_impl(v, expo):
             if expo == 2:
                 return v.tau2
             else:
-                return v.tau ** expo
+                return v.tau**expo
 
         return numpy_power_impl
 

src/vector/_compute/lorentz/Et.py

@@ -57,7 +57,7 @@ def xy_eta_tau(lib, x, y, eta, tau):
 
 
 def rhophi_z_t(lib, rho, phi, z, t):
-    return t * rho / lib.sqrt(rho ** 2 + z ** 2)
+    return t * rho / lib.sqrt(rho**2 + z**2)
 
 
 def rhophi_z_tau(lib, rho, phi, z, tau):

src/vector/_compute/lorentz/Et2.py

@@ -32,8 +32,8 @@
 
 
 def xy_z_t(lib, x, y, z, t):
-    pt2 = x ** 2 + y ** 2
-    return t ** 2 * pt2 / (pt2 + z ** 2)
+    pt2 = x**2 + y**2
+    return t**2 * pt2 / (pt2 + z**2)
 
 
 def xy_z_tau(lib, x, y, z, tau):
@@ -58,8 +58,8 @@ def xy_eta_tau(lib, x, y, eta, tau):
 
 
 def rhophi_z_t(lib, rho, phi, z, t):
-    pt2 = rho ** 2
-    return t ** 2 * pt2 / (pt2 + z ** 2)
+    pt2 = rho**2
+    return t**2 * pt2 / (pt2 + z**2)
 
 
 def rhophi_z_tau(lib, rho, phi, z, tau):

src/vector/_compute/lorentz/Mt2.py

@@ -33,51 +33,51 @@
 
 
 def xy_z_t(lib, x, y, z, t):
-    return t ** 2 - z ** 2
+    return t**2 - z**2
 
 
 def xy_z_tau(lib, x, y, z, tau):
-    return lib.maximum(tau2.xy_z_tau(lib, x, y, z, tau) + x ** 2 + y ** 2, 0)
+    return lib.maximum(tau2.xy_z_tau(lib, x, y, z, tau) + x**2 + y**2, 0)
 
 
 def xy_theta_t(lib, x, y, theta, t):
     return xy_z_t(lib, x, y, z.xy_theta(lib, x, y, theta), t)
 
 
 def xy_theta_tau(lib, x, y, theta, tau):
-    return lib.maximum(tau2.xy_theta_tau(lib, x, y, theta, tau) + x ** 2 + y ** 2, 0)
+    return lib.maximum(tau2.xy_theta_tau(lib, x, y, theta, tau) + x**2 + y**2, 0)
 
 
 def xy_eta_t(lib, x, y, eta, t):
     return xy_z_t(lib, x, y, z.xy_eta(lib, x, y, eta), t)
 
 
 def xy_eta_tau(lib, x, y, eta, tau):
-    return lib.maximum(tau2.xy_eta_tau(lib, x, y, eta, tau) + x ** 2 + y ** 2, 0)
+    return lib.maximum(tau2.xy_eta_tau(lib, x, y, eta, tau) + x**2 + y**2, 0)
 
 
 def rhophi_z_t(lib, rho, phi, z, t):
-    return t ** 2 - z ** 2
+    return t**2 - z**2
 
 
 def rhophi_z_tau(lib, rho, phi, z, tau):
-    return lib.maximum(tau2.rhophi_z_tau(lib, rho, phi, z, tau) + rho ** 2, 0)
+    return lib.maximum(tau2.rhophi_z_tau(lib, rho, phi, z, tau) + rho**2, 0)
 
 
 def rhophi_theta_t(lib, rho, phi, theta, t):
     return rhophi_z_t(lib, rho, phi, z.rhophi_theta(lib, rho, phi, theta), t)
 
 
 def rhophi_theta_tau(lib, rho, phi, theta, tau):
-    return lib.maximum(tau2.rhophi_theta_tau(lib, rho, phi, theta, tau) + rho ** 2, 0)
+    return lib.maximum(tau2.rhophi_theta_tau(lib, rho, phi, theta, tau) + rho**2, 0)
 
 
 def rhophi_eta_t(lib, rho, phi, eta, t):
     return rhophi_z_t(lib, rho, phi, z.rhophi_eta(lib, rho, phi, eta), t)
 
 
 def rhophi_eta_tau(lib, rho, phi, eta, tau):
-    return lib.maximum(tau2.rhophi_eta_tau(lib, rho, phi, eta, tau) + rho ** 2, 0)
+    return lib.maximum(tau2.rhophi_eta_tau(lib, rho, phi, eta, tau) + rho**2, 0)
 
 
 dispatch_map = {

src/vector/_compute/lorentz/boostX_beta.py

@@ -33,7 +33,7 @@
 
 
 def xy_z_t(lib, beta, x, y, z, t):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x
     why = y
@@ -43,7 +43,7 @@ def xy_z_t(lib, beta, x, y, z, t):
 
 
 def xy_z_tau(lib, beta, x, y, z, tau):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x
     why = y
@@ -53,7 +53,7 @@ def xy_z_tau(lib, beta, x, y, z, tau):
 
 
 def xy_theta_t(lib, beta, x, y, theta, t):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x
     why = y
@@ -63,7 +63,7 @@ def xy_theta_t(lib, beta, x, y, theta, t):
 
 
 def xy_theta_tau(lib, beta, x, y, theta, tau):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x
     why = y
@@ -73,7 +73,7 @@ def xy_theta_tau(lib, beta, x, y, theta, tau):
 
 
 def xy_eta_t(lib, beta, x, y, eta, t):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x
     why = y
@@ -83,7 +83,7 @@ def xy_eta_t(lib, beta, x, y, eta, t):
 
 
 def xy_eta_tau(lib, beta, x, y, eta, tau):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x
     why = y
@@ -93,7 +93,7 @@ def xy_eta_tau(lib, beta, x, y, eta, tau):
 
 
 def rhophi_z_t(lib, beta, rho, phi, z, t):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x.rhophi(lib, rho, phi)
     why = y.rhophi(lib, rho, phi)
@@ -103,7 +103,7 @@ def rhophi_z_t(lib, beta, rho, phi, z, t):
 
 
 def rhophi_z_tau(lib, beta, rho, phi, z, tau):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x.rhophi(lib, rho, phi)
     why = y.rhophi(lib, rho, phi)
@@ -113,7 +113,7 @@ def rhophi_z_tau(lib, beta, rho, phi, z, tau):
 
 
 def rhophi_theta_t(lib, beta, rho, phi, theta, t):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x.rhophi(lib, rho, phi)
     why = y.rhophi(lib, rho, phi)
@@ -123,7 +123,7 @@ def rhophi_theta_t(lib, beta, rho, phi, theta, t):
 
 
 def rhophi_theta_tau(lib, beta, rho, phi, theta, tau):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x.rhophi(lib, rho, phi)
     why = y.rhophi(lib, rho, phi)
@@ -133,7 +133,7 @@ def rhophi_theta_tau(lib, beta, rho, phi, theta, tau):
 
 
 def rhophi_eta_t(lib, beta, rho, phi, eta, t):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x.rhophi(lib, rho, phi)
     why = y.rhophi(lib, rho, phi)
@@ -143,7 +143,7 @@ def rhophi_eta_t(lib, beta, rho, phi, eta, t):
 
 
 def rhophi_eta_tau(lib, beta, rho, phi, eta, tau):
-    gam = (1 - beta ** 2) ** -0.5
+    gam = (1 - beta**2) ** -0.5
     bgam = beta * gam
     exx = x.rhophi(lib, rho, phi)
     why = y.rhophi(lib, rho, phi)