black

tests/gflownet/envs/common.py

@@ -66,7 +66,6 @@ def test__step__returns_same_state_action_and_invalid_if_done(self, n_repeat=1):
     def test__sample_actions__backward__returns_eos_if_done(
         self, n_repeat=1, n_states=5
     ):
-
         if _get_current_method_name() in self.n_states:
             n_states = self.n_states[_get_current_method_name()]
 
@@ -96,7 +95,6 @@ def test__sample_actions__backward__returns_eos_if_done(
     def test__get_logprobs__backward__returns_zero_if_done(
         self, n_repeat=1, n_states=5
     ):
-
         if _get_current_method_name() in self.n_states:
             n_states = self.n_states[_get_current_method_name()]
 
@@ -161,7 +159,6 @@ def test__forward_actions_have_nonzero_backward_prob(self, n_repeat=1):
     def test__backward_actions_have_nonzero_forward_prob(
         self, n_repeat=1, n_states=100
     ):
-
         if _get_current_method_name() in self.n_states:
             n_states = self.n_states[_get_current_method_name()]
 
@@ -398,7 +395,6 @@ def test__state2readable__is_reversible(self, n_repeat=1):
     def test__get_parents__returns_same_state_and_eos_if_done(
         self, n_repeat=1, n_states=10
     ):
-
         if _get_current_method_name() in self.n_states:
             n_states = self.n_states[_get_current_method_name()]