replace with_module calls with @scope

Author: Sergio Sánchez Ramírez

src/Ops.jl

@@ -3239,15 +3239,11 @@ end
     )
 
     sym_name = Reactant.TracedUtils.__lookup_unique_name_in_module(mod, sym_name)
-
-    mesh_op = MLIR.IR.with_module(mod) do
-        return MLIR.Dialects.sdy.mesh(; sym_name, mesh=mesh_attr, location)
-    end
+    mesh_op = MLIR.Dialects.sdy.mesh(; sym_name, mesh=mesh_attr, location)
 
     # mesh_op needs to be moved to the beginning of the module
     mesh_op = MLIR.IR.rmfromparent!(mesh_op)
-    mod_body = MLIR.IR.body(mod)
-    pushfirst!(mod_body, mesh_op)
+    pushfirst!(MLIR.IR.body(mod), mesh_op)
 
     # We return the name of the mesh, since the operation is a Symbol op
     return (;