chore: apply suggestions from code review

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

Author: avik-pal

test/deeponet_tests.jl

@@ -50,7 +50,7 @@
             @test setup.out_size == size(pred)
 
             __f = (u, y, ps) -> sum(abs2, first(deeponet((u, y), ps, st)))
-            test_gradients(__f, u, y, ps; atol=1f-3, rtol=1f-3)
+            test_gradients(__f, u, y, ps; atol=1.0f-3, rtol=1.0f-3)
         end
 
         @testset "Embedding layer mismatch" begin
@@ -64,7 +64,7 @@
             @test_throws ArgumentError deeponet((u, y), ps, st)
 
             __f = (u, y, ps) -> sum(abs2, first(deeponet((u, y), ps, st)))
-            test_gradients(__f, u, y, ps; atol=1f-3, rtol=1f-3)
+            test_gradients(__f, u, y, ps; atol=1.0f-3, rtol=1.0f-3)
         end
     end
 end

test/fno_tests.jl

@@ -29,7 +29,7 @@
             end broken=broken
 
             __f = (x, ps) -> sum(abs2, first(fno(x, ps, st)))
-            test_gradients(__f, x, ps; atol=1f-3, rtol=1f-3)
+            test_gradients(__f, x, ps; atol=1.0f-3, rtol=1.0f-3)
         end
     end
 end

test/layers_tests.jl

@@ -38,7 +38,7 @@
             end broken=broken
 
             __f = (x, ps) -> sum(abs2, first(m(x, ps, st)))
-            test_gradients(__f, x, ps; atol=1f-3, rtol=1f-3)
+            test_gradients(__f, x, ps; atol=1.0f-3, rtol=1.0f-3)
         end
     end
 end