From 6da67bd65203146da74cb1979729e4532273645f Mon Sep 17 00:00:00 2001 From: alexhernandezgarcia Date: Thu, 6 Jun 2024 08:38:42 -0400 Subject: [PATCH] Fix evaluator test after adding new figure --- tests/gflownet/evaluator/test_base.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/gflownet/evaluator/test_base.py b/tests/gflownet/evaluator/test_base.py index 85f5f81b5..f9616aa82 100644 --- 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}")