From 8d8a4c14b568968954103a44bed43543c2806a0a Mon Sep 17 00:00:00 2001 From: Joseph Viviano Date: Fri, 8 Dec 2023 11:40:23 -0500 Subject: [PATCH] rename method --- src/gfn/gym/helpers/test_box_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/gfn/gym/helpers/test_box_utils.py b/src/gfn/gym/helpers/test_box_utils.py index fb004140..e0ac56dc 100644 --- a/src/gfn/gym/helpers/test_box_utils.py +++ b/src/gfn/gym/helpers/test_box_utils.py @@ -30,7 +30,7 @@ def test_mixed_distributions(n_components: int, n_components_s0: int): R2=2.0, device_str="cpu", ) - States = environment.make_States_class() + States = environment.make_states_class() # Three cases: when all states are s0, some are s0, and none are s0. centers_mixed = States(torch.FloatTensor([[0.03, 0.06], [0.0, 0.0], [0.0, 0.0]]))