Respect Op class in clone

pytensor/scalar/basic.py

@@ -4301,7 +4301,7 @@ class Composite(ScalarInnerGraphOp):
         return self._fgraph
 
     def clone(self):
-        return Composite(self.fgraph.inputs, self.fgraph.outputs)
+        return self.__class__(self.fgraph.inputs, self.fgraph.outputs)
 
     def output_types(self, input_types):
         if tuple(input_types) != self.inputs_type:

pytensor/scalar/loop.py

@@ -123,7 +123,7 @@ class ScalarLoop(ScalarInnerGraphOp):
             update, until = self.outputs, None
         init = self.inputs[: len(update)]
         constant = self.inputs[len(update) :]
-        return ScalarLoop(
+        return self.__class__(
             init=init,
             update=update,
             constant=constant,