rename method

GFNOrg · Dec 8, 2023 · 8d8a4c1 · 8d8a4c1
1 parent 364e52d
commit 8d8a4c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]]))