fixup

Author: wsmoses

test/train.jl

@@ -54,9 +54,9 @@ for (trainfn!, name) in ((Flux.train!, "Zygote"), (train_enzyme!, "Enzyme"))
     end
     @test CNT[] == 51  # stopped early
     if name != "Enzyme"
-      @test m1.weight[1] ≈ 0.0  # did not corrupt weights
-    else
       @test m1.weight[1] ≈ -5  # did not corrupt weights
+    else
+      @test m1.weight[1] ≈ 0.0  # did not corrupt weights
     end
   end