standardize module docstring

This commit is contained in:
giovanni
2025-03-14 16:32:19 +01:00
committed by FilippoOlivo
parent 8cfee0d1bc
commit 8dfc9d19db
60 changed files with 61 additions and 122 deletions

View File

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

View File

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

View File

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

View File

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

View File

@@ -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

View File

@@ -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

View File

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

View File

@@ -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

View File

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

View File

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

View File

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

View File

@@ -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

View File

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

View File

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

View File

@@ -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

View File

@@ -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

View File

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

View File

@@ -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

View File

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