Fix a docstring

torchmd · May 22, 2024 · e224dc5 · e224dc5
1 parent a6a7efb
commit e224dc5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmdnet/models/utils.py b/torchmdnet/models/utils.py
@@ -436,7 +436,7 @@ def forward(self, distances: Tensor) -> Tensor:
 
 
 class MLP(nn.Module):
-    """A simple multi-layer perceptron with a given number of layers and hidden channels.
+    r"""A simple multi-layer perceptron with a given number of layers and hidden channels.
 
     The simplest MLP has no hidden layers and is composed of two linear layers with a non-linear activation function in between: