Skip to content

Commit

Permalink
Evaluator tests: fix issue with replay buffer path
Browse files Browse the repository at this point in the history
  • Loading branch information
alexhernandezgarcia committed Jun 11, 2024
1 parent 0e1bdef commit 7b11c38
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/gflownet/evaluator/test_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -221,7 +221,7 @@ def test__should_eval_top_k(constant_evaluator, period, step, target, first_it,
],
)
def test__eval(gflownet_for_tests, parameterization):
assert Path("./replay.pkl").exists()
assert gflownet_for_tests.buffer.replay_pkl.exists()
# results: {"metrics": dict[str, float], "figs": list[plt.Figure]}
results = gflownet_for_tests.evaluator.eval()
figs = gflownet_for_tests.evaluator.plot(**results["data"])
Expand Down

0 comments on commit 7b11c38

Please sign in to comment.