fix plots

BatyLeo · BatyLeo · commit ab7f421be5aa · 2024-09-17T16:53:10.000+02:00
diff --git a/docs/src/tutorials/warcraft.jl b/docs/src/tutorials/warcraft.jl
@@ -85,4 +85,4 @@ final_gap = compute_gap(b, test_dataset, model, maximizer)
 #
 θ = model(x)
 y = maximizer(θ)
-plot_data(b, DataSample(; x, θ, y))
+plot_data(b, DataSample(; x, θ, y); θ_true=θ_true)
diff --git a/src/Warcraft/Warcraft.jl b/src/Warcraft/Warcraft.jl
@@ -29,7 +29,7 @@ Plot the image `im`, the weights `weights`, and the path `path` on the same Figu
 """
 function Utils.plot_data(
     ::WarcraftBenchmark,
-    sample::DataSample,
+    sample::DataSample;
     θ_title="Weights",
     y_title="Path",
     θ_true=sample.θ,

Original file line number	Diff line number	Diff line change
`@@ -85,4 +85,4 @@ final_gap = compute_gap(b, test_dataset, model, maximizer)`
`85`	`85`	`#`
`86`	`86`	`θ = model(x)`
`87`	`87`	`y = maximizer(θ)`
`88`		`-plot_data(b, DataSample(; x, θ, y))`
	`88`	`+plot_data(b, DataSample(; x, θ, y); θ_true=θ_true)`