Fixup tests and tweak env var handling

Author: ToucheSir

.buildkite/pipeline.yml

@@ -21,7 +21,7 @@ steps:
     env:
       GROUP: "All"
       TEST_FAST: true
-      TEST_WORKERS: 4
+      RETESTITEMS_NWORKERS: 4
     if: build.message !~ /\[skip tests\]/
     timeout_in_minutes: 180
     matrix:

test/convnet_tests.jl

@@ -170,7 +170,7 @@ end
 end
 
 @testitem "EfficientNet" setup=[TestModels] begin
-    config = TEST_FAST ? [:b0] : [:b0, :b1, :b2, :b3, :b4, :b5] #:b6, :b7, :b8]
+    configs = TEST_FAST ? [:b0] : [:b0, :b1, :b2, :b3, :b4, :b5] #:b6, :b7, :b8]
     @testset "EfficientNet($config)" for config in configs
         # preferred image resolution scaling
         r = Metalhead.EFFICIENTNET_GLOBAL_CONFIGS[config][1]
@@ -235,7 +235,11 @@ end
     m = SqueezeNet() |> gpu
     @test size(m(x_224)) == (1000, 1)
     if SqueezeNet in PRETRAINED_MODELS
-        @test acctest(SqueezeNet(; pretrain = true))
+        if VERSION >= v"1.7"
+            @test acctest(SqueezeNet(; pretrain = true))
+        else
+            @test_broken acctest(SqueezeNet(; pretrain = true))
+        end
     else
         @test_throws ArgumentError SqueezeNet(pretrain = true)
     end

test/model_tests.jl

@@ -72,7 +72,7 @@ end
 const TEST_LBLS = readlines(download("https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt"))
 
 function acctest(model)
-    ypred = gpu(model)(TEST_X) |> vec
+    ypred = gpu(model)(TEST_X) |> collect |> vec
     top5 = TEST_LBLS[sortperm(ypred; rev = true)]
     return "monarch" in top5
 end

test/runtests.jl

@@ -7,9 +7,12 @@ const name_filter = test_group == "All" ? nothing : Regex(test_group)
 
 @static if VERSION >= v"1.7"
     using ReTestItems
-    verbose_results = get(ENV, "CI", "false") == "true"
-    nworkers = parse(Int, get(ENV, "TEST_WORKERS", "0"))
-    runtests(Metalhead; name = name_filter, verbose_results, nworkers)
+    if parse(Bool, get(ENV, "CI", "false"))
+        runtests(Metalhead; name = name_filter, verbose_results = true)
+    else
+        # For running locally
+        runtests(Metalhead; name = name_filter)
+    end
 else
     using TestItemRunner
     function testitem_filter(ti)