standardize module docstring

pina/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the Pina library."""
-Module for the Pina library.
-"""
 
 __all__ = [
     "Trainer",

pina/adaptive_function/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Adaptive Activation Functions Module."""
-Adaptive Activation Functions Module.
-"""
 
 __all__ = [
     "AdaptiveActivationFunctionInterface",

pina/adaptive_function/adaptive_function.py

@@ -1,4 +1,4 @@
-"""Module for adaptive functions."""
+"""Module for the Adaptive Functions."""
 
 import torch
 from ..utils import check_consistency

pina/adaptive_function/adaptive_function_interface.py

@@ -1,4 +1,4 @@
-"""Module for adaptive functions."""
+"""Module for the Adaptive Function interface."""
 
 from abc import ABCMeta
 import torch

pina/adaptive_functions/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Old module for adaptive functions. Deprecated in 0.2.0."""
-Old module for adaptive functions. Deprecated in 0.2.0.
-"""
 
 import warnings
 

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

pina/callbacks/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Old module for callbacks. Deprecated in 0.2.0."""
-Old module for callbacks. Deprecated in 0.2.0.
-"""
 
 import warnings
 

pina/collector.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the Collector class."""
-Module for the Collector class.
-"""
 
 from .graph import Graph
 from .utils import check_consistency

pina/condition/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Module for PINA Conditions classes."""
-Module for importing Conditions objects.
-"""
 
 __all__ = [
     "Condition",

pina/condition/condition.py

@@ -1,4 +1,4 @@
-"""Condition module."""
+"""Module for the Condition class."""
 
 import warnings
 from .data_condition import DataCondition

pina/condition/condition_interface.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the Condition interface."""
-Module that defines the ConditionInterface class.
-"""
 
 from abc import ABCMeta
 from torch_geometric.data import Data

pina/condition/data_condition.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the DataCondition class."""
-DataCondition class
-"""
 
 import torch
 from torch_geometric.data import Data

pina/condition/domain_equation_condition.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the DomainEquationCondition class."""
-DomainEquationCondition class definition.
-"""
 
 from .condition_interface import ConditionInterface
 from ..utils import check_consistency

pina/condition/input_equation_condition.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the InputEquationCondition class and its subclasses."""
-Module to define InputEquationCondition class and its subclasses.
-"""
 
 from torch_geometric.data import Data
 from .condition_interface import ConditionInterface

pina/data/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Module for data, data module, and dataset."""
-Module for data data module and dataset.
-"""
 
 __all__ = ["PinaDataModule", "PinaDataset"]
 

pina/data/dataset.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the PINA dataset classes."""
-Module for the PINA dataset
-"""
 
 from abc import abstractmethod, ABC
 from torch.utils.data import Dataset

pina/equation/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Module to define equations and systems of equations."""
-Module to define equations and systems of equations.
-"""
 
 __all__ = [
     "SystemEquation",

pina/equation/equation_interface.py

@@ -1,4 +1,4 @@
-"""Module for the Equation Interface"""
+"""Module for the Equation Interface."""
 
 from abc import ABCMeta, abstractmethod
 

pina/geometry/__init__.py

@@ -1,7 +1,4 @@
-"""
+"""Old module for geometry classes and functions. Deprecated in 0.2.0."""
-Old module for geometry related classes and functions.
-Deprecated in 0.2.0.
-"""
 
 import warnings
 

pina/graph.py

@@ -1,6 +1,4 @@
-"""
+"""Module to build Graph objects and perform operations on them."""
-Module to build Graph objects and perform operations on them.
-"""
 
 import torch
 from torch_geometric.data import Data, Batch

pina/loss/loss_interface.py

@@ -1,4 +1,4 @@
-"""Module for the Loss Interface"""
+"""Module for the Loss Interface."""
 
 from abc import ABCMeta, abstractmethod
 from torch.nn.modules.loss import _Loss

pina/loss/lp_loss.py

@@ -1,4 +1,4 @@
-"""Module for LpLoss class"""
+"""Module for the LpLoss class."""
 
 import torch
 

pina/loss/power_loss.py

@@ -1,4 +1,4 @@
-"""Module for PowerLoss class"""
+"""Module for the PowerLoss class."""
 
 import torch
 

pina/loss/weighting_interface.py

@@ -1,4 +1,4 @@
-"""Module for the Weighting Interface"""
+"""Module for the Weighting Interface."""
 
 from abc import ABCMeta, abstractmethod
 

pina/model/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the Neural model classes."""
-Module for the Neural model classes.
-"""
 
 __all__ = [
     "FeedForward",

pina/model/block/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the building blocks of the neural models."""
-Module for the building blocks of the neural models.
-"""
 
 __all__ = [
     "ContinuousConvBlock",

pina/model/block/convolution_2d.py

@@ -1,4 +1,4 @@
-"""Module for Continuous Convolution class"""
+"""Module for the Continuous Convolution class."""
 
 import torch
 from .convolution import BaseContinuousConv

pina/model/block/embedding.py

@@ -1,4 +1,4 @@
-"""Modules for the Embedding blocks."""
+"""Modules for the the Embedding blocks."""
 
 import torch
 from pina.utils import check_consistency

pina/model/block/fourier_block.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the Fourier Neural Operator Block class."""
-Module for the Fourier Neural Operator Block class.
-"""
 
 import torch
 from torch import nn

pina/model/block/gno_block.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the Graph Neural Operator Block class."""
-Module for the Graph Neural Operator Block class.
-"""
 
 import torch
 from torch_geometric.nn import MessagePassing

pina/model/block/integral.py

@@ -1,6 +1,4 @@
-"""
+"""Module to perform integration for continuous convolution."""
-Module to perform integration for continuous convolution.
-"""
 
 import torch
 

pina/model/block/residual.py

@@ -1,6 +1,4 @@
-"""
+"""Module for residual blocks and enhanced linear layers."""
-Module for residual blocks and enhanced linear layers.
-"""
 
 import torch
 from torch import nn

pina/model/block/spectral.py

@@ -1,6 +1,4 @@
-"""
+"""Module for spectral convolution blocks."""
-Module for spectral convolution blocks.
-"""
 
 import torch
 from torch import nn

pina/model/block/stride.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the Stride class."""
-Module for the Stride class.
-"""
 
 import torch
 

pina/model/block/utils_convolution.py

@@ -1,6 +1,4 @@
-"""
+"""Module for utility functions for the convolutional layer."""
-Module for utility functions for the convolutional layer.
-"""
 
 import torch
 

pina/model/deeponet.py

@@ -1,4 +1,4 @@
-"""Module for the DeepONet and MIONet model classes"""
+"""Module for the DeepONet and MIONet model classes."""
 
 from functools import partial
 import torch

pina/model/feed_forward.py

@@ -1,4 +1,4 @@
-"""Module for the Feed Forward model class"""
+"""Module for the Feed Forward model class."""
 
 import torch
 from torch import nn

pina/model/fourier_neural_operator.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the Fourier Neural Operator model class."""
-Module for the Fourier Neural Operator model class.
-"""
 
 import warnings
 import torch

pina/model/graph_neural_operator.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the Graph Neural Operator model class."""
-Module for the Graph Neural Operator model class.
-"""
 
 import torch
 from torch.nn import Tanh

pina/model/kernel_neural_operator.py

@@ -1,6 +1,4 @@
-"""
+"""Module for the Kernel Neural Operator model class."""
-Module for the Kernel Neural Operator model class.
-"""
 
 import torch
 from ..utils import check_consistency

pina/model/layers/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Old layers module, deprecated in 0.2.0."""
-Old layers module, deprecated in 0.2.0.
-"""
 
 import warnings
 

pina/model/multi_feed_forward.py

@@ -1,4 +1,4 @@
-"""Module for the Multi Feed Forward model class"""
+"""Module for the Multi Feed Forward model class."""
 
 from abc import ABC, abstractmethod
 import torch

pina/model/spline.py

@@ -1,4 +1,4 @@
-"""Module for the Spline model class"""
+"""Module for the Spline model class."""
 
 import torch
 from ..utils import check_consistency

pina/operators.py

@@ -1,6 +1,4 @@
-"""
+"""Old module for operators. Deprecated in 0.2.0."""
-Old module for operators. Deprecated in 0.2.0.
-"""
 
 import warnings
 

pina/problem/abstract_problem.py

@@ -1,4 +1,4 @@
-"""Module for the AbstractProblem class"""
+"""Module for the AbstractProblem class."""
 
 from abc import ABCMeta, abstractmethod
 from copy import deepcopy

pina/problem/inverse_problem.py

@@ -1,4 +1,4 @@
-"""Module for the InverseProblem class"""
+"""Module for the InverseProblem class."""
 
 from abc import abstractmethod
 import torch

pina/problem/parametric_problem.py

@@ -1,4 +1,4 @@
-"""Module for the ParametricProblem class"""
+"""Module for the ParametricProblem class."""
 
 from abc import abstractmethod
 

pina/problem/spatial_problem.py

@@ -1,4 +1,4 @@
-"""Module for the SpatialProblem class"""
+"""Module for the SpatialProblem class."""
 
 from abc import abstractmethod
 

pina/problem/time_dependent_problem.py

@@ -1,4 +1,4 @@
-"""Module for the TimeDependentProblem class"""
+"""Module for the TimeDependentProblem class."""
 
 from abc import abstractmethod
 

pina/solver/physic_informed_solver/__init__.py

@@ -1,4 +1,4 @@
-"""Module for the physics-informed solvers."""
+"""Module for the Physics-Informed solvers."""
 
 __all__ = [
     "PINNInterface",

pina/solver/physic_informed_solver/pinn_interface.py

@@ -1,4 +1,4 @@
-"""Module for the Physics Informed Neural Network Interface."""
+"""Module for the Physics-Informed Neural Network Interface."""
 
 from abc import ABCMeta, abstractmethod
 import torch

pina/solver/physic_informed_solver/self_adaptive_pinn.py

@@ -1,4 +1,4 @@
-"""Module for Self-Adaptive PINN."""
+"""Module for the Self-Adaptive PINN solver."""
 
 from copy import deepcopy
 import torch

pina/solver/reduced_order_model.py

@@ -1,4 +1,4 @@
-"""Module for ReducedOrderModelSolver"""
+"""Module for the Reduced Order Model solver"""
 
 import torch
 from .supervised import SupervisedSolver

pina/solver/supervised.py

@@ -1,4 +1,4 @@
-"""Module for the Supervised Solver."""
+"""Module for the Supervised solver."""
 
 import torch
 from torch.nn.modules.loss import _Loss

pina/solvers/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Old module for solvers. Deprecated in 0.2.0 ."""
-Old module for solvers. Deprecated in 0.2.0 .
-"""
 
 import warnings
 

pina/solvers/pinns/__init__.py

@@ -1,6 +1,4 @@
-"""
+"""Old module for the PINNs solver. Deprecated in 0.2.0."""
-Old module for the PINNs solver. Deprecated in 0.2.0.
-"""
 
 import warnings