Skip to content

Commit

Permalink
black
Browse files Browse the repository at this point in the history
  • Loading branch information
josephdviviano committed Feb 24, 2024
1 parent 6811a82 commit 45d9893
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
1 change: 1 addition & 0 deletions src/gfn/gflownet/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ class GFlowNet(ABC, nn.Module, Generic[TrainingSampleType]):
A formal definition of GFlowNets is given in Sec. 3 of [GFlowNet Foundations](https://arxiv.org/pdf/2111.09266).
"""

log_reward_clip_min = float("-inf") # Default off.

@abstractmethod
Expand Down
4 changes: 3 additions & 1 deletion testing/test_environments.py
Original file line number Diff line number Diff line change
Expand Up @@ -209,7 +209,9 @@ def test_box_fwd_step(delta: float):
]

for failing_actions_list in failing_actions_lists_at_s0:
actions = env.actions_from_tensor(format_tensor(failing_actions_list, discrete=False))
actions = env.actions_from_tensor(
format_tensor(failing_actions_list, discrete=False)
)
with pytest.raises(NonValidActionsError):
states = env._step(states, actions)

Expand Down

0 comments on commit 45d9893

Please sign in to comment.