standardize module docstring

pina/callback/__init__.py

@@ -1,3 +1,5 @@
+"""Module for the Pina Callbacks."""
+
 __all__ = [
     "SwitchOptimizer",
     "R3Refinement",

pina/callback/adaptive_refinement_callback.py

@@ -1,4 +1,4 @@
-"""PINA Callbacks Implementations"""
+"""Module for the R3Refinement callback."""
 
 import importlib.metadata
 import torch

pina/callback/linear_weight_update_callback.py

@@ -1,4 +1,4 @@
-"""PINA Callbacks Implementations"""
+"""Module for the LinearWeightUpdate callback."""
 
 import warnings
 from lightning.pytorch.callbacks import Callback

pina/callback/optimizer_callback.py

@@ -1,4 +1,4 @@
-"""PINA Callbacks Implementations"""
+"""Module for the SwitchOptimizer callback."""
 
 from lightning.pytorch.callbacks import Callback
 from ..optim import TorchOptimizer

pina/callback/processing_callback.py

@@ -1,4 +1,4 @@
-"""PINA Callbacks Implementations"""
+"""Module for the Processing Callbacks."""
 
 import copy
 import torch