Register useless_switch in infer_shape

ricardoV94 · ricardoV94 · commit 75b8ee9bbcac · 2025-07-29T16:40:59.000+02:00
diff --git a/pytensor/tensor/rewriting/basic.py b/pytensor/tensor/rewriting/basic.py
@@ -949,6 +949,7 @@ def equivalent_up_to_constant_casting(a, b) -> bool:
     return False
 
 
+@register_infer_shape
 @register_useless("shape_unsafe")
 @register_canonicalize("fast_compile", "shape_unsafe")
 @register_specialize("shape_unsafe")
