Skip to content

Commit 3e65827

Browse files
Address reviewer feedback
1 parent 32b14d2 commit 3e65827

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pytensor/tensor/pad.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -379,7 +379,8 @@ def pad(x: TensorLike, pad_width: TensorLike, mode: PadMode = "constant", **kwar
379379

380380
elif mode in ["maximum", "minimum", "mean", "median"]:
381381
if mode == "median":
382-
# TODO: pt.quantile? pt.median?
382+
# TODO: Revisit this after we implement a quantile function.
383+
# See https://github.com/pymc-devs/pytensor/issues/53
383384
raise NotImplementedError("Median padding not implemented")
384385
stat_func = cast(Callable, stat_funcs[mode])
385386
stat_length = kwargs.get("stat_length")
@@ -428,4 +429,4 @@ def pad(x: TensorLike, pad_width: TensorLike, mode: PadMode = "constant", **kwar
428429
return op
429430

430431

431-
__all__ = ["pad"]
432+
__all__ = ["pad", "flip"]

0 commit comments

Comments
 (0)