Move Identity operator initialization close to class definition

pymc/aesaraf.py

@@ -584,6 +584,10 @@ def __hash__(self):
         return hash(type(self))
 
 
+scalar_identity = IdentityOp(scalar.upgrade_to_float, name="scalar_identity")
+identity = Elemwise(scalar_identity, name="identity")
+
+
 def make_shared_replacements(point, vars, model):
     """
     Makes shared replacements for all *other* variables than the ones passed.
@@ -688,10 +692,6 @@ def __call__(self, input):
         return aesara.clone_replace(self.tensor, {oldinput: input}, strict=False)
 
 
-scalar_identity = IdentityOp(scalar.upgrade_to_float, name="scalar_identity")
-identity = Elemwise(scalar_identity, name="identity")
-
-
 class GeneratorOp(Op):
     """
     Generator Op is designed for storing python generators inside aesara graph.