Update training.py

Signed-off-by: Tsz Wai Ko <[email protected]>

src/matgl/utils/training.py

@@ -137,7 +137,7 @@ def predict_step(self, batch, batch_idx, dataloader_idx=0):
 
 
 class ModelLightningModule(MatglLightningModuleMixin, pl.LightningModule):
-    """A PyTorch.LightningModule for training MEGNet and M3GNet models."""
+    """A PyTorch.LightningModule for training MatGL structure-wise property models."""
 
     def __init__(
         self,