Skip to content

Commit 828be9c

Browse files
authored
[Doc] _make_collector helper function (#678)
1 parent d918200 commit 828be9c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchrl/trainers/helpers/collectors.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,7 @@ def _make_collector(
175175
num_env_per_collector: Optional[int] = None,
176176
num_collectors: Optional[int] = None,
177177
**kwargs,
178-
) -> _MultiDataCollector:
178+
) -> _DataCollector:
179179
if env_kwargs is None:
180180
env_kwargs = dict()
181181
if isinstance(env_fns, list):

0 commit comments

Comments
 (0)