fix: syntax

Author: avik-pal

test/ops.jl

@@ -108,8 +108,8 @@ end
         ],
     )
 
-    @test cholesky(Array(x)).U ≈ @jit g1(x) broken = RunningOnTPU
-    @test adjoint(cholesky(Array(x)).U) ≈ @jit g2(x) broken = RunningOnTPU
+    @test cholesky(Array(x)).U ≈ @jit(g1(x)) broken = RunningOnTPU
+    @test adjoint(cholesky(Array(x)).U) ≈ @jit(g2(x)) broken = RunningOnTPU
 end
 
 @testset "clamp" begin

test/optimize_comm.jl

@@ -37,7 +37,6 @@ if length(addressable_devices) ≥ 8
         @test !contains(hlo, "all-gather")
         @test contains(hlo, "collective-permute")
 
-        rotate(x)
         @jit shardy_passes = :to_mhlo_shardings rotate(rx)
         @test all(x .== convert(Array, rx))
     end
@@ -75,7 +74,6 @@ if length(addressable_devices) ≥ 8
         ry = Reactant.to_rarray(y; sharding)
 
         hlo = repr(@code_xla shardy_passes = :to_mhlo_shardings dus(rx, ry))
-        println(hlo)
         @test !contains(hlo, "all-to-all")
         @test !contains(hlo, "all-gather")
         @test contains(hlo, "collective-permute")