Unverified Commit f5ab9a18 authored by Benjamin Lefaudeux's avatar Benjamin Lefaudeux Committed by GitHub
Browse files

[fix] Fixing some docstrings which did not pass CI (#330)

* docfix
parent cae9b638
...@@ -202,7 +202,7 @@ class Pipe(Module): ...@@ -202,7 +202,7 @@ class Pipe(Module):
``'except_last'``, or ``'never'`` (default: ``'except_last'``) ``'except_last'``, or ``'never'`` (default: ``'except_last'``)
deferred_batch_norm (bool): deferred_batch_norm (bool):
whether to use deferred BatchNorm moving statistics (default: whether to use deferred BatchNorm moving statistics (default:
:data:`False`, see :ref:`Deferred Batch Normalization` for more :data:`False`, see :class:`Deferred Batch Normalization <DeferredBatchNorm>` for more
details) details)
Raises: Raises:
...@@ -339,15 +339,13 @@ class Pipe(Module): ...@@ -339,15 +339,13 @@ class Pipe(Module):
raise MOVING_DENIED raise MOVING_DENIED
def to(self, *args: Any, **kwargs: Any) -> "Pipe": def to(self, *args: Any, **kwargs: Any) -> "Pipe":
# Deny these usages: """ Deny these usages:
# - to(device[, dtype, non_blocking])
# - to(device[, dtype, non_blocking]) - to(tensor[, non_blocking])
# - to(tensor[, non_blocking])
# But allow this:
# But allow this: - to(dtype[, non_blocking])"""
#
# - to(dtype[, non_blocking])
#
if "device" in kwargs or "tensor" in kwargs: if "device" in kwargs or "tensor" in kwargs:
raise MOVING_DENIED raise MOVING_DENIED
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment