Skip to content

Commit fac4b3e

Browse files
authored
[Doc] WeightUpdaterBase docs update after renaming (#3007)
1 parent 43d5333 commit fac4b3e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchrl/collectors/weight_update.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ class WeightUpdaterBase(metaclass=abc.ABCMeta):
4747
4848
Attributes:
4949
collector: The collector (or any container) of the weight receiver. The collector is registered via
50-
:meth:`~torchrl.collectors.WeightUpdateReceiverBase.register_collector`.
50+
:meth:`~torchrl.collectors.WeightUpdaterBase.register_collector`.
5151
5252
Methods:
5353
push_weights: Updates the weights on specified or all remote workers.
@@ -317,7 +317,7 @@ class VanillaWeightUpdater(WeightUpdaterBase):
317317
This class is used by default in the `SyncDataCollector` when no custom weight sender
318318
is provided.
319319
320-
.. seealso:: :class:`~torchrl.collectors.WeightUpdateReceiverBase` and :class:`~torchrl.collectors.SyncDataCollector`.
320+
.. seealso:: :class:`~torchrl.collectors.WeightUpdaterBase` and :class:`~torchrl.collectors.SyncDataCollector`.
321321
322322
Keyword Args:
323323
weight_getter (Callable[[], TensorDictBase], optional): a callable that returns the weights from the server.

0 commit comments

Comments
 (0)