Skip to content

Commit 0b9bf7e

Browse files
pre-commit-ci[bot]pre-commit-ci[bot]
authored
[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>
1 parent d49183b commit 0b9bf7e

27 files changed

Lines changed: 196 additions & 196 deletions

‎.pre-commit-config.yaml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ repos:
66
args: [--py36-plus]
77

88
- repo: https://github.com/psf/black
9-
rev: 21.12b0
9+
rev: 22.1.0
1010
hooks:
1111
- id: black
1212

‎src/vector/_backends/awkward_.py‎

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -865,37 +865,37 @@ class MomentumRecord4D(MomentumAwkward4D, ak.Record): # type: ignore[misc]
865865
behavior[numpy.square, "Momentum3D"] = lambda v: v.mag2
866866
behavior[numpy.square, "Momentum4D"] = lambda v: v.tau2
867867

868-
behavior[numpy.sqrt, "Vector2D"] = lambda v: v.rho2 ** 0.25
869-
behavior[numpy.sqrt, "Vector3D"] = lambda v: v.mag2 ** 0.25
870-
behavior[numpy.sqrt, "Vector4D"] = lambda v: v.tau2 ** 0.25
871-
behavior[numpy.sqrt, "Momentum2D"] = lambda v: v.rho2 ** 0.25
872-
behavior[numpy.sqrt, "Momentum3D"] = lambda v: v.mag2 ** 0.25
873-
behavior[numpy.sqrt, "Momentum4D"] = lambda v: v.tau2 ** 0.25
874-
875-
behavior[numpy.cbrt, "Vector2D"] = lambda v: v.rho2 ** 0.16666666666666666
876-
behavior[numpy.cbrt, "Vector3D"] = lambda v: v.mag2 ** 0.16666666666666666
877-
behavior[numpy.cbrt, "Vector4D"] = lambda v: v.tau2 ** 0.16666666666666666
878-
behavior[numpy.cbrt, "Momentum2D"] = lambda v: v.rho2 ** 0.16666666666666666
879-
behavior[numpy.cbrt, "Momentum3D"] = lambda v: v.mag2 ** 0.16666666666666666
880-
behavior[numpy.cbrt, "Momentum4D"] = lambda v: v.tau2 ** 0.16666666666666666
868+
behavior[numpy.sqrt, "Vector2D"] = lambda v: v.rho2**0.25
869+
behavior[numpy.sqrt, "Vector3D"] = lambda v: v.mag2**0.25
870+
behavior[numpy.sqrt, "Vector4D"] = lambda v: v.tau2**0.25
871+
behavior[numpy.sqrt, "Momentum2D"] = lambda v: v.rho2**0.25
872+
behavior[numpy.sqrt, "Momentum3D"] = lambda v: v.mag2**0.25
873+
behavior[numpy.sqrt, "Momentum4D"] = lambda v: v.tau2**0.25
874+
875+
behavior[numpy.cbrt, "Vector2D"] = lambda v: v.rho2**0.16666666666666666
876+
behavior[numpy.cbrt, "Vector3D"] = lambda v: v.mag2**0.16666666666666666
877+
behavior[numpy.cbrt, "Vector4D"] = lambda v: v.tau2**0.16666666666666666
878+
behavior[numpy.cbrt, "Momentum2D"] = lambda v: v.rho2**0.16666666666666666
879+
behavior[numpy.cbrt, "Momentum3D"] = lambda v: v.mag2**0.16666666666666666
880+
behavior[numpy.cbrt, "Momentum4D"] = lambda v: v.tau2**0.16666666666666666
881881

882882
behavior[numpy.power, "Vector2D", numbers.Real] = (
883-
lambda v, expo: v.rho2 if expo == 2 else v.rho ** expo
883+
lambda v, expo: v.rho2 if expo == 2 else v.rho**expo
884884
)
885885
behavior[numpy.power, "Vector3D", numbers.Real] = (
886-
lambda v, expo: v.mag2 if expo == 2 else v.mag ** expo
886+
lambda v, expo: v.mag2 if expo == 2 else v.mag**expo
887887
)
888888
behavior[numpy.power, "Vector4D", numbers.Real] = (
889-
lambda v, expo: v.tau2 if expo == 2 else v.tau ** expo
889+
lambda v, expo: v.tau2 if expo == 2 else v.tau**expo
890890
)
891891
behavior[numpy.power, "Momentum2D", numbers.Real] = (
892-
lambda v, expo: v.rho2 if expo == 2 else v.rho ** expo
892+
lambda v, expo: v.rho2 if expo == 2 else v.rho**expo
893893
)
894894
behavior[numpy.power, "Momentum3D", numbers.Real] = (
895-
lambda v, expo: v.mag2 if expo == 2 else v.mag ** expo
895+
lambda v, expo: v.mag2 if expo == 2 else v.mag**expo
896896
)
897897
behavior[numpy.power, "Momentum4D", numbers.Real] = (
898-
lambda v, expo: v.tau2 if expo == 2 else v.tau ** expo
898+
lambda v, expo: v.tau2 if expo == 2 else v.tau**expo
899899
)
900900

901901
behavior["__cast__", VectorNumpy2D] = lambda v: vector.Array(v)

‎src/vector/_backends/numba_object.py‎

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3170,21 +3170,21 @@ def numpy_sqrt(v):
31703170
if isinstance(v, VectorObject2DType):
31713171

31723172
def numpy_sqrt_impl(v):
3173-
return v.rho2 ** 0.25
3173+
return v.rho2**0.25
31743174

31753175
return numpy_sqrt_impl
31763176

31773177
elif isinstance(v, VectorObject3DType):
31783178

31793179
def numpy_sqrt_impl(v):
3180-
return v.mag2 ** 0.25
3180+
return v.mag2**0.25
31813181

31823182
return numpy_sqrt_impl
31833183

31843184
elif isinstance(v, VectorObject4DType):
31853185

31863186
def numpy_sqrt_impl(v):
3187-
return v.tau2 ** 0.25
3187+
return v.tau2**0.25
31883188

31893189
return numpy_sqrt_impl
31903190

@@ -3194,21 +3194,21 @@ def numpy_cbrt(v):
31943194
if isinstance(v, VectorObject2DType):
31953195

31963196
def numpy_cbrt_impl(v):
3197-
return v.rho2 ** 0.16666666666666666
3197+
return v.rho2**0.16666666666666666
31983198

31993199
return numpy_cbrt_impl
32003200

32013201
elif isinstance(v, VectorObject3DType):
32023202

32033203
def numpy_cbrt_impl(v):
3204-
return v.mag2 ** 0.16666666666666666
3204+
return v.mag2**0.16666666666666666
32053205

32063206
return numpy_cbrt_impl
32073207

32083208
elif isinstance(v, VectorObject4DType):
32093209

32103210
def numpy_cbrt_impl(v):
3211-
return v.tau2 ** 0.16666666666666666
3211+
return v.tau2**0.16666666666666666
32123212

32133213
return numpy_cbrt_impl
32143214

@@ -3223,7 +3223,7 @@ def numpy_power_impl(v, expo):
32233223
if expo == 2:
32243224
return v.rho2
32253225
else:
3226-
return v.rho ** expo
3226+
return v.rho**expo
32273227

32283228
return numpy_power_impl
32293229

@@ -3235,7 +3235,7 @@ def numpy_power_impl(v, expo):
32353235
if expo == 2:
32363236
return v.mag2
32373237
else:
3238-
return v.mag ** expo
3238+
return v.mag**expo
32393239

32403240
return numpy_power_impl
32413241

@@ -3247,7 +3247,7 @@ def numpy_power_impl(v, expo):
32473247
if expo == 2:
32483248
return v.tau2
32493249
else:
3250-
return v.tau ** expo
3250+
return v.tau**expo
32513251

32523252
return numpy_power_impl
32533253

‎src/vector/_compute/lorentz/Et.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def xy_eta_tau(lib, x, y, eta, tau):
5757

5858

5959
def rhophi_z_t(lib, rho, phi, z, t):
60-
return t * rho / lib.sqrt(rho ** 2 + z ** 2)
60+
return t * rho / lib.sqrt(rho**2 + z**2)
6161

6262

6363
def rhophi_z_tau(lib, rho, phi, z, tau):

‎src/vector/_compute/lorentz/Et2.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@
3232

3333

3434
def xy_z_t(lib, x, y, z, t):
35-
pt2 = x ** 2 + y ** 2
36-
return t ** 2 * pt2 / (pt2 + z ** 2)
35+
pt2 = x**2 + y**2
36+
return t**2 * pt2 / (pt2 + z**2)
3737

3838

3939
def xy_z_tau(lib, x, y, z, tau):
@@ -58,8 +58,8 @@ def xy_eta_tau(lib, x, y, eta, tau):
5858

5959

6060
def rhophi_z_t(lib, rho, phi, z, t):
61-
pt2 = rho ** 2
62-
return t ** 2 * pt2 / (pt2 + z ** 2)
61+
pt2 = rho**2
62+
return t**2 * pt2 / (pt2 + z**2)
6363

6464

6565
def rhophi_z_tau(lib, rho, phi, z, tau):

‎src/vector/_compute/lorentz/Mt2.py‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -33,51 +33,51 @@
3333

3434

3535
def xy_z_t(lib, x, y, z, t):
36-
return t ** 2 - z ** 2
36+
return t**2 - z**2
3737

3838

3939
def xy_z_tau(lib, x, y, z, tau):
40-
return lib.maximum(tau2.xy_z_tau(lib, x, y, z, tau) + x ** 2 + y ** 2, 0)
40+
return lib.maximum(tau2.xy_z_tau(lib, x, y, z, tau) + x**2 + y**2, 0)
4141

4242

4343
def xy_theta_t(lib, x, y, theta, t):
4444
return xy_z_t(lib, x, y, z.xy_theta(lib, x, y, theta), t)
4545

4646

4747
def xy_theta_tau(lib, x, y, theta, tau):
48-
return lib.maximum(tau2.xy_theta_tau(lib, x, y, theta, tau) + x ** 2 + y ** 2, 0)
48+
return lib.maximum(tau2.xy_theta_tau(lib, x, y, theta, tau) + x**2 + y**2, 0)
4949

5050

5151
def xy_eta_t(lib, x, y, eta, t):
5252
return xy_z_t(lib, x, y, z.xy_eta(lib, x, y, eta), t)
5353

5454

5555
def xy_eta_tau(lib, x, y, eta, tau):
56-
return lib.maximum(tau2.xy_eta_tau(lib, x, y, eta, tau) + x ** 2 + y ** 2, 0)
56+
return lib.maximum(tau2.xy_eta_tau(lib, x, y, eta, tau) + x**2 + y**2, 0)
5757

5858

5959
def rhophi_z_t(lib, rho, phi, z, t):
60-
return t ** 2 - z ** 2
60+
return t**2 - z**2
6161

6262

6363
def rhophi_z_tau(lib, rho, phi, z, tau):
64-
return lib.maximum(tau2.rhophi_z_tau(lib, rho, phi, z, tau) + rho ** 2, 0)
64+
return lib.maximum(tau2.rhophi_z_tau(lib, rho, phi, z, tau) + rho**2, 0)
6565

6666

6767
def rhophi_theta_t(lib, rho, phi, theta, t):
6868
return rhophi_z_t(lib, rho, phi, z.rhophi_theta(lib, rho, phi, theta), t)
6969

7070

7171
def rhophi_theta_tau(lib, rho, phi, theta, tau):
72-
return lib.maximum(tau2.rhophi_theta_tau(lib, rho, phi, theta, tau) + rho ** 2, 0)
72+
return lib.maximum(tau2.rhophi_theta_tau(lib, rho, phi, theta, tau) + rho**2, 0)
7373

7474

7575
def rhophi_eta_t(lib, rho, phi, eta, t):
7676
return rhophi_z_t(lib, rho, phi, z.rhophi_eta(lib, rho, phi, eta), t)
7777

7878

7979
def rhophi_eta_tau(lib, rho, phi, eta, tau):
80-
return lib.maximum(tau2.rhophi_eta_tau(lib, rho, phi, eta, tau) + rho ** 2, 0)
80+
return lib.maximum(tau2.rhophi_eta_tau(lib, rho, phi, eta, tau) + rho**2, 0)
8181

8282

8383
dispatch_map = {

‎src/vector/_compute/lorentz/boostX_beta.py‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333

3434

3535
def xy_z_t(lib, beta, x, y, z, t):
36-
gam = (1 - beta ** 2) ** -0.5
36+
gam = (1 - beta**2) ** -0.5
3737
bgam = beta * gam
3838
exx = x
3939
why = y
@@ -43,7 +43,7 @@ def xy_z_t(lib, beta, x, y, z, t):
4343

4444

4545
def xy_z_tau(lib, beta, x, y, z, tau):
46-
gam = (1 - beta ** 2) ** -0.5
46+
gam = (1 - beta**2) ** -0.5
4747
bgam = beta * gam
4848
exx = x
4949
why = y
@@ -53,7 +53,7 @@ def xy_z_tau(lib, beta, x, y, z, tau):
5353

5454

5555
def xy_theta_t(lib, beta, x, y, theta, t):
56-
gam = (1 - beta ** 2) ** -0.5
56+
gam = (1 - beta**2) ** -0.5
5757
bgam = beta * gam
5858
exx = x
5959
why = y
@@ -63,7 +63,7 @@ def xy_theta_t(lib, beta, x, y, theta, t):
6363

6464

6565
def xy_theta_tau(lib, beta, x, y, theta, tau):
66-
gam = (1 - beta ** 2) ** -0.5
66+
gam = (1 - beta**2) ** -0.5
6767
bgam = beta * gam
6868
exx = x
6969
why = y
@@ -73,7 +73,7 @@ def xy_theta_tau(lib, beta, x, y, theta, tau):
7373

7474

7575
def xy_eta_t(lib, beta, x, y, eta, t):
76-
gam = (1 - beta ** 2) ** -0.5
76+
gam = (1 - beta**2) ** -0.5
7777
bgam = beta * gam
7878
exx = x
7979
why = y
@@ -83,7 +83,7 @@ def xy_eta_t(lib, beta, x, y, eta, t):
8383

8484

8585
def xy_eta_tau(lib, beta, x, y, eta, tau):
86-
gam = (1 - beta ** 2) ** -0.5
86+
gam = (1 - beta**2) ** -0.5
8787
bgam = beta * gam
8888
exx = x
8989
why = y
@@ -93,7 +93,7 @@ def xy_eta_tau(lib, beta, x, y, eta, tau):
9393

9494

9595
def rhophi_z_t(lib, beta, rho, phi, z, t):
96-
gam = (1 - beta ** 2) ** -0.5
96+
gam = (1 - beta**2) ** -0.5
9797
bgam = beta * gam
9898
exx = x.rhophi(lib, rho, phi)
9999
why = y.rhophi(lib, rho, phi)
@@ -103,7 +103,7 @@ def rhophi_z_t(lib, beta, rho, phi, z, t):
103103

104104

105105
def rhophi_z_tau(lib, beta, rho, phi, z, tau):
106-
gam = (1 - beta ** 2) ** -0.5
106+
gam = (1 - beta**2) ** -0.5
107107
bgam = beta * gam
108108
exx = x.rhophi(lib, rho, phi)
109109
why = y.rhophi(lib, rho, phi)
@@ -113,7 +113,7 @@ def rhophi_z_tau(lib, beta, rho, phi, z, tau):
113113

114114

115115
def rhophi_theta_t(lib, beta, rho, phi, theta, t):
116-
gam = (1 - beta ** 2) ** -0.5
116+
gam = (1 - beta**2) ** -0.5
117117
bgam = beta * gam
118118
exx = x.rhophi(lib, rho, phi)
119119
why = y.rhophi(lib, rho, phi)
@@ -123,7 +123,7 @@ def rhophi_theta_t(lib, beta, rho, phi, theta, t):
123123

124124

125125
def rhophi_theta_tau(lib, beta, rho, phi, theta, tau):
126-
gam = (1 - beta ** 2) ** -0.5
126+
gam = (1 - beta**2) ** -0.5
127127
bgam = beta * gam
128128
exx = x.rhophi(lib, rho, phi)
129129
why = y.rhophi(lib, rho, phi)
@@ -133,7 +133,7 @@ def rhophi_theta_tau(lib, beta, rho, phi, theta, tau):
133133

134134

135135
def rhophi_eta_t(lib, beta, rho, phi, eta, t):
136-
gam = (1 - beta ** 2) ** -0.5
136+
gam = (1 - beta**2) ** -0.5
137137
bgam = beta * gam
138138
exx = x.rhophi(lib, rho, phi)
139139
why = y.rhophi(lib, rho, phi)
@@ -143,7 +143,7 @@ def rhophi_eta_t(lib, beta, rho, phi, eta, t):
143143

144144

145145
def rhophi_eta_tau(lib, beta, rho, phi, eta, tau):
146-
gam = (1 - beta ** 2) ** -0.5
146+
gam = (1 - beta**2) ** -0.5
147147
bgam = beta * gam
148148
exx = x.rhophi(lib, rho, phi)
149149
why = y.rhophi(lib, rho, phi)

0 commit comments

Comments
 (0)