Raise ShapeError instead of NotImplementedError for unimplemented infer_shape in Op class

Author: Aarsh-Wankar

pytensor/graph/op.py

@@ -602,7 +602,9 @@ def infer_shape(self, fgraph, node, input_shapes):
 
             return _gufunc_to_out_shape(self.gufunc_signature, input_shapes)
         else:
-            raise NotImplementedError(f"Op {self} does not implement infer_shape")
+            from pytensor.tensor.exceptions import ShapeError
+
+            raise ShapeError(f"Op {self} does not implement infer_shape")
 
     def __str__(self):
         return getattr(type(self), "__name__", super().__str__())