Skip to content

Commit acb1a54

Browse files
committed
Remove unused position tags
These should have been provided as keyword arguments, but they don't seem to matter anyway
1 parent 5598298 commit acb1a54

File tree

8 files changed

+11
-20
lines changed

8 files changed

+11
-20
lines changed

pymc/logprob/censoring.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,6 @@ def find_measurable_clips(fgraph: FunctionGraph, node: Node) -> Optional[List[Me
110110
measurable_ir_rewrites_db.register(
111111
"find_measurable_clips",
112112
find_measurable_clips,
113-
0,
114113
"basic",
115114
"censoring",
116115
)
@@ -220,7 +219,6 @@ def find_measurable_roundings(fgraph: FunctionGraph, node: Node) -> Optional[Lis
220219
measurable_ir_rewrites_db.register(
221220
"find_measurable_roundings",
222221
find_measurable_roundings,
223-
0,
224222
"basic",
225223
"censoring",
226224
)

pymc/logprob/cumsum.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,6 @@ def find_measurable_cumsums(fgraph, node) -> Optional[List[MeasurableCumsum]]:
122122
measurable_ir_rewrites_db.register(
123123
"find_measurable_cumsums",
124124
find_measurable_cumsums,
125-
0,
126125
"basic",
127126
"cumsum",
128127
)

pymc/logprob/mixture.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -461,7 +461,6 @@ def logprob_MixtureRV(
461461
[mixture_replace, switch_mixture_replace],
462462
max_use_ratio=pytensor.config.optdb__max_use_ratio,
463463
),
464-
0,
465464
"basic",
466465
"mixture",
467466
)

pymc/logprob/rewriting.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -252,23 +252,23 @@ def incsubtensor_rv_replace(fgraph, node):
252252

253253
logprob_rewrites_db = SequenceDB()
254254
logprob_rewrites_db.name = "logprob_rewrites_db"
255-
logprob_rewrites_db.register("pre-canonicalize", optdb.query("+canonicalize"), -10, "basic")
255+
logprob_rewrites_db.register("pre-canonicalize", optdb.query("+canonicalize"), "basic")
256256

257257
# These rewrites convert un-measurable variables into their measurable forms,
258258
# but they need to be reapplied, because some of the measurable forms require
259259
# their inputs to be measurable.
260260
measurable_ir_rewrites_db = NoCallbackEquilibriumDB()
261261
measurable_ir_rewrites_db.name = "measurable_ir_rewrites_db"
262262

263-
logprob_rewrites_db.register("measurable_ir_rewrites", measurable_ir_rewrites_db, -10, "basic")
263+
logprob_rewrites_db.register("measurable_ir_rewrites", measurable_ir_rewrites_db, "basic")
264264

265265
# These rewrites push random/measurable variables "down", making them closer to
266266
# (or eventually) the graph outputs. Often this is done by lifting other `Op`s
267267
# "up" through the random/measurable variables and into their inputs.
268-
measurable_ir_rewrites_db.register("subtensor_lift", local_subtensor_rv_lift, -5, "basic")
269-
measurable_ir_rewrites_db.register("incsubtensor_lift", incsubtensor_rv_replace, -5, "basic")
268+
measurable_ir_rewrites_db.register("subtensor_lift", local_subtensor_rv_lift, "basic")
269+
measurable_ir_rewrites_db.register("incsubtensor_lift", incsubtensor_rv_replace, "basic")
270270

271-
logprob_rewrites_db.register("post-canonicalize", optdb.query("+canonicalize"), 10, "basic")
271+
logprob_rewrites_db.register("post-canonicalize", optdb.query("+canonicalize"), "basic")
272272

273273

274274
def construct_ir_fgraph(

pymc/logprob/scan.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -530,19 +530,17 @@ def _get_measurable_outputs_MeasurableScan(op, node):
530530
# out2in(
531531
# add_opts_to_inner_graphs, name="add_opts_to_inner_graphs", ignore_newtrees=True
532532
# ),
533-
-100,
534533
"basic",
535534
"scan",
536535
)
537536

538537
measurable_ir_rewrites_db.register(
539538
"find_measurable_scans",
540539
find_measurable_scans,
541-
0,
542540
"basic",
543541
"scan",
544542
)
545543

546544
# Add scan canonicalizations that aren't in the canonicalization DB
547-
logprob_rewrites_db.register("scan_eqopt1", scan_eqopt1, -9, "basic", "scan")
548-
logprob_rewrites_db.register("scan_eqopt2", scan_eqopt2, -9, "basic", "scan")
545+
logprob_rewrites_db.register("scan_eqopt1", scan_eqopt1, "basic", "scan")
546+
logprob_rewrites_db.register("scan_eqopt2", scan_eqopt2, "basic", "scan")

pymc/logprob/tensor.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -311,24 +311,21 @@ def find_measurable_dimshuffles(fgraph, node) -> Optional[List[MeasurableDimShuf
311311
return [measurable_dimshuffle]
312312

313313

314-
measurable_ir_rewrites_db.register(
315-
"dimshuffle_lift", local_dimshuffle_rv_lift, -5, "basic", "tensor"
316-
)
314+
measurable_ir_rewrites_db.register("dimshuffle_lift", local_dimshuffle_rv_lift, "basic", "tensor")
317315

318316

319317
# We register this later than `dimshuffle_lift` so that it is only applied as a fallback
320318
measurable_ir_rewrites_db.register(
321-
"find_measurable_dimshuffles", find_measurable_dimshuffles, 0, "basic", "tensor"
319+
"find_measurable_dimshuffles", find_measurable_dimshuffles, "basic", "tensor"
322320
)
323321

324322

325-
measurable_ir_rewrites_db.register("broadcast_to_lift", naive_bcast_rv_lift, -5, "basic", "tensor")
323+
measurable_ir_rewrites_db.register("broadcast_to_lift", naive_bcast_rv_lift, "basic", "tensor")
326324

327325

328326
measurable_ir_rewrites_db.register(
329327
"find_measurable_stacks",
330328
find_measurable_stacks,
331-
0,
332329
"basic",
333330
"tensor",
334331
)

pymc/logprob/transforms.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -373,7 +373,6 @@ def find_measurable_transforms(fgraph: FunctionGraph, node: Node) -> Optional[Li
373373
measurable_ir_rewrites_db.register(
374374
"find_measurable_transforms",
375375
find_measurable_transforms,
376-
0,
377376
"basic",
378377
"transform",
379378
)

scripts/run_mypy.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@
4747
pymc/gp/util.py
4848
pymc/logprob/__init__.py
4949
pymc/logprob/abstract.py
50+
pymc/logprob/cumsum.py
5051
pymc/math.py
5152
pymc/ode/__init__.py
5253
pymc/ode/ode.py

0 commit comments

Comments
 (0)