Fix evaluator test after adding new figure

alexhernandezgarcia · Jun 6, 2024 · 6da67bd · 6da67bd
1 parent 8e582a3
commit 6da67bd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/gflownet/evaluator/test_base.py b/tests/gflownet/evaluator/test_base.py
@@ -237,6 +237,9 @@ def test__eval(gflownet_for_tests, parameterization):
     elif parameterization == "ctorus":
         for figname, fig in figs.items():
             assert isinstance(figname, str)
+            # plot_samples_topk not implemented in ctorus
+            if figname == "Samples TopK":
+                continue
             assert isinstance(fig, plt.Figure)
     else:
         raise ValueError(f"Unknown parameterization: {parameterization}")