fix doc

awslabs · Jun 16, 2024 · c6a195c · c6a195c
1 parent fcccea0
commit c6a195c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gluonts/torch/model/samformer/lightning_module.py b/src/gluonts/torch/model/samformer/lightning_module.py
@@ -23,8 +23,8 @@
 
 class SamFormerLightningModule(pl.LightningModule):
     """
-    A ``pl.LightningModule`` class that can be used to train a
-    ``SamFormerModel`` with PyTorch Lightning.
+    A ``pl.LightningModule`` class that can be used to train a ``SamFormerModel`` with
+    PyTorch Lightning.
 
     This is a thin layer around a (wrapped) ``SamFormerModel`` object,
     that exposes the methods to evaluate training and validation loss.