update warcraft

Author: BatyLeo

src/Warcraft/Warcraft.jl

@@ -60,8 +60,8 @@ The embedding is made as follows:
 4) The element-wize `neg_tensor` function to get cell weights of proper sign to apply shortest path algorithms.
 5) A squeeze function to forget the two last dimensions.
 """
-function Utils.generate_statistical_model(::WarcraftBenchmark)
-    Random.seed!(67)
+function Utils.generate_statistical_model(::WarcraftBenchmark; seed=0)
+    Random.seed!(seed)
     resnet18 = ResNet(18; pretrain=false, nclasses=1)
     model_embedding = Chain(
         resnet18.layers[1][1][1],

test/warcraft.jl

@@ -26,6 +26,8 @@
 
         y_bellman = bellman_maximizer(θ)
         y_dijkstra = dijkstra_maximizer(θ)
+        @test size(y_bellman) == size(y_true)
+        @test size(y_dijkstra) == size(y_true)
         @test objective_value(b, θ_true, y_bellman) ==
             objective_value(b, θ_true, y_dijkstra)