Skip to content

Commit 3e99824

Browse files
committed
#79 Refactor API
1 parent 8e50ba3 commit 3e99824

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/utils/replay_buffer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,8 +114,8 @@ def get_torch__tensor_info_item_as_torch_tensor(self, name_attr: str) -> torch.
114114

115115
return torch.stack(vals)
116116

117-
def add(self, state: Any, action: Any, reward: float,
118-
next_state: Any, done: bool, info: dict = {}) -> None:
117+
def add(self, state: Any, action: Any, reward: Any,
118+
next_state: Any, done: Any, info: dict = {}) -> None:
119119
"""Add a new experience tuple in the buffer
120120
121121
Parameters

0 commit comments

Comments
 (0)