Skip to content

Commit

Permalink
Fix doc of ToTensorImage transforms.py
Browse files Browse the repository at this point in the history
  • Loading branch information
skandermoalla authored Jan 21, 2024
1 parent a10cdbf commit 26f490a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchrl/envs/transforms/transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -1163,7 +1163,7 @@ class ToTensorImage(ObservationTransform):
from_int (bool, optional): if ``True``, the tensor will be scaled from
the range [0, 255] to the range [0.0, 1.0]. if `False``, the tensor
will not be scaled. if `None`, the tensor will be scaled if
it's a floating-point tensor. default=None.
it's not a floating-point tensor. default=None.
unsqueeze (bool): if ``True``, the observation tensor is unsqueezed
along the first dimension. default=False.
dtype (torch.dtype, optional): dtype to use for the resulting
Expand Down

0 comments on commit 26f490a

Please sign in to comment.