From ad8641a52c9b2c4f20fb7b4af5072b8e29c90b8b Mon Sep 17 00:00:00 2001 From: Adam Dendek Date: Tue, 20 Jan 2026 15:47:01 +0100 Subject: [PATCH 1/4] Add error suppression logic fot test --- pina/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pina/__init__.py b/pina/__init__.py index 0d38804fe..cfcb28833 100644 --- a/pina/__init__.py +++ b/pina/__init__.py @@ -22,3 +22,7 @@ from pina._src.core.trainer import Trainer from pina._src.condition.condition import Condition from pina._src.data.data_module import PinaDataModule + +# suppress errors +import torch._dynamo +torch._dynamo.config.suppress_errors = True From 3bb8a91f884be5d8a18b8058ae48b1a7e85e3122 Mon Sep 17 00:00:00 2001 From: Adam Dendek Date: Tue, 20 Jan 2026 15:53:40 +0100 Subject: [PATCH 2/4] Fix missing imports --- pina/__init__.py | 4 --- .../condition/domain_equation_condition.py | 2 +- pina/_src/data/data_module.py | 2 +- pina/_src/model/average_neural_operator.py | 6 ++--- pina/_src/model/deeponet.py | 2 +- .../equivariant_graph_neural_operator.py | 6 +++-- pina/_src/model/feed_forward.py | 4 +-- pina/_src/model/fourier_neural_operator.py | 12 ++++++--- pina/_src/model/graph_neural_operator.py | 4 +-- pina/_src/model/kernel_neural_operator.py | 2 +- pina/_src/model/low_rank_neural_operator.py | 6 ++--- pina/_src/model/multi_feed_forward.py | 2 +- pina/_src/model/pirate_network.py | 5 ++-- pina/_src/model/sindy.py | 2 +- pina/_src/model/spline.py | 2 +- pina/_src/model/spline_surface.py | 4 +-- pina/_src/problem/abstract_problem.py | 13 ++++++---- .../solver/ensemble_solver/ensemble_pinn.py | 6 +++-- .../physics_informed_solver/pinn_interface.py | 4 ++- .../_src/solver/supervised_solver/__init__.py | 11 -------- pina/data/__init__.py | 26 ++++++++++++++++--- pina/solver/__init__.py | 7 +++-- pina/type_checker.py | 11 ++++++++ pina/utils.py | 19 ++++++++++++++ tests/test_block/test_pod.py | 2 +- tests/test_block/test_rbf.py | 2 +- tests/test_condition.py | 2 +- tests/test_data/test_data_module.py | 4 +-- tests/test_data/test_graph_dataset.py | 2 +- tests/test_data/test_tensor_dataset.py | 2 +- tests/test_loss/test_power_loss.py | 1 - .../test_supervised_problem.py | 2 +- 32 files changed, 115 insertions(+), 64 deletions(-) create mode 100644 pina/type_checker.py create mode 100644 pina/utils.py diff --git a/pina/__init__.py b/pina/__init__.py index cfcb28833..0d38804fe 100644 --- a/pina/__init__.py +++ b/pina/__init__.py @@ -22,7 +22,3 @@ from pina._src.core.trainer import Trainer from pina._src.condition.condition import Condition from pina._src.data.data_module import PinaDataModule - -# suppress errors -import torch._dynamo -torch._dynamo.config.suppress_errors = True diff --git a/pina/_src/condition/domain_equation_condition.py b/pina/_src/condition/domain_equation_condition.py index 93e76892a..0b75269ce 100644 --- a/pina/_src/condition/domain_equation_condition.py +++ b/pina/_src/condition/domain_equation_condition.py @@ -2,7 +2,7 @@ from pina._src.condition.condition_interface import ConditionInterface from pina._src.core.utils import check_consistency -from pina._src.domain import DomainInterface +from pina._src.domain.domain_interface import DomainInterface from pina._src.equation.equation_interface import EquationInterface diff --git a/pina/_src/data/data_module.py b/pina/_src/data/data_module.py index 664d200f8..f45236f0f 100644 --- a/pina/_src/data/data_module.py +++ b/pina/_src/data/data_module.py @@ -8,7 +8,7 @@ from lightning.pytorch import LightningDataModule import torch from torch_geometric.data import Data -from torch.utils.data import DataLoader, SequentialSampler, RandomSampler +from torch.utils.data import DataLoader, SequentialSampler from torch.utils.data.distributed import DistributedSampler from pina._src.core.label_tensor import LabelTensor from pina._src.data.dataset import PinaDatasetFactory, PinaTensorDataset diff --git a/pina/_src/model/average_neural_operator.py b/pina/_src/model/average_neural_operator.py index 6019b96c6..e16e3430f 100644 --- a/pina/_src/model/average_neural_operator.py +++ b/pina/_src/model/average_neural_operator.py @@ -2,9 +2,9 @@ import torch from torch import nn -from .block.average_neural_operator_block import AVNOBlock -from .kernel_neural_operator import KernelNeuralOperator -from ..utils import check_consistency +from pina._src.model.block.average_neural_operator_block import AVNOBlock +from pina._src.model.kernel_neural_operator import KernelNeuralOperator +from pina._src.core.utils import check_consistency class AveragingNeuralOperator(KernelNeuralOperator): diff --git a/pina/_src/model/deeponet.py b/pina/_src/model/deeponet.py index c65f6b316..800f2acc3 100644 --- a/pina/_src/model/deeponet.py +++ b/pina/_src/model/deeponet.py @@ -3,7 +3,7 @@ from functools import partial import torch from torch import nn -from ..utils import check_consistency, is_function +from pina._src.core.utils import check_consistency, is_function class MIONet(torch.nn.Module): diff --git a/pina/_src/model/equivariant_graph_neural_operator.py b/pina/_src/model/equivariant_graph_neural_operator.py index 6b33df6db..3aa7dde69 100644 --- a/pina/_src/model/equivariant_graph_neural_operator.py +++ b/pina/_src/model/equivariant_graph_neural_operator.py @@ -1,8 +1,10 @@ """Module for the Equivariant Graph Neural Operator model.""" import torch -from ..utils import check_positive_integer -from .block.message_passing import EquivariantGraphNeuralOperatorBlock +from pina._src.core.utils import check_positive_integer +from pina._src.model.block.message_passing.equivariant_graph_neural_operator_block import ( + EquivariantGraphNeuralOperatorBlock, +) class EquivariantGraphNeuralOperator(torch.nn.Module): diff --git a/pina/_src/model/feed_forward.py b/pina/_src/model/feed_forward.py index a1651b38b..fdf6bc91e 100644 --- a/pina/_src/model/feed_forward.py +++ b/pina/_src/model/feed_forward.py @@ -2,8 +2,8 @@ import torch from torch import nn -from ..utils import check_consistency -from .block.residual import EnhancedLinear +from pina._src.core.utils import check_consistency +from pina._src.model.block.residual import EnhancedLinear class FeedForward(torch.nn.Module): diff --git a/pina/_src/model/fourier_neural_operator.py b/pina/_src/model/fourier_neural_operator.py index e1336c999..7517b39b4 100644 --- a/pina/_src/model/fourier_neural_operator.py +++ b/pina/_src/model/fourier_neural_operator.py @@ -3,10 +3,14 @@ import warnings import torch from torch import nn -from ..label_tensor import LabelTensor -from ..utils import check_consistency -from .block.fourier_block import FourierBlock1D, FourierBlock2D, FourierBlock3D -from .kernel_neural_operator import KernelNeuralOperator +from pina._src.core.label_tensor import LabelTensor +from pina._src.core.utils import check_consistency +from pina._src.model.block.fourier_block import ( + FourierBlock1D, + FourierBlock2D, + FourierBlock3D, +) +from pina._src.model.kernel_neural_operator import KernelNeuralOperator class FourierIntegralKernel(torch.nn.Module): diff --git a/pina/_src/model/graph_neural_operator.py b/pina/_src/model/graph_neural_operator.py index 3cb5cdd31..e4d844fcb 100644 --- a/pina/_src/model/graph_neural_operator.py +++ b/pina/_src/model/graph_neural_operator.py @@ -2,8 +2,8 @@ import torch from torch.nn import Tanh -from .block.gno_block import GNOBlock -from .kernel_neural_operator import KernelNeuralOperator +from pina._src.model.block.gno_block import GNOBlock +from pina._src.model.kernel_neural_operator import KernelNeuralOperator class GraphNeuralKernel(torch.nn.Module): diff --git a/pina/_src/model/kernel_neural_operator.py b/pina/_src/model/kernel_neural_operator.py index e3cb790e5..81d1be45d 100644 --- a/pina/_src/model/kernel_neural_operator.py +++ b/pina/_src/model/kernel_neural_operator.py @@ -1,7 +1,7 @@ """Module for the Kernel Neural Operator model class.""" import torch -from ..utils import check_consistency +from pina._src.core.utils import check_consistency class KernelNeuralOperator(torch.nn.Module): diff --git a/pina/_src/model/low_rank_neural_operator.py b/pina/_src/model/low_rank_neural_operator.py index 1a7082dff..049894001 100644 --- a/pina/_src/model/low_rank_neural_operator.py +++ b/pina/_src/model/low_rank_neural_operator.py @@ -3,10 +3,10 @@ import torch from torch import nn -from ..utils import check_consistency +from pina._src.core.utils import check_consistency -from .kernel_neural_operator import KernelNeuralOperator -from .block.low_rank_block import LowRankBlock +from pina._src.model.kernel_neural_operator import KernelNeuralOperator +from pina._src.model.block.low_rank_block import LowRankBlock class LowRankNeuralOperator(KernelNeuralOperator): diff --git a/pina/_src/model/multi_feed_forward.py b/pina/_src/model/multi_feed_forward.py index f2f149ca6..df8fb19e2 100644 --- a/pina/_src/model/multi_feed_forward.py +++ b/pina/_src/model/multi_feed_forward.py @@ -2,7 +2,7 @@ from abc import ABC, abstractmethod import torch -from .feed_forward import FeedForward +from pina._src.model.feed_forward import FeedForward class MultiFeedForward(torch.nn.Module, ABC): diff --git a/pina/_src/model/pirate_network.py b/pina/_src/model/pirate_network.py index 96102b41f..09aad269d 100644 --- a/pina/_src/model/pirate_network.py +++ b/pina/_src/model/pirate_network.py @@ -1,8 +1,9 @@ """Module for the PirateNet model class.""" import torch -from .block import FourierFeatureEmbedding, PirateNetBlock -from ..utils import check_consistency, check_positive_integer +from pina._src.model.block.embedding import FourierFeatureEmbedding +from pina._src.model.block.pirate_network_block import PirateNetBlock +from pina._src.core.utils import check_consistency, check_positive_integer class PirateNet(torch.nn.Module): diff --git a/pina/_src/model/sindy.py b/pina/_src/model/sindy.py index a40fa37b4..f69842a54 100644 --- a/pina/_src/model/sindy.py +++ b/pina/_src/model/sindy.py @@ -2,7 +2,7 @@ from typing import Callable import torch -from ..utils import check_consistency, check_positive_integer +from pina._src.core.utils import check_consistency, check_positive_integer class SINDy(torch.nn.Module): diff --git a/pina/_src/model/spline.py b/pina/_src/model/spline.py index d9141fe8c..5e5b133c3 100644 --- a/pina/_src/model/spline.py +++ b/pina/_src/model/spline.py @@ -2,7 +2,7 @@ import warnings import torch -from ..utils import check_positive_integer, check_consistency +from pina._src.core.utils import check_consistency, check_positive_integer class Spline(torch.nn.Module): diff --git a/pina/_src/model/spline_surface.py b/pina/_src/model/spline_surface.py index 767e5b0dc..d54a0c7bb 100644 --- a/pina/_src/model/spline_surface.py +++ b/pina/_src/model/spline_surface.py @@ -2,8 +2,8 @@ import torch from .spline import Spline -from ..label_tensor import LabelTensor -from ..utils import check_consistency, check_positive_integer +from pina._src.core.label_tensor import LabelTensor +from pina._src.core.utils import check_consistency, check_positive_integer class SplineSurface(torch.nn.Module): diff --git a/pina/_src/problem/abstract_problem.py b/pina/_src/problem/abstract_problem.py index 9280d48d8..381186e00 100644 --- a/pina/_src/problem/abstract_problem.py +++ b/pina/_src/problem/abstract_problem.py @@ -3,11 +3,14 @@ from abc import ABCMeta, abstractmethod import warnings from copy import deepcopy -from ..utils import check_consistency -from ..domain import DomainInterface, CartesianDomain -from ..condition.domain_equation_condition import DomainEquationCondition -from ..label_tensor import LabelTensor -from ..utils import merge_tensors, custom_warning_format +from pina._src.core.utils import check_consistency +from pina._src.domain.domain_interface import DomainInterface +from pina._src.domain.cartesian_domain import CartesianDomain +from pina._src.condition.domain_equation_condition import ( + DomainEquationCondition, +) +from pina._src.core.label_tensor import LabelTensor +from pina._src.core.utils import merge_tensors, custom_warning_format class AbstractProblem(metaclass=ABCMeta): diff --git a/pina/_src/solver/ensemble_solver/ensemble_pinn.py b/pina/_src/solver/ensemble_solver/ensemble_pinn.py index 002d4bd53..f010753ec 100644 --- a/pina/_src/solver/ensemble_solver/ensemble_pinn.py +++ b/pina/_src/solver/ensemble_solver/ensemble_pinn.py @@ -5,8 +5,10 @@ from pina._src.solver.ensemble_solver.ensemble_solver_interface import ( DeepEnsembleSolverInterface, ) -from pina._src.solver.physics_informed_solver import PINNInterface -from pina._src.problem import InverseProblem +from pina._src.solver.physics_informed_solver.pinn_interface import ( + PINNInterface, +) +from pina._src.problem.inverse_problem import InverseProblem class DeepEnsemblePINN(PINNInterface, DeepEnsembleSolverInterface): diff --git a/pina/_src/solver/physics_informed_solver/pinn_interface.py b/pina/_src/solver/physics_informed_solver/pinn_interface.py index 60330372a..517b48082 100644 --- a/pina/_src/solver/physics_informed_solver/pinn_interface.py +++ b/pina/_src/solver/physics_informed_solver/pinn_interface.py @@ -5,7 +5,9 @@ import torch from pina._src.core.utils import custom_warning_format -from pina._src.solver.supervised_solver import SupervisedSolverInterface +from pina._src.solver.supervised_solver.supervised_solver_interface import ( + SupervisedSolverInterface, +) from pina._src.condition.input_target_condition import InputTargetCondition from pina._src.condition.input_equation_condition import InputEquationCondition from pina._src.condition.domain_equation_condition import ( diff --git a/pina/_src/solver/supervised_solver/__init__.py b/pina/_src/solver/supervised_solver/__init__.py index f681d2dd3..e69de29bb 100644 --- a/pina/_src/solver/supervised_solver/__init__.py +++ b/pina/_src/solver/supervised_solver/__init__.py @@ -1,11 +0,0 @@ -"""Module for the Supervised solvers.""" - -__all__ = [ - "SupervisedSolverInterface", - "SupervisedSolver", - "ReducedOrderModelSolver", -] - -from .supervised_solver_interface import SupervisedSolverInterface -from .supervised import SupervisedSolver -from .reduced_order_model import ReducedOrderModelSolver diff --git a/pina/data/__init__.py b/pina/data/__init__.py index 164a6c7aa..2ecebecdd 100644 --- a/pina/data/__init__.py +++ b/pina/data/__init__.py @@ -5,8 +5,28 @@ graph-structured data within the PINA training pipeline. """ -__all__ = ["PinaDataModule", "PinaDataset"] +from pina._src.data.data_module import ( + PinaDataModule, + PinaSampler, + DummyDataloader, + Collator, + PinaSampler, +) +from pina._src.data.dataset import ( + PinaDataset, + PinaTensorDataset, + PinaGraphDataset, + PinaDatasetFactory, +) -from pina._src.data.data_module import PinaDataModule -from pina._src.data.dataset import PinaDataset +__all__ = [ + "PinaDataModule", + "PinaDataset", + "PinaSampler", + "DummyDataloader", + "Collator", + "PinaTensorDataset", + "PinaGraphDataset", + "PinaDatasetFactory", +] diff --git a/pina/solver/__init__.py b/pina/solver/__init__.py index 327056035..d576c47be 100644 --- a/pina/solver/__init__.py +++ b/pina/solver/__init__.py @@ -53,9 +53,12 @@ SupervisedSolver, ReducedOrderModelSolver, ) -from pina._src.solver.ensemble_solver import ( +from pina._src.solver.ensemble_solver.ensemble_solver_interface import ( DeepEnsembleSolverInterface, +) +from pina._src.solver.ensemble_solver.ensemble_pinn import DeepEnsemblePINN +from pina._src.solver.ensemble_solver.ensemble_supervised import ( DeepEnsembleSupervisedSolver, - DeepEnsemblePINN, ) + from pina._src.solver.garom import GAROM diff --git a/pina/type_checker.py b/pina/type_checker.py new file mode 100644 index 000000000..cfa886da3 --- /dev/null +++ b/pina/type_checker.py @@ -0,0 +1,11 @@ +"""Runtime type enforcement and validation utilities. + +This module provides decorators to validate function arguments against type +annotations at runtime. This ensures that PINA components receive inputs +adhering to expected specifications, improving the robustness of the +scientific computing pipeline. +""" + +from pina._src.core.type_checker import enforce_types + +__all__ = ["enforce_types"] diff --git a/pina/utils.py b/pina/utils.py new file mode 100644 index 000000000..8e40ae9cd --- /dev/null +++ b/pina/utils.py @@ -0,0 +1,19 @@ +""" +Utility functions for tensor manipulation and input validation. + +This module provides helper functions to manage tensor operations and ensure +data consistency across the PINA framework, supporting robust input checking +and seamless data merging. +""" + +from pina._src.core.utils import ( + merge_tensors, + check_consistency, + check_positive_integer, +) + +__all__ = [ + "merge_tensors", + "check_consistency", + "check_positive_integer", +] diff --git a/tests/test_block/test_pod.py b/tests/test_block/test_pod.py index d10625fc3..ffe9a8a81 100644 --- a/tests/test_block/test_pod.py +++ b/tests/test_block/test_pod.py @@ -1,7 +1,7 @@ import torch import pytest -from pina.model.block.pod_block import PODBlock +from pina.model.block import PODBlock x = torch.linspace(-1, 1, 100) toy_snapshots = torch.vstack( diff --git a/tests/test_block/test_rbf.py b/tests/test_block/test_rbf.py index 65912fb76..c36836743 100644 --- a/tests/test_block/test_rbf.py +++ b/tests/test_block/test_rbf.py @@ -2,7 +2,7 @@ import pytest import math -from pina.model.block.rbf_block import RBFBlock +from pina.model.block import RBFBlock x = torch.linspace(-1, 1, 100) toy_params = torch.linspace(0, 1, 10).unsqueeze(1) diff --git a/tests/test_condition.py b/tests/test_condition.py index 9199f2bd9..266233179 100644 --- a/tests/test_condition.py +++ b/tests/test_condition.py @@ -18,7 +18,7 @@ GraphDataCondition, ) from pina.domain import CartesianDomain -from pina.equation.equation_factory import FixedValue +from pina.equation import FixedValue from pina.graph import RadiusGraph example_domain = CartesianDomain({"x": [0, 1], "y": [0, 1]}) diff --git a/tests/test_data/test_data_module.py b/tests/test_data/test_data_module.py index 53e7334ec..9fd2d36ee 100644 --- a/tests/test_data/test_data_module.py +++ b/tests/test_data/test_data_module.py @@ -1,10 +1,10 @@ import torch import pytest from pina.data import PinaDataModule -from pina.data.dataset import PinaTensorDataset, PinaGraphDataset +from pina.data import PinaTensorDataset, PinaGraphDataset from pina.problem.zoo import SupervisedProblem from pina.graph import RadiusGraph -from pina.data.data_module import DummyDataloader +from pina.data import DummyDataloader from pina import Trainer from pina.solver import SupervisedSolver from torch_geometric.data import Batch diff --git a/tests/test_data/test_graph_dataset.py b/tests/test_data/test_graph_dataset.py index 81d6a2c5d..3a63f7ec6 100644 --- a/tests/test_data/test_graph_dataset.py +++ b/tests/test_data/test_graph_dataset.py @@ -1,6 +1,6 @@ import torch import pytest -from pina.data.dataset import PinaDatasetFactory, PinaGraphDataset +from pina.data import PinaDatasetFactory, PinaGraphDataset from pina.graph import KNNGraph from torch_geometric.data import Data diff --git a/tests/test_data/test_tensor_dataset.py b/tests/test_data/test_tensor_dataset.py index 81a122f2f..9e348c942 100644 --- a/tests/test_data/test_tensor_dataset.py +++ b/tests/test_data/test_tensor_dataset.py @@ -1,6 +1,6 @@ import torch import pytest -from pina.data.dataset import PinaDatasetFactory, PinaTensorDataset +from pina.data import PinaDatasetFactory, PinaTensorDataset input_tensor = torch.rand((100, 10)) output_tensor = torch.rand((100, 2)) diff --git a/tests/test_loss/test_power_loss.py b/tests/test_loss/test_power_loss.py index 4ea90282b..3781f66d3 100644 --- a/tests/test_loss/test_power_loss.py +++ b/tests/test_loss/test_power_loss.py @@ -1,5 +1,4 @@ import torch -import pytest from pina.loss import PowerLoss diff --git a/tests/test_problem_zoo/test_supervised_problem.py b/tests/test_problem_zoo/test_supervised_problem.py index 19b3920ce..da18d6146 100644 --- a/tests/test_problem_zoo/test_supervised_problem.py +++ b/tests/test_problem_zoo/test_supervised_problem.py @@ -1,7 +1,7 @@ import torch from pina.problem import AbstractProblem from pina.condition import InputTargetCondition -from pina.problem.zoo.supervised_problem import SupervisedProblem +from pina.problem.zoo import SupervisedProblem from pina.graph import RadiusGraph From 4f9f12ce4b959c14b5de92188095414066564332 Mon Sep 17 00:00:00 2001 From: Adam Dendek Date: Tue, 20 Jan 2026 15:53:40 +0100 Subject: [PATCH 3/4] Fix missing imports --- pina/__init__.py | 4 --- .../condition/domain_equation_condition.py | 2 +- pina/_src/data/data_module.py | 2 +- pina/_src/model/average_neural_operator.py | 6 ++--- pina/_src/model/deeponet.py | 2 +- .../equivariant_graph_neural_operator.py | 6 +++-- pina/_src/model/feed_forward.py | 4 +-- pina/_src/model/fourier_neural_operator.py | 12 ++++++--- pina/_src/model/graph_neural_operator.py | 4 +-- pina/_src/model/kernel_neural_operator.py | 2 +- pina/_src/model/low_rank_neural_operator.py | 6 ++--- pina/_src/model/multi_feed_forward.py | 2 +- pina/_src/model/pirate_network.py | 5 ++-- pina/_src/model/sindy.py | 2 +- pina/_src/model/spline.py | 2 +- pina/_src/model/spline_surface.py | 4 +-- pina/_src/problem/abstract_problem.py | 13 ++++++---- .../solver/ensemble_solver/ensemble_pinn.py | 6 +++-- .../physics_informed_solver/pinn_interface.py | 4 ++- .../_src/solver/supervised_solver/__init__.py | 11 -------- pina/data/__init__.py | 26 ++++++++++++++++--- pina/solver/__init__.py | 14 +++++++--- pina/type_checker.py | 11 ++++++++ pina/utils.py | 19 ++++++++++++++ tests/test_block/test_pod.py | 2 +- tests/test_block/test_rbf.py | 2 +- tests/test_condition.py | 2 +- tests/test_data/test_data_module.py | 4 +-- tests/test_data/test_graph_dataset.py | 2 +- tests/test_data/test_tensor_dataset.py | 2 +- tests/test_loss/test_power_loss.py | 1 - .../test_supervised_problem.py | 2 +- 32 files changed, 120 insertions(+), 66 deletions(-) create mode 100644 pina/type_checker.py create mode 100644 pina/utils.py diff --git a/pina/__init__.py b/pina/__init__.py index cfcb28833..0d38804fe 100644 --- a/pina/__init__.py +++ b/pina/__init__.py @@ -22,7 +22,3 @@ from pina._src.core.trainer import Trainer from pina._src.condition.condition import Condition from pina._src.data.data_module import PinaDataModule - -# suppress errors -import torch._dynamo -torch._dynamo.config.suppress_errors = True diff --git a/pina/_src/condition/domain_equation_condition.py b/pina/_src/condition/domain_equation_condition.py index 93e76892a..0b75269ce 100644 --- a/pina/_src/condition/domain_equation_condition.py +++ b/pina/_src/condition/domain_equation_condition.py @@ -2,7 +2,7 @@ from pina._src.condition.condition_interface import ConditionInterface from pina._src.core.utils import check_consistency -from pina._src.domain import DomainInterface +from pina._src.domain.domain_interface import DomainInterface from pina._src.equation.equation_interface import EquationInterface diff --git a/pina/_src/data/data_module.py b/pina/_src/data/data_module.py index 664d200f8..f45236f0f 100644 --- a/pina/_src/data/data_module.py +++ b/pina/_src/data/data_module.py @@ -8,7 +8,7 @@ from lightning.pytorch import LightningDataModule import torch from torch_geometric.data import Data -from torch.utils.data import DataLoader, SequentialSampler, RandomSampler +from torch.utils.data import DataLoader, SequentialSampler from torch.utils.data.distributed import DistributedSampler from pina._src.core.label_tensor import LabelTensor from pina._src.data.dataset import PinaDatasetFactory, PinaTensorDataset diff --git a/pina/_src/model/average_neural_operator.py b/pina/_src/model/average_neural_operator.py index 6019b96c6..e16e3430f 100644 --- a/pina/_src/model/average_neural_operator.py +++ b/pina/_src/model/average_neural_operator.py @@ -2,9 +2,9 @@ import torch from torch import nn -from .block.average_neural_operator_block import AVNOBlock -from .kernel_neural_operator import KernelNeuralOperator -from ..utils import check_consistency +from pina._src.model.block.average_neural_operator_block import AVNOBlock +from pina._src.model.kernel_neural_operator import KernelNeuralOperator +from pina._src.core.utils import check_consistency class AveragingNeuralOperator(KernelNeuralOperator): diff --git a/pina/_src/model/deeponet.py b/pina/_src/model/deeponet.py index c65f6b316..800f2acc3 100644 --- a/pina/_src/model/deeponet.py +++ b/pina/_src/model/deeponet.py @@ -3,7 +3,7 @@ from functools import partial import torch from torch import nn -from ..utils import check_consistency, is_function +from pina._src.core.utils import check_consistency, is_function class MIONet(torch.nn.Module): diff --git a/pina/_src/model/equivariant_graph_neural_operator.py b/pina/_src/model/equivariant_graph_neural_operator.py index 6b33df6db..3aa7dde69 100644 --- a/pina/_src/model/equivariant_graph_neural_operator.py +++ b/pina/_src/model/equivariant_graph_neural_operator.py @@ -1,8 +1,10 @@ """Module for the Equivariant Graph Neural Operator model.""" import torch -from ..utils import check_positive_integer -from .block.message_passing import EquivariantGraphNeuralOperatorBlock +from pina._src.core.utils import check_positive_integer +from pina._src.model.block.message_passing.equivariant_graph_neural_operator_block import ( + EquivariantGraphNeuralOperatorBlock, +) class EquivariantGraphNeuralOperator(torch.nn.Module): diff --git a/pina/_src/model/feed_forward.py b/pina/_src/model/feed_forward.py index a1651b38b..fdf6bc91e 100644 --- a/pina/_src/model/feed_forward.py +++ b/pina/_src/model/feed_forward.py @@ -2,8 +2,8 @@ import torch from torch import nn -from ..utils import check_consistency -from .block.residual import EnhancedLinear +from pina._src.core.utils import check_consistency +from pina._src.model.block.residual import EnhancedLinear class FeedForward(torch.nn.Module): diff --git a/pina/_src/model/fourier_neural_operator.py b/pina/_src/model/fourier_neural_operator.py index e1336c999..7517b39b4 100644 --- a/pina/_src/model/fourier_neural_operator.py +++ b/pina/_src/model/fourier_neural_operator.py @@ -3,10 +3,14 @@ import warnings import torch from torch import nn -from ..label_tensor import LabelTensor -from ..utils import check_consistency -from .block.fourier_block import FourierBlock1D, FourierBlock2D, FourierBlock3D -from .kernel_neural_operator import KernelNeuralOperator +from pina._src.core.label_tensor import LabelTensor +from pina._src.core.utils import check_consistency +from pina._src.model.block.fourier_block import ( + FourierBlock1D, + FourierBlock2D, + FourierBlock3D, +) +from pina._src.model.kernel_neural_operator import KernelNeuralOperator class FourierIntegralKernel(torch.nn.Module): diff --git a/pina/_src/model/graph_neural_operator.py b/pina/_src/model/graph_neural_operator.py index 3cb5cdd31..e4d844fcb 100644 --- a/pina/_src/model/graph_neural_operator.py +++ b/pina/_src/model/graph_neural_operator.py @@ -2,8 +2,8 @@ import torch from torch.nn import Tanh -from .block.gno_block import GNOBlock -from .kernel_neural_operator import KernelNeuralOperator +from pina._src.model.block.gno_block import GNOBlock +from pina._src.model.kernel_neural_operator import KernelNeuralOperator class GraphNeuralKernel(torch.nn.Module): diff --git a/pina/_src/model/kernel_neural_operator.py b/pina/_src/model/kernel_neural_operator.py index e3cb790e5..81d1be45d 100644 --- a/pina/_src/model/kernel_neural_operator.py +++ b/pina/_src/model/kernel_neural_operator.py @@ -1,7 +1,7 @@ """Module for the Kernel Neural Operator model class.""" import torch -from ..utils import check_consistency +from pina._src.core.utils import check_consistency class KernelNeuralOperator(torch.nn.Module): diff --git a/pina/_src/model/low_rank_neural_operator.py b/pina/_src/model/low_rank_neural_operator.py index 1a7082dff..049894001 100644 --- a/pina/_src/model/low_rank_neural_operator.py +++ b/pina/_src/model/low_rank_neural_operator.py @@ -3,10 +3,10 @@ import torch from torch import nn -from ..utils import check_consistency +from pina._src.core.utils import check_consistency -from .kernel_neural_operator import KernelNeuralOperator -from .block.low_rank_block import LowRankBlock +from pina._src.model.kernel_neural_operator import KernelNeuralOperator +from pina._src.model.block.low_rank_block import LowRankBlock class LowRankNeuralOperator(KernelNeuralOperator): diff --git a/pina/_src/model/multi_feed_forward.py b/pina/_src/model/multi_feed_forward.py index f2f149ca6..df8fb19e2 100644 --- a/pina/_src/model/multi_feed_forward.py +++ b/pina/_src/model/multi_feed_forward.py @@ -2,7 +2,7 @@ from abc import ABC, abstractmethod import torch -from .feed_forward import FeedForward +from pina._src.model.feed_forward import FeedForward class MultiFeedForward(torch.nn.Module, ABC): diff --git a/pina/_src/model/pirate_network.py b/pina/_src/model/pirate_network.py index 96102b41f..09aad269d 100644 --- a/pina/_src/model/pirate_network.py +++ b/pina/_src/model/pirate_network.py @@ -1,8 +1,9 @@ """Module for the PirateNet model class.""" import torch -from .block import FourierFeatureEmbedding, PirateNetBlock -from ..utils import check_consistency, check_positive_integer +from pina._src.model.block.embedding import FourierFeatureEmbedding +from pina._src.model.block.pirate_network_block import PirateNetBlock +from pina._src.core.utils import check_consistency, check_positive_integer class PirateNet(torch.nn.Module): diff --git a/pina/_src/model/sindy.py b/pina/_src/model/sindy.py index a40fa37b4..f69842a54 100644 --- a/pina/_src/model/sindy.py +++ b/pina/_src/model/sindy.py @@ -2,7 +2,7 @@ from typing import Callable import torch -from ..utils import check_consistency, check_positive_integer +from pina._src.core.utils import check_consistency, check_positive_integer class SINDy(torch.nn.Module): diff --git a/pina/_src/model/spline.py b/pina/_src/model/spline.py index d9141fe8c..5e5b133c3 100644 --- a/pina/_src/model/spline.py +++ b/pina/_src/model/spline.py @@ -2,7 +2,7 @@ import warnings import torch -from ..utils import check_positive_integer, check_consistency +from pina._src.core.utils import check_consistency, check_positive_integer class Spline(torch.nn.Module): diff --git a/pina/_src/model/spline_surface.py b/pina/_src/model/spline_surface.py index 767e5b0dc..d54a0c7bb 100644 --- a/pina/_src/model/spline_surface.py +++ b/pina/_src/model/spline_surface.py @@ -2,8 +2,8 @@ import torch from .spline import Spline -from ..label_tensor import LabelTensor -from ..utils import check_consistency, check_positive_integer +from pina._src.core.label_tensor import LabelTensor +from pina._src.core.utils import check_consistency, check_positive_integer class SplineSurface(torch.nn.Module): diff --git a/pina/_src/problem/abstract_problem.py b/pina/_src/problem/abstract_problem.py index 9280d48d8..381186e00 100644 --- a/pina/_src/problem/abstract_problem.py +++ b/pina/_src/problem/abstract_problem.py @@ -3,11 +3,14 @@ from abc import ABCMeta, abstractmethod import warnings from copy import deepcopy -from ..utils import check_consistency -from ..domain import DomainInterface, CartesianDomain -from ..condition.domain_equation_condition import DomainEquationCondition -from ..label_tensor import LabelTensor -from ..utils import merge_tensors, custom_warning_format +from pina._src.core.utils import check_consistency +from pina._src.domain.domain_interface import DomainInterface +from pina._src.domain.cartesian_domain import CartesianDomain +from pina._src.condition.domain_equation_condition import ( + DomainEquationCondition, +) +from pina._src.core.label_tensor import LabelTensor +from pina._src.core.utils import merge_tensors, custom_warning_format class AbstractProblem(metaclass=ABCMeta): diff --git a/pina/_src/solver/ensemble_solver/ensemble_pinn.py b/pina/_src/solver/ensemble_solver/ensemble_pinn.py index 002d4bd53..f010753ec 100644 --- a/pina/_src/solver/ensemble_solver/ensemble_pinn.py +++ b/pina/_src/solver/ensemble_solver/ensemble_pinn.py @@ -5,8 +5,10 @@ from pina._src.solver.ensemble_solver.ensemble_solver_interface import ( DeepEnsembleSolverInterface, ) -from pina._src.solver.physics_informed_solver import PINNInterface -from pina._src.problem import InverseProblem +from pina._src.solver.physics_informed_solver.pinn_interface import ( + PINNInterface, +) +from pina._src.problem.inverse_problem import InverseProblem class DeepEnsemblePINN(PINNInterface, DeepEnsembleSolverInterface): diff --git a/pina/_src/solver/physics_informed_solver/pinn_interface.py b/pina/_src/solver/physics_informed_solver/pinn_interface.py index 60330372a..517b48082 100644 --- a/pina/_src/solver/physics_informed_solver/pinn_interface.py +++ b/pina/_src/solver/physics_informed_solver/pinn_interface.py @@ -5,7 +5,9 @@ import torch from pina._src.core.utils import custom_warning_format -from pina._src.solver.supervised_solver import SupervisedSolverInterface +from pina._src.solver.supervised_solver.supervised_solver_interface import ( + SupervisedSolverInterface, +) from pina._src.condition.input_target_condition import InputTargetCondition from pina._src.condition.input_equation_condition import InputEquationCondition from pina._src.condition.domain_equation_condition import ( diff --git a/pina/_src/solver/supervised_solver/__init__.py b/pina/_src/solver/supervised_solver/__init__.py index f681d2dd3..e69de29bb 100644 --- a/pina/_src/solver/supervised_solver/__init__.py +++ b/pina/_src/solver/supervised_solver/__init__.py @@ -1,11 +0,0 @@ -"""Module for the Supervised solvers.""" - -__all__ = [ - "SupervisedSolverInterface", - "SupervisedSolver", - "ReducedOrderModelSolver", -] - -from .supervised_solver_interface import SupervisedSolverInterface -from .supervised import SupervisedSolver -from .reduced_order_model import ReducedOrderModelSolver diff --git a/pina/data/__init__.py b/pina/data/__init__.py index 164a6c7aa..2ecebecdd 100644 --- a/pina/data/__init__.py +++ b/pina/data/__init__.py @@ -5,8 +5,28 @@ graph-structured data within the PINA training pipeline. """ -__all__ = ["PinaDataModule", "PinaDataset"] +from pina._src.data.data_module import ( + PinaDataModule, + PinaSampler, + DummyDataloader, + Collator, + PinaSampler, +) +from pina._src.data.dataset import ( + PinaDataset, + PinaTensorDataset, + PinaGraphDataset, + PinaDatasetFactory, +) -from pina._src.data.data_module import PinaDataModule -from pina._src.data.dataset import PinaDataset +__all__ = [ + "PinaDataModule", + "PinaDataset", + "PinaSampler", + "DummyDataloader", + "Collator", + "PinaTensorDataset", + "PinaGraphDataset", + "PinaDatasetFactory", +] diff --git a/pina/solver/__init__.py b/pina/solver/__init__.py index 327056035..9e1cb4021 100644 --- a/pina/solver/__init__.py +++ b/pina/solver/__init__.py @@ -48,14 +48,20 @@ SupervisedSolverInterface, ) -from pina._src.solver.supervised_solver import ( +from pina._src.solver.supervised_solver.supervised_solver_interface import ( SupervisedSolverInterface, - SupervisedSolver, +) +from pina._src.solver.supervised_solver.supervised import SupervisedSolver +from pina._src.solver.supervised_solver.reduced_order_model import ( ReducedOrderModelSolver, ) -from pina._src.solver.ensemble_solver import ( + +from pina._src.solver.ensemble_solver.ensemble_solver_interface import ( DeepEnsembleSolverInterface, +) +from pina._src.solver.ensemble_solver.ensemble_pinn import DeepEnsemblePINN +from pina._src.solver.ensemble_solver.ensemble_supervised import ( DeepEnsembleSupervisedSolver, - DeepEnsemblePINN, ) + from pina._src.solver.garom import GAROM diff --git a/pina/type_checker.py b/pina/type_checker.py new file mode 100644 index 000000000..cfa886da3 --- /dev/null +++ b/pina/type_checker.py @@ -0,0 +1,11 @@ +"""Runtime type enforcement and validation utilities. + +This module provides decorators to validate function arguments against type +annotations at runtime. This ensures that PINA components receive inputs +adhering to expected specifications, improving the robustness of the +scientific computing pipeline. +""" + +from pina._src.core.type_checker import enforce_types + +__all__ = ["enforce_types"] diff --git a/pina/utils.py b/pina/utils.py new file mode 100644 index 000000000..8e40ae9cd --- /dev/null +++ b/pina/utils.py @@ -0,0 +1,19 @@ +""" +Utility functions for tensor manipulation and input validation. + +This module provides helper functions to manage tensor operations and ensure +data consistency across the PINA framework, supporting robust input checking +and seamless data merging. +""" + +from pina._src.core.utils import ( + merge_tensors, + check_consistency, + check_positive_integer, +) + +__all__ = [ + "merge_tensors", + "check_consistency", + "check_positive_integer", +] diff --git a/tests/test_block/test_pod.py b/tests/test_block/test_pod.py index d10625fc3..ffe9a8a81 100644 --- a/tests/test_block/test_pod.py +++ b/tests/test_block/test_pod.py @@ -1,7 +1,7 @@ import torch import pytest -from pina.model.block.pod_block import PODBlock +from pina.model.block import PODBlock x = torch.linspace(-1, 1, 100) toy_snapshots = torch.vstack( diff --git a/tests/test_block/test_rbf.py b/tests/test_block/test_rbf.py index 65912fb76..c36836743 100644 --- a/tests/test_block/test_rbf.py +++ b/tests/test_block/test_rbf.py @@ -2,7 +2,7 @@ import pytest import math -from pina.model.block.rbf_block import RBFBlock +from pina.model.block import RBFBlock x = torch.linspace(-1, 1, 100) toy_params = torch.linspace(0, 1, 10).unsqueeze(1) diff --git a/tests/test_condition.py b/tests/test_condition.py index 9199f2bd9..266233179 100644 --- a/tests/test_condition.py +++ b/tests/test_condition.py @@ -18,7 +18,7 @@ GraphDataCondition, ) from pina.domain import CartesianDomain -from pina.equation.equation_factory import FixedValue +from pina.equation import FixedValue from pina.graph import RadiusGraph example_domain = CartesianDomain({"x": [0, 1], "y": [0, 1]}) diff --git a/tests/test_data/test_data_module.py b/tests/test_data/test_data_module.py index 53e7334ec..9fd2d36ee 100644 --- a/tests/test_data/test_data_module.py +++ b/tests/test_data/test_data_module.py @@ -1,10 +1,10 @@ import torch import pytest from pina.data import PinaDataModule -from pina.data.dataset import PinaTensorDataset, PinaGraphDataset +from pina.data import PinaTensorDataset, PinaGraphDataset from pina.problem.zoo import SupervisedProblem from pina.graph import RadiusGraph -from pina.data.data_module import DummyDataloader +from pina.data import DummyDataloader from pina import Trainer from pina.solver import SupervisedSolver from torch_geometric.data import Batch diff --git a/tests/test_data/test_graph_dataset.py b/tests/test_data/test_graph_dataset.py index 81d6a2c5d..3a63f7ec6 100644 --- a/tests/test_data/test_graph_dataset.py +++ b/tests/test_data/test_graph_dataset.py @@ -1,6 +1,6 @@ import torch import pytest -from pina.data.dataset import PinaDatasetFactory, PinaGraphDataset +from pina.data import PinaDatasetFactory, PinaGraphDataset from pina.graph import KNNGraph from torch_geometric.data import Data diff --git a/tests/test_data/test_tensor_dataset.py b/tests/test_data/test_tensor_dataset.py index 81a122f2f..9e348c942 100644 --- a/tests/test_data/test_tensor_dataset.py +++ b/tests/test_data/test_tensor_dataset.py @@ -1,6 +1,6 @@ import torch import pytest -from pina.data.dataset import PinaDatasetFactory, PinaTensorDataset +from pina.data import PinaDatasetFactory, PinaTensorDataset input_tensor = torch.rand((100, 10)) output_tensor = torch.rand((100, 2)) diff --git a/tests/test_loss/test_power_loss.py b/tests/test_loss/test_power_loss.py index 4ea90282b..3781f66d3 100644 --- a/tests/test_loss/test_power_loss.py +++ b/tests/test_loss/test_power_loss.py @@ -1,5 +1,4 @@ import torch -import pytest from pina.loss import PowerLoss diff --git a/tests/test_problem_zoo/test_supervised_problem.py b/tests/test_problem_zoo/test_supervised_problem.py index 19b3920ce..da18d6146 100644 --- a/tests/test_problem_zoo/test_supervised_problem.py +++ b/tests/test_problem_zoo/test_supervised_problem.py @@ -1,7 +1,7 @@ import torch from pina.problem import AbstractProblem from pina.condition import InputTargetCondition -from pina.problem.zoo.supervised_problem import SupervisedProblem +from pina.problem.zoo import SupervisedProblem from pina.graph import RadiusGraph From 9a70520040533924091f893212969131cf093926 Mon Sep 17 00:00:00 2001 From: Adam Dendek Date: Wed, 21 Jan 2026 03:24:44 +0100 Subject: [PATCH 4/4] Update documentation --- .../AdaptiveActivationFunctionInterface.rst | 2 +- .../_rst/adaptive_function/AdaptiveCELU.rst | 4 ++-- .../_rst/adaptive_function/AdaptiveELU.rst | 2 +- .../_rst/adaptive_function/AdaptiveExp.rst | 2 +- .../_rst/adaptive_function/AdaptiveGELU.rst | 2 +- .../_rst/adaptive_function/AdaptiveMish.rst | 2 +- .../_rst/adaptive_function/AdaptiveReLU.rst | 2 +- .../_rst/adaptive_function/AdaptiveSIREN.rst | 2 +- .../_rst/adaptive_function/AdaptiveSiLU.rst | 2 +- .../adaptive_function/AdaptiveSigmoid.rst | 2 +- .../adaptive_function/AdaptiveSoftmax.rst | 2 +- .../adaptive_function/AdaptiveSoftmin.rst | 2 +- .../_rst/adaptive_function/AdaptiveTanh.rst | 2 +- .../_rst/callback/optim/switch_optimizer.rst | 2 ++ .../_rst/callback/optim/switch_scheduler.rst | 2 ++ .../callback/processing/metric_tracker.rst | 3 ++- .../processing/normalizer_data_callback.rst | 2 ++ .../callback/processing/pina_progress_bar.rst | 3 ++- .../callback/refinement/r3_refinement.rst | 4 ++-- .../refinement/refinement_interface.rst | 4 ++-- docs/source/_rst/condition/condition.rst | 2 +- .../_rst/condition/condition_interface.rst | 2 +- docs/source/_rst/condition/data_condition.rst | 6 +++--- .../condition/domain_equation_condition.rst | 2 +- .../condition/input_equation_condition.rst | 6 +++--- .../_rst/condition/input_target_condition.rst | 10 +++++----- docs/source/_rst/data/data_module.rst | 6 +++--- docs/source/_rst/data/dataset.rst | 8 ++++---- docs/source/_rst/domain/base_domain.rst | 4 ++-- docs/source/_rst/domain/base_operation.rst | 4 ++-- docs/source/_rst/domain/cartesian_domain.rst | 4 ++-- docs/source/_rst/domain/difference.rst | 4 ++-- docs/source/_rst/domain/domain_interface.rst | 4 ++-- docs/source/_rst/domain/ellipsoid_domain.rst | 4 ++-- docs/source/_rst/domain/exclusion.rst | 4 ++-- docs/source/_rst/domain/intersection.rst | 4 ++-- .../_rst/domain/operation_interface.rst | 4 ++-- docs/source/_rst/domain/simplex_domain.rst | 4 ++-- docs/source/_rst/domain/union.rst | 4 ++-- docs/source/_rst/equation/equation.rst | 2 +- .../source/_rst/equation/equation_factory.rst | 20 +++++++++---------- .../_rst/equation/equation_interface.rst | 2 +- docs/source/_rst/equation/system_equation.rst | 2 +- docs/source/_rst/graph/graph.rst | 2 +- docs/source/_rst/graph/graph_builder.rst | 2 +- docs/source/_rst/graph/knn_graph.rst | 2 +- docs/source/_rst/graph/label_batch.rst | 2 +- docs/source/_rst/graph/radius_graph.rst | 2 +- docs/source/_rst/label_tensor.rst | 5 ++++- docs/source/_rst/loss/linear_weighting.rst | 10 ++++++---- docs/source/_rst/loss/loss_interface.rst | 4 ++-- docs/source/_rst/loss/lploss.rst | 5 ++++- docs/source/_rst/loss/ntk_weighting.rst | 4 ++-- docs/source/_rst/loss/powerloss.rst | 4 +++- docs/source/_rst/loss/scalar_weighting.rst | 4 ++-- .../_rst/loss/self_adaptive_weighting.rst | 4 ++-- docs/source/_rst/loss/weighting_interface.rst | 4 ++-- .../_rst/model/average_neural_operator.rst | 2 +- .../block/average_neural_operator_block.rst | 2 +- docs/source/_rst/model/block/convolution.rst | 2 +- .../model/block/convolution_interface.rst | 2 +- .../_rst/model/block/enhanced_linear.rst | 2 +- .../source/_rst/model/block/fourier_block.rst | 6 +++--- .../_rst/model/block/fourier_embedding.rst | 2 +- docs/source/_rst/model/block/gno_block.rst | 2 +- .../_rst/model/block/low_rank_block.rst | 2 +- .../deep_tensor_network_block.rst | 2 +- .../en_equivariant_network_block.rst | 2 +- ...quivariant_graph_neural_operator_block.rst | 2 +- .../interaction_network_block.rst | 2 +- .../radial_field_network_block.rst | 2 +- docs/source/_rst/model/block/orthogonal.rst | 2 +- .../source/_rst/model/block/pbc_embedding.rst | 2 +- .../_rst/model/block/pirate_network_block.rst | 2 +- docs/source/_rst/model/block/pod_block.rst | 2 +- docs/source/_rst/model/block/rbf_block.rst | 2 +- docs/source/_rst/model/block/residual.rst | 2 +- docs/source/_rst/model/block/spectral.rst | 6 +++--- docs/source/_rst/model/deeponet.rst | 2 +- .../equivariant_graph_neural_operator.rst | 2 +- docs/source/_rst/model/feed_forward.rst | 2 +- .../_rst/model/fourier_integral_kernel.rst | 2 +- .../_rst/model/fourier_neural_operator.rst | 2 +- .../_rst/model/graph_neural_operator.rst | 2 +- .../graph_neural_operator_integral_kernel.rst | 2 +- .../_rst/model/kernel_neural_operator.rst | 2 +- .../_rst/model/low_rank_neural_operator.rst | 2 +- docs/source/_rst/model/mionet.rst | 2 +- docs/source/_rst/model/multi_feed_forward.rst | 2 +- docs/source/_rst/model/pirate_network.rst | 2 +- .../_rst/model/residual_feed_forward.rst | 2 +- docs/source/_rst/model/sindy.rst | 2 +- docs/source/_rst/model/spline.rst | 2 +- docs/source/_rst/model/spline_surface.rst | 2 +- docs/source/_rst/operator.rst | 3 ++- .../source/_rst/optim/optimizer_interface.rst | 2 +- .../source/_rst/optim/scheduler_interface.rst | 2 +- docs/source/_rst/optim/torch_optimizer.rst | 2 +- docs/source/_rst/optim/torch_scheduler.rst | 2 +- docs/source/_rst/problem/abstract_problem.rst | 4 ++-- docs/source/_rst/problem/inverse_problem.rst | 4 ++-- .../_rst/problem/parametric_problem.rst | 4 ++-- docs/source/_rst/problem/spatial_problem.rst | 4 ++-- .../_rst/problem/time_dependent_problem.rst | 4 ++-- .../source/_rst/problem/zoo/acoustic_wave.rst | 4 ++-- docs/source/_rst/problem/zoo/advection.rst | 4 ++-- docs/source/_rst/problem/zoo/allen_cahn.rst | 4 ++-- .../_rst/problem/zoo/diffusion_reaction.rst | 4 ++-- docs/source/_rst/problem/zoo/helmholtz.rst | 4 ++-- .../problem/zoo/inverse_poisson_2d_square.rst | 4 ++-- .../_rst/problem/zoo/poisson_2d_square.rst | 4 ++-- .../_rst/problem/zoo/supervised_problem.rst | 4 ++-- .../solver/ensemble_solver/ensemble_pinn.rst | 2 +- .../ensemble_solver_interface.rst | 2 +- .../ensemble_solver/ensemble_supervised.rst | 2 +- docs/source/_rst/solver/garom.rst | 2 +- .../_rst/solver/multi_solver_interface.rst | 2 +- .../physics_informed_solver/causal_pinn.rst | 2 +- .../competitive_pinn.rst | 2 +- .../physics_informed_solver/gradient_pinn.rst | 2 +- .../solver/physics_informed_solver/pinn.rst | 2 +- .../pinn_interface.rst | 2 +- .../physics_informed_solver/rba_pinn.rst | 2 +- .../self_adaptive_pinn.rst | 2 +- .../_rst/solver/single_solver_interface.rst | 2 +- docs/source/_rst/solver/solver_interface.rst | 2 +- .../supervised_solver/reduced_order_model.rst | 2 +- .../solver/supervised_solver/supervised.rst | 2 +- .../supervised_solver_interface.rst | 2 +- docs/source/_rst/trainer.rst | 2 +- 130 files changed, 207 insertions(+), 188 deletions(-) diff --git a/docs/source/_rst/adaptive_function/AdaptiveActivationFunctionInterface.rst b/docs/source/_rst/adaptive_function/AdaptiveActivationFunctionInterface.rst index cf8b6551d..db035b46b 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveActivationFunctionInterface.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveActivationFunctionInterface.rst @@ -3,6 +3,6 @@ AdaptiveActivationFunctionInterface .. currentmodule:: pina.adaptive_function.adaptive_function_interface -.. automodule:: pina.adaptive_function.adaptive_function_interface +.. automodule:: pina._src.adaptive_function.adaptive_function_interface :members: :show-inheritance: diff --git a/docs/source/_rst/adaptive_function/AdaptiveCELU.rst b/docs/source/_rst/adaptive_function/AdaptiveCELU.rst index c4d6d5429..5c04ecde3 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveCELU.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveCELU.rst @@ -1,9 +1,9 @@ AdaptiveCELU ============ -.. currentmodule:: pina.adaptive_function.adaptive_function +.. currentmodule:: pina.adaptive_function -.. autoclass:: AdaptiveCELU +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveCELU :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveELU.rst b/docs/source/_rst/adaptive_function/AdaptiveELU.rst index aab273b08..2b27c4038 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveELU.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveELU.rst @@ -3,7 +3,7 @@ AdaptiveELU .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveELU +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveELU :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveExp.rst b/docs/source/_rst/adaptive_function/AdaptiveExp.rst index a7ee52b20..000f5bab2 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveExp.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveExp.rst @@ -3,7 +3,7 @@ AdaptiveExp .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveExp +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveExp :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveGELU.rst b/docs/source/_rst/adaptive_function/AdaptiveGELU.rst index b4aef14dc..35ae98382 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveGELU.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveGELU.rst @@ -3,7 +3,7 @@ AdaptiveGELU .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveGELU +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveGELU :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveMish.rst b/docs/source/_rst/adaptive_function/AdaptiveMish.rst index d006df054..6b440f5d2 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveMish.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveMish.rst @@ -3,7 +3,7 @@ AdaptiveMish .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveMish +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveMish :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveReLU.rst b/docs/source/_rst/adaptive_function/AdaptiveReLU.rst index d0fe4de68..379ee1d66 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveReLU.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveReLU.rst @@ -3,7 +3,7 @@ AdaptiveReLU .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveReLU +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveReLU :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveSIREN.rst b/docs/source/_rst/adaptive_function/AdaptiveSIREN.rst index 9f132547b..6e4aaf6f0 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveSIREN.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveSIREN.rst @@ -3,7 +3,7 @@ AdaptiveSIREN .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveSIREN +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveSIREN :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveSiLU.rst b/docs/source/_rst/adaptive_function/AdaptiveSiLU.rst index 722678611..b1fa345f1 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveSiLU.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveSiLU.rst @@ -3,7 +3,7 @@ AdaptiveSiLU .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveSiLU +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveSiLU :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveSigmoid.rst b/docs/source/_rst/adaptive_function/AdaptiveSigmoid.rst index 6002ffb31..3a2c19a9b 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveSigmoid.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveSigmoid.rst @@ -3,7 +3,7 @@ AdaptiveSigmoid .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveSigmoid +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveSigmoid :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveSoftmax.rst b/docs/source/_rst/adaptive_function/AdaptiveSoftmax.rst index c2b4c9f09..0a2352508 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveSoftmax.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveSoftmax.rst @@ -3,7 +3,7 @@ AdaptiveSoftmax .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveSoftmax +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveSoftmax :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveSoftmin.rst b/docs/source/_rst/adaptive_function/AdaptiveSoftmin.rst index 5189cb391..d842c5f26 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveSoftmin.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveSoftmin.rst @@ -3,7 +3,7 @@ AdaptiveSoftmin .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveSoftmin +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveSoftmin :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/adaptive_function/AdaptiveTanh.rst b/docs/source/_rst/adaptive_function/AdaptiveTanh.rst index 9a9b380a3..ca183abec 100644 --- a/docs/source/_rst/adaptive_function/AdaptiveTanh.rst +++ b/docs/source/_rst/adaptive_function/AdaptiveTanh.rst @@ -3,7 +3,7 @@ AdaptiveTanh .. currentmodule:: pina.adaptive_function.adaptive_function -.. autoclass:: AdaptiveTanh +.. autoclass:: pina._src.adaptive_function.adaptive_function.AdaptiveTanh :members: :show-inheritance: :inherited-members: AdaptiveActivationFunctionInterface diff --git a/docs/source/_rst/callback/optim/switch_optimizer.rst b/docs/source/_rst/callback/optim/switch_optimizer.rst index 635e79a18..13b7db7ad 100644 --- a/docs/source/_rst/callback/optim/switch_optimizer.rst +++ b/docs/source/_rst/callback/optim/switch_optimizer.rst @@ -2,6 +2,8 @@ Switch Optimizer ===================== .. currentmodule:: pina.callback.optim.switch_optimizer +.. automodule:: pina._src.callback.optim.switch_optimizer + :show-inheritance: .. autoclass:: SwitchOptimizer :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/callback/optim/switch_scheduler.rst b/docs/source/_rst/callback/optim/switch_scheduler.rst index 3176904da..42d5e6be0 100644 --- a/docs/source/_rst/callback/optim/switch_scheduler.rst +++ b/docs/source/_rst/callback/optim/switch_scheduler.rst @@ -2,6 +2,8 @@ Switch Scheduler ===================== .. currentmodule:: pina.callback.optim.switch_scheduler +.. automodule:: pina._src.callback.optim.switch_scheduler + :show-inheritance: .. autoclass:: SwitchScheduler :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/callback/processing/metric_tracker.rst b/docs/source/_rst/callback/processing/metric_tracker.rst index f21cc7730..202522831 100644 --- a/docs/source/_rst/callback/processing/metric_tracker.rst +++ b/docs/source/_rst/callback/processing/metric_tracker.rst @@ -1,7 +1,8 @@ Metric Tracker ================== .. currentmodule:: pina.callback.processing.metric_tracker - +.. automodule:: pina._src.callback.processing.metric_tracker + :show-inheritance: .. autoclass:: MetricTracker :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/callback/processing/normalizer_data_callback.rst b/docs/source/_rst/callback/processing/normalizer_data_callback.rst index a44f0c402..31fd769c8 100644 --- a/docs/source/_rst/callback/processing/normalizer_data_callback.rst +++ b/docs/source/_rst/callback/processing/normalizer_data_callback.rst @@ -2,6 +2,8 @@ Normalizer Data ======================= .. currentmodule:: pina.callback.processing.normalizer_data_callback +.. automodule:: pina._src.callback.processing.normalizer_data_callback + :show-inheritance: .. autoclass:: NormalizerDataCallback :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/callback/processing/pina_progress_bar.rst b/docs/source/_rst/callback/processing/pina_progress_bar.rst index 1d42ad120..da3a878ba 100644 --- a/docs/source/_rst/callback/processing/pina_progress_bar.rst +++ b/docs/source/_rst/callback/processing/pina_progress_bar.rst @@ -1,7 +1,8 @@ PINA Progress Bar ================== .. currentmodule:: pina.callback.processing.pina_progress_bar - +.. automodule:: pina._src.callback.processing.pina_progress_bar + :show-inheritance: .. autoclass:: PINAProgressBar :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/callback/refinement/r3_refinement.rst b/docs/source/_rst/callback/refinement/r3_refinement.rst index eb3bfebf2..5f0da6ea6 100644 --- a/docs/source/_rst/callback/refinement/r3_refinement.rst +++ b/docs/source/_rst/callback/refinement/r3_refinement.rst @@ -1,7 +1,7 @@ Refinments callbacks ======================= -.. currentmodule:: pina.callback.refinement -.. autoclass:: R3Refinement +.. currentmodule:: pina.callback +.. autoclass:: pina._src.callback.refinement.r3_refinement.R3Refinement :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/callback/refinement/refinement_interface.rst b/docs/source/_rst/callback/refinement/refinement_interface.rst index 5e02f2dc3..d1de6429b 100644 --- a/docs/source/_rst/callback/refinement/refinement_interface.rst +++ b/docs/source/_rst/callback/refinement/refinement_interface.rst @@ -1,7 +1,7 @@ Refinement Interface ======================= -.. currentmodule:: pina.callback.refinement -.. autoclass:: RefinementInterface +.. currentmodule:: pina.callback +.. autoclass:: pina._src.callback.refinement.refinement_interface.RefinementInterface :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/condition/condition.rst b/docs/source/_rst/condition/condition.rst index 51edfafff..cea9371f7 100644 --- a/docs/source/_rst/condition/condition.rst +++ b/docs/source/_rst/condition/condition.rst @@ -2,6 +2,6 @@ Conditions ============= .. currentmodule:: pina.condition.condition -.. autoclass:: Condition +.. autoclass:: pina._src.condition.condition.Condition :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/condition/condition_interface.rst b/docs/source/_rst/condition/condition_interface.rst index 88459629b..6c675c275 100644 --- a/docs/source/_rst/condition/condition_interface.rst +++ b/docs/source/_rst/condition/condition_interface.rst @@ -2,6 +2,6 @@ ConditionInterface ====================== .. currentmodule:: pina.condition.condition_interface -.. autoclass:: ConditionInterface +.. autoclass:: pina._src.condition.condition_interface.ConditionInterface :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/condition/data_condition.rst b/docs/source/_rst/condition/data_condition.rst index b7c322ea1..79d3ea13d 100644 --- a/docs/source/_rst/condition/data_condition.rst +++ b/docs/source/_rst/condition/data_condition.rst @@ -2,14 +2,14 @@ Data Conditions ================== .. currentmodule:: pina.condition.data_condition -.. autoclass:: DataCondition +.. autoclass:: pina._src.condition.data_condition.DataCondition :members: :show-inheritance: -.. autoclass:: GraphDataCondition +.. autoclass:: pina._src.condition.data_condition.GraphDataCondition :members: :show-inheritance: -.. autoclass:: TensorDataCondition +.. autoclass:: pina._src.condition.data_condition.TensorDataCondition :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/condition/domain_equation_condition.rst b/docs/source/_rst/condition/domain_equation_condition.rst index 505c8b839..10f1395ca 100644 --- a/docs/source/_rst/condition/domain_equation_condition.rst +++ b/docs/source/_rst/condition/domain_equation_condition.rst @@ -2,6 +2,6 @@ Domain Equation Condition =========================== .. currentmodule:: pina.condition.domain_equation_condition -.. autoclass:: DomainEquationCondition +.. autoclass:: pina._src.condition.domain_equation_condition.DomainEquationCondition :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/condition/input_equation_condition.rst b/docs/source/_rst/condition/input_equation_condition.rst index 4f5450e93..41b499fff 100644 --- a/docs/source/_rst/condition/input_equation_condition.rst +++ b/docs/source/_rst/condition/input_equation_condition.rst @@ -2,14 +2,14 @@ Input Equation Condition =========================== .. currentmodule:: pina.condition.input_equation_condition -.. autoclass:: InputEquationCondition +.. autoclass:: pina._src.condition.input_equation_condition.InputEquationCondition :members: :show-inheritance: -.. autoclass:: InputTensorEquationCondition +.. autoclass:: pina._src.condition.input_equation_condition.InputTensorEquationCondition :members: :show-inheritance: -.. autoclass:: InputGraphEquationCondition +.. autoclass:: pina._src.condition.input_equation_condition.InputGraphEquationCondition :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/condition/input_target_condition.rst b/docs/source/_rst/condition/input_target_condition.rst index 960b7d6f4..b4bbdcfc3 100644 --- a/docs/source/_rst/condition/input_target_condition.rst +++ b/docs/source/_rst/condition/input_target_condition.rst @@ -2,22 +2,22 @@ Input Target Condition =========================== .. currentmodule:: pina.condition.input_target_condition -.. autoclass:: InputTargetCondition +.. autoclass:: pina._src.condition.input_target_condition.InputTargetCondition :members: :show-inheritance: -.. autoclass:: TensorInputTensorTargetCondition +.. autoclass:: pina._src.condition.input_target_condition.TensorInputTensorTargetCondition :members: :show-inheritance: -.. autoclass:: TensorInputGraphTargetCondition +.. autoclass:: pina._src.condition.input_target_condition.TensorInputGraphTargetCondition :members: :show-inheritance: -.. autoclass:: GraphInputTensorTargetCondition +.. autoclass:: pina._src.condition.input_target_condition.GraphInputTensorTargetCondition :members: :show-inheritance: -.. autoclass:: GraphInputGraphTargetCondition +.. autoclass:: pina._src.condition.input_target_condition.GraphInputGraphTargetCondition :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/data/data_module.rst b/docs/source/_rst/data/data_module.rst index b7ffb14e0..7c57fbf62 100644 --- a/docs/source/_rst/data/data_module.rst +++ b/docs/source/_rst/data/data_module.rst @@ -2,14 +2,14 @@ DataModule ====================== .. currentmodule:: pina.data.data_module -.. autoclass:: Collator +.. autoclass:: pina._src.data.data_module.Collator :members: :show-inheritance: -.. autoclass:: PinaDataModule +.. autoclass:: pina._src.data.data_module.PinaDataModule :members: :show-inheritance: -.. autoclass:: PinaSampler +.. autoclass:: pina._src.data.data_module.PinaSampler :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/data/dataset.rst b/docs/source/_rst/data/dataset.rst index b49b41db1..264722b07 100644 --- a/docs/source/_rst/data/dataset.rst +++ b/docs/source/_rst/data/dataset.rst @@ -2,18 +2,18 @@ Dataset ====================== .. currentmodule:: pina.data.dataset -.. autoclass:: PinaDataset +.. autoclass:: pina._src.data.dataset.PinaDataset :members: :show-inheritance: -.. autoclass:: PinaDatasetFactory +.. autoclass:: pina._src.data.dataset.PinaDatasetFactory :members: :show-inheritance: -.. autoclass:: PinaGraphDataset +.. autoclass:: pina._src.data.dataset.PinaGraphDataset :members: :show-inheritance: -.. autoclass:: PinaTensorDataset +.. autoclass:: pina._src.data.dataset.PinaTensorDataset :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/domain/base_domain.rst b/docs/source/_rst/domain/base_domain.rst index e6b9ce88c..3850ba4fa 100644 --- a/docs/source/_rst/domain/base_domain.rst +++ b/docs/source/_rst/domain/base_domain.rst @@ -2,8 +2,8 @@ BaseDomain =========== .. currentmodule:: pina.domain.base_domain -.. automodule:: pina.domain.base_domain +.. automodule:: pina._src.domain.base_domain -.. autoclass:: BaseDomain +.. autoclass:: pina._src.domain.base_domain.BaseDomain :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/domain/base_operation.rst b/docs/source/_rst/domain/base_operation.rst index cfa145f03..122048d81 100644 --- a/docs/source/_rst/domain/base_operation.rst +++ b/docs/source/_rst/domain/base_operation.rst @@ -2,8 +2,8 @@ BaseOperation ============== .. currentmodule:: pina.domain.base_operation -.. automodule:: pina.domain.base_operation +.. automodule:: pina._src.domain.base_operation -.. autoclass:: BaseOperation +.. autoclass:: pina._src.domain.base_operation.BaseOperation :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/domain/cartesian_domain.rst b/docs/source/_rst/domain/cartesian_domain.rst index 15491be8c..bc2afec03 100644 --- a/docs/source/_rst/domain/cartesian_domain.rst +++ b/docs/source/_rst/domain/cartesian_domain.rst @@ -2,9 +2,9 @@ CartesianDomain ====================== .. currentmodule:: pina.domain.cartesian_domain -.. automodule:: pina.domain.cartesian_domain +.. automodule:: pina._src.domain.cartesian_domain -.. autoclass:: CartesianDomain +.. autoclass:: pina._src.domain.cartesian_domain.CartesianDomain :members: :show-inheritance: :noindex: diff --git a/docs/source/_rst/domain/difference.rst b/docs/source/_rst/domain/difference.rst index 0167c3062..91ffd4ec9 100644 --- a/docs/source/_rst/domain/difference.rst +++ b/docs/source/_rst/domain/difference.rst @@ -2,8 +2,8 @@ Difference ====================== .. currentmodule:: pina.domain.difference -.. automodule:: pina.domain.difference +.. automodule:: pina._src.domain.difference -.. autoclass:: Difference +.. autoclass:: pina._src.domain.difference.Difference :members: :show-inheritance: diff --git a/docs/source/_rst/domain/domain_interface.rst b/docs/source/_rst/domain/domain_interface.rst index 898896ba3..96594a23b 100644 --- a/docs/source/_rst/domain/domain_interface.rst +++ b/docs/source/_rst/domain/domain_interface.rst @@ -2,8 +2,8 @@ DomainInterface ================ .. currentmodule:: pina.domain.domain_interface -.. automodule:: pina.domain.domain_interface +.. automodule:: pina._src.domain.domain_interface -.. autoclass:: DomainInterface +.. autoclass:: pina._src.domain.domain_interface.DomainInterface :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/domain/ellipsoid_domain.rst b/docs/source/_rst/domain/ellipsoid_domain.rst index 4a9799e29..2cbc5f7ec 100644 --- a/docs/source/_rst/domain/ellipsoid_domain.rst +++ b/docs/source/_rst/domain/ellipsoid_domain.rst @@ -2,9 +2,9 @@ EllipsoidDomain ====================== .. currentmodule:: pina.domain.ellipsoid_domain -.. automodule:: pina.domain.ellipsoid_domain +.. automodule:: pina._src.domain.ellipsoid_domain -.. autoclass:: EllipsoidDomain +.. autoclass:: pina._src.domain.ellipsoid_domain.EllipsoidDomain :members: :show-inheritance: :noindex: diff --git a/docs/source/_rst/domain/exclusion.rst b/docs/source/_rst/domain/exclusion.rst index f624122ae..040b48416 100644 --- a/docs/source/_rst/domain/exclusion.rst +++ b/docs/source/_rst/domain/exclusion.rst @@ -2,8 +2,8 @@ Exclusion ====================== .. currentmodule:: pina.domain.exclusion -.. automodule:: pina.domain.exclusion +.. automodule:: pina._src.domain.exclusion -.. autoclass:: Exclusion +.. autoclass:: pina._src.domain.exclusion.Exclusion :members: :show-inheritance: diff --git a/docs/source/_rst/domain/intersection.rst b/docs/source/_rst/domain/intersection.rst index fade1d042..666fe0f00 100644 --- a/docs/source/_rst/domain/intersection.rst +++ b/docs/source/_rst/domain/intersection.rst @@ -2,8 +2,8 @@ Intersection ====================== .. currentmodule:: pina.domain.intersection -.. automodule:: pina.domain.intersection +.. automodule:: pina._src.domain.intersection -.. autoclass:: Intersection +.. autoclass:: pina._src.domain.intersection.Intersection :members: :show-inheritance: diff --git a/docs/source/_rst/domain/operation_interface.rst b/docs/source/_rst/domain/operation_interface.rst index 0acd393dc..42e92fbe8 100644 --- a/docs/source/_rst/domain/operation_interface.rst +++ b/docs/source/_rst/domain/operation_interface.rst @@ -2,8 +2,8 @@ OperationInterface ====================== .. currentmodule:: pina.domain.operation_interface -.. automodule:: pina.domain.operation_interface +.. automodule:: pina._src.domain.operation_interface -.. autoclass:: OperationInterface +.. autoclass:: pina._src.domain.operation_interface.OperationInterface :members: :show-inheritance: diff --git a/docs/source/_rst/domain/simplex_domain.rst b/docs/source/_rst/domain/simplex_domain.rst index 5f1d31c9b..0aba5f912 100644 --- a/docs/source/_rst/domain/simplex_domain.rst +++ b/docs/source/_rst/domain/simplex_domain.rst @@ -2,9 +2,9 @@ SimplexDomain ====================== .. currentmodule:: pina.domain.simplex_domain -.. automodule:: pina.domain.simplex_domain +.. automodule:: pina._src.domain.simplex_domain -.. autoclass:: SimplexDomain +.. autoclass:: pina._src.domain.simplex_domain.SimplexDomain :members: :show-inheritance: :noindex: diff --git a/docs/source/_rst/domain/union.rst b/docs/source/_rst/domain/union.rst index 614bb351c..fc5ff92a9 100644 --- a/docs/source/_rst/domain/union.rst +++ b/docs/source/_rst/domain/union.rst @@ -2,8 +2,8 @@ Union ====================== .. currentmodule:: pina.domain.union -.. automodule:: pina.domain.union +.. automodule:: pina._src.domain.union -.. autoclass:: Union +.. autoclass:: pina._src.domain.union.Union :members: :show-inheritance: diff --git a/docs/source/_rst/equation/equation.rst b/docs/source/_rst/equation/equation.rst index 33e19c957..edb350090 100644 --- a/docs/source/_rst/equation/equation.rst +++ b/docs/source/_rst/equation/equation.rst @@ -2,6 +2,6 @@ Equation ========== .. currentmodule:: pina.equation.equation -.. autoclass:: Equation +.. autoclass:: pina._src.equation.equation.Equation :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/equation/equation_factory.rst b/docs/source/_rst/equation/equation_factory.rst index 86390c6bd..5282aa948 100644 --- a/docs/source/_rst/equation/equation_factory.rst +++ b/docs/source/_rst/equation/equation_factory.rst @@ -2,42 +2,42 @@ Equation Factory ================== .. currentmodule:: pina.equation.equation_factory -.. autoclass:: FixedValue +.. autoclass:: pina._src.equation.equation_factory.FixedValue :members: :show-inheritance: -.. autoclass:: FixedGradient +.. autoclass:: pina._src.equation.equation_factory.FixedGradient :members: :show-inheritance: -.. autoclass:: FixedFlux +.. autoclass:: pina._src.equation.equation_factory.FixedFlux :members: :show-inheritance: -.. autoclass:: FixedLaplacian +.. autoclass:: pina._src.equation.equation_factory.FixedLaplacian :members: :show-inheritance: -.. autoclass:: Laplace +.. autoclass:: pina._src.equation.equation_factory.Laplace :members: :show-inheritance: -.. autoclass:: Advection +.. autoclass:: pina._src.equation.equation_factory.Advection :members: :show-inheritance: -.. autoclass:: AllenCahn +.. autoclass:: pina._src.equation.equation_factory.AllenCahn :members: :show-inheritance: -.. autoclass:: DiffusionReaction +.. autoclass:: pina._src.equation.equation_factory.DiffusionReaction :members: :show-inheritance: -.. autoclass:: Helmholtz +.. autoclass:: pina._src.equation.equation_factory.Helmholtz :members: :show-inheritance: -.. autoclass:: Poisson +.. autoclass:: pina._src.equation.equation_factory.Poisson :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/equation/equation_interface.rst b/docs/source/_rst/equation/equation_interface.rst index cde7b0012..f16502831 100644 --- a/docs/source/_rst/equation/equation_interface.rst +++ b/docs/source/_rst/equation/equation_interface.rst @@ -2,6 +2,6 @@ Equation Interface ==================== .. currentmodule:: pina.equation.equation_interface -.. autoclass:: EquationInterface +.. autoclass:: pina._src.equation.equation_interface.EquationInterface :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/equation/system_equation.rst b/docs/source/_rst/equation/system_equation.rst index 33c931cd9..88d1554f8 100644 --- a/docs/source/_rst/equation/system_equation.rst +++ b/docs/source/_rst/equation/system_equation.rst @@ -2,6 +2,6 @@ System Equation ================= .. currentmodule:: pina.equation.system_equation -.. autoclass:: SystemEquation +.. autoclass:: pina._src.equation.system_equation.SystemEquation :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/graph/graph.rst b/docs/source/_rst/graph/graph.rst index 1921f83e0..58180f50f 100644 --- a/docs/source/_rst/graph/graph.rst +++ b/docs/source/_rst/graph/graph.rst @@ -3,7 +3,7 @@ Graph .. currentmodule:: pina.graph -.. autoclass:: Graph +.. autoclass:: pina._src.core.graph.Graph :members: :private-members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/graph/graph_builder.rst b/docs/source/_rst/graph/graph_builder.rst index 2508aecb7..f576fe7c7 100644 --- a/docs/source/_rst/graph/graph_builder.rst +++ b/docs/source/_rst/graph/graph_builder.rst @@ -3,7 +3,7 @@ GraphBuilder .. currentmodule:: pina.graph -.. autoclass:: GraphBuilder +.. autoclass:: pina._src.core.graph.GraphBuilder :members: :private-members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/graph/knn_graph.rst b/docs/source/_rst/graph/knn_graph.rst index 8ef0b190b..e31a004ab 100644 --- a/docs/source/_rst/graph/knn_graph.rst +++ b/docs/source/_rst/graph/knn_graph.rst @@ -3,7 +3,7 @@ KNNGraph .. currentmodule:: pina.graph -.. autoclass:: KNNGraph +.. autoclass:: pina._src.core.graph.KNNGraph :members: :private-members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/graph/label_batch.rst b/docs/source/_rst/graph/label_batch.rst index 7cd4d2684..5a68bde60 100644 --- a/docs/source/_rst/graph/label_batch.rst +++ b/docs/source/_rst/graph/label_batch.rst @@ -3,7 +3,7 @@ LabelBatch .. currentmodule:: pina.graph -.. autoclass:: LabelBatch +.. autoclass:: pina._src.core.graph.LabelBatch :members: :private-members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/graph/radius_graph.rst b/docs/source/_rst/graph/radius_graph.rst index 7414d2dc1..9db9fb174 100644 --- a/docs/source/_rst/graph/radius_graph.rst +++ b/docs/source/_rst/graph/radius_graph.rst @@ -3,7 +3,7 @@ RadiusGraph .. currentmodule:: pina.graph -.. autoclass:: RadiusGraph +.. autoclass:: pina._src.core.graph.RadiusGraph :members: :private-members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/label_tensor.rst b/docs/source/_rst/label_tensor.rst index 9eb227369..1b750ad97 100644 --- a/docs/source/_rst/label_tensor.rst +++ b/docs/source/_rst/label_tensor.rst @@ -2,8 +2,11 @@ LabelTensor =========== .. currentmodule:: pina.label_tensor +.. automodule:: pina._src.core.label_tensor + :no-members: -.. autoclass:: LabelTensor + +.. autoclass:: pina._src.core.label_tensor.LabelTensor :members: :private-members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/loss/linear_weighting.rst b/docs/source/_rst/loss/linear_weighting.rst index 16e6232d0..359a4fbd1 100644 --- a/docs/source/_rst/loss/linear_weighting.rst +++ b/docs/source/_rst/loss/linear_weighting.rst @@ -1,9 +1,11 @@ LinearWeighting ============================= -.. currentmodule:: pina.loss.linear_weighting -.. automodule:: pina.loss.linear_weighting +.. currentmodule:: pina.loss -.. autoclass:: LinearWeighting +.. automodule:: pina._src.loss.linear_weighting + :no-members: + +.. autoclass:: pina._src.loss.linear_weighting.LinearWeighting :members: - :show-inheritance: + :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/loss/loss_interface.rst b/docs/source/_rst/loss/loss_interface.rst index 8ff78c01e..76a84d9c6 100644 --- a/docs/source/_rst/loss/loss_interface.rst +++ b/docs/source/_rst/loss/loss_interface.rst @@ -2,8 +2,8 @@ LossInterface =============== .. currentmodule:: pina.loss.loss_interface -.. automodule:: pina.loss.loss_interface +.. automodule:: pina._src.loss.loss_interface -.. autoclass:: LossInterface +.. autoclass:: pina._src.loss.loss_interface.LossInterface :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/loss/lploss.rst b/docs/source/_rst/loss/lploss.rst index 37dfdfe3c..9f6113a71 100644 --- a/docs/source/_rst/loss/lploss.rst +++ b/docs/source/_rst/loss/lploss.rst @@ -2,6 +2,9 @@ LpLoss =============== .. currentmodule:: pina.loss.lp_loss -.. autoclass:: LpLoss +.. automodule:: pina._src.loss.lp_loss + :no-members: + +.. autoclass:: pina._src.loss.lp_loss.LpLoss :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/loss/ntk_weighting.rst b/docs/source/_rst/loss/ntk_weighting.rst index 6d9d8816d..488e1923f 100644 --- a/docs/source/_rst/loss/ntk_weighting.rst +++ b/docs/source/_rst/loss/ntk_weighting.rst @@ -2,8 +2,8 @@ NeuralTangentKernelWeighting ============================= .. currentmodule:: pina.loss.ntk_weighting -.. automodule:: pina.loss.ntk_weighting +.. automodule:: pina._src.loss.ntk_weighting -.. autoclass:: NeuralTangentKernelWeighting +.. autoclass:: pina._src.loss.ntk_weighting.NeuralTangentKernelWeighting :members: :show-inheritance: diff --git a/docs/source/_rst/loss/powerloss.rst b/docs/source/_rst/loss/powerloss.rst index e4dee43b8..7a1438f4c 100644 --- a/docs/source/_rst/loss/powerloss.rst +++ b/docs/source/_rst/loss/powerloss.rst @@ -2,6 +2,8 @@ PowerLoss ==================== .. currentmodule:: pina.loss.power_loss -.. autoclass:: PowerLoss +.. automodule:: pina._src.loss.power_loss + +.. autoclass:: pina._src.loss.power_loss.PowerLoss :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/loss/scalar_weighting.rst b/docs/source/_rst/loss/scalar_weighting.rst index 5ee82a785..bbf8c06ec 100644 --- a/docs/source/_rst/loss/scalar_weighting.rst +++ b/docs/source/_rst/loss/scalar_weighting.rst @@ -2,8 +2,8 @@ ScalarWeighting =================== .. currentmodule:: pina.loss.scalar_weighting -.. automodule:: pina.loss.scalar_weighting +.. automodule:: pina._src.loss.scalar_weighting -.. autoclass:: ScalarWeighting +.. autoclass:: pina._src.loss.scalar_weighting.ScalarWeighting :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/loss/self_adaptive_weighting.rst b/docs/source/_rst/loss/self_adaptive_weighting.rst index cd1daed1f..6c39873d6 100644 --- a/docs/source/_rst/loss/self_adaptive_weighting.rst +++ b/docs/source/_rst/loss/self_adaptive_weighting.rst @@ -2,8 +2,8 @@ SelfAdaptiveWeighting ============================= .. currentmodule:: pina.loss.self_adaptive_weighting -.. automodule:: pina.loss.self_adaptive_weighting +.. automodule:: pina._src.loss.self_adaptive_weighting -.. autoclass:: SelfAdaptiveWeighting +.. autoclass:: pina._src.loss.self_adaptive_weighting.SelfAdaptiveWeighting :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/loss/weighting_interface.rst b/docs/source/_rst/loss/weighting_interface.rst index 2b0fa1bdc..d4aedd9d3 100644 --- a/docs/source/_rst/loss/weighting_interface.rst +++ b/docs/source/_rst/loss/weighting_interface.rst @@ -2,8 +2,8 @@ WeightingInterface =================== .. currentmodule:: pina.loss.weighting_interface -.. automodule:: pina.loss.weighting_interface +.. automodule:: pina._src.loss.weighting_interface -.. autoclass:: WeightingInterface +.. autoclass:: pina._src.loss.weighting_interface.WeightingInterface :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/average_neural_operator.rst b/docs/source/_rst/model/average_neural_operator.rst index 02211e9a8..a54107620 100644 --- a/docs/source/_rst/model/average_neural_operator.rst +++ b/docs/source/_rst/model/average_neural_operator.rst @@ -2,6 +2,6 @@ Averaging Neural Operator ============================== .. currentmodule:: pina.model.average_neural_operator -.. autoclass:: AveragingNeuralOperator +.. autoclass:: pina._src.model.average_neural_operator.AveragingNeuralOperator :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/block/average_neural_operator_block.rst b/docs/source/_rst/model/block/average_neural_operator_block.rst index 0072ec9d0..1e38fc215 100644 --- a/docs/source/_rst/model/block/average_neural_operator_block.rst +++ b/docs/source/_rst/model/block/average_neural_operator_block.rst @@ -2,7 +2,7 @@ Averaging Neural Operator Block ================================== .. currentmodule:: pina.model.block.average_neural_operator_block -.. autoclass:: AVNOBlock +.. autoclass:: pina._src.model.block.average_neural_operator_block.AVNOBlock :members: :show-inheritance: :noindex: diff --git a/docs/source/_rst/model/block/convolution.rst b/docs/source/_rst/model/block/convolution.rst index 4033d5d56..bd0d32e71 100644 --- a/docs/source/_rst/model/block/convolution.rst +++ b/docs/source/_rst/model/block/convolution.rst @@ -2,7 +2,7 @@ Continuous Convolution Block =============================== .. currentmodule:: pina.model.block.convolution_2d -.. autoclass:: ContinuousConvBlock +.. autoclass:: pina._src.model.block.convolution_2d.ContinuousConvBlock :members: :show-inheritance: :noindex: diff --git a/docs/source/_rst/model/block/convolution_interface.rst b/docs/source/_rst/model/block/convolution_interface.rst index f8e61c16c..c6708ca94 100644 --- a/docs/source/_rst/model/block/convolution_interface.rst +++ b/docs/source/_rst/model/block/convolution_interface.rst @@ -2,7 +2,7 @@ Continuous Convolution Interface ================================== .. currentmodule:: pina.model.block.convolution -.. autoclass:: BaseContinuousConv +.. autoclass:: pina._src.model.block.convolution.BaseContinuousConv :members: :show-inheritance: :noindex: diff --git a/docs/source/_rst/model/block/enhanced_linear.rst b/docs/source/_rst/model/block/enhanced_linear.rst index d08cf79bf..92e8d5581 100644 --- a/docs/source/_rst/model/block/enhanced_linear.rst +++ b/docs/source/_rst/model/block/enhanced_linear.rst @@ -2,7 +2,7 @@ EnhancedLinear Block ===================== .. currentmodule:: pina.model.block.residual -.. autoclass:: EnhancedLinear +.. autoclass:: pina._src.model.block.residual.EnhancedLinear :members: :show-inheritance: :noindex: \ No newline at end of file diff --git a/docs/source/_rst/model/block/fourier_block.rst b/docs/source/_rst/model/block/fourier_block.rst index c0fff4deb..9b601bb3d 100644 --- a/docs/source/_rst/model/block/fourier_block.rst +++ b/docs/source/_rst/model/block/fourier_block.rst @@ -3,14 +3,14 @@ Fourier Neural Operator Block .. currentmodule:: pina.model.block.fourier_block -.. autoclass:: FourierBlock1D +.. autoclass:: pina._src.model.block.fourier_block.FourierBlock1D :members: :show-inheritance: -.. autoclass:: FourierBlock2D +.. autoclass:: pina._src.model.block.fourier_block.FourierBlock2D :members: :show-inheritance: -.. autoclass:: FourierBlock3D +.. autoclass:: pina._src.model.block.fourier_block.FourierBlock3D :members: :show-inheritance: diff --git a/docs/source/_rst/model/block/fourier_embedding.rst b/docs/source/_rst/model/block/fourier_embedding.rst index 77eb3960c..48c8df41c 100644 --- a/docs/source/_rst/model/block/fourier_embedding.rst +++ b/docs/source/_rst/model/block/fourier_embedding.rst @@ -2,7 +2,7 @@ Fourier Feature Embedding ======================================= .. currentmodule:: pina.model.block.embedding -.. autoclass:: FourierFeatureEmbedding +.. autoclass:: pina._src.model.block.embedding.FourierFeatureEmbedding :members: :show-inheritance: diff --git a/docs/source/_rst/model/block/gno_block.rst b/docs/source/_rst/model/block/gno_block.rst index 19a532bab..8ce3f2f30 100644 --- a/docs/source/_rst/model/block/gno_block.rst +++ b/docs/source/_rst/model/block/gno_block.rst @@ -2,7 +2,7 @@ Graph Neural Operator Block =============================== .. currentmodule:: pina.model.block.gno_block -.. autoclass:: GNOBlock +.. autoclass:: pina._src.model.block.gno_block.GNOBlock :members: :show-inheritance: :noindex: diff --git a/docs/source/_rst/model/block/low_rank_block.rst b/docs/source/_rst/model/block/low_rank_block.rst index 366068f79..83c7a11a0 100644 --- a/docs/source/_rst/model/block/low_rank_block.rst +++ b/docs/source/_rst/model/block/low_rank_block.rst @@ -2,7 +2,7 @@ Low Rank Neural Operator Block ================================= .. currentmodule:: pina.model.block.low_rank_block -.. autoclass:: LowRankBlock +.. autoclass:: pina._src.model.block.low_rank_block.LowRankBlock :members: :show-inheritance: :noindex: diff --git a/docs/source/_rst/model/block/message_passing/deep_tensor_network_block.rst b/docs/source/_rst/model/block/message_passing/deep_tensor_network_block.rst index 30121e5a6..51482496a 100644 --- a/docs/source/_rst/model/block/message_passing/deep_tensor_network_block.rst +++ b/docs/source/_rst/model/block/message_passing/deep_tensor_network_block.rst @@ -2,7 +2,7 @@ Deep Tensor Network Block ================================== .. currentmodule:: pina.model.block.message_passing.deep_tensor_network_block -.. autoclass:: DeepTensorNetworkBlock +.. autoclass:: pina._src.model.block.message_passing.deep_tensor_network_block.DeepTensorNetworkBlock :members: :show-inheritance: :noindex: diff --git a/docs/source/_rst/model/block/message_passing/en_equivariant_network_block.rst b/docs/source/_rst/model/block/message_passing/en_equivariant_network_block.rst index e2755c665..09966ea0a 100644 --- a/docs/source/_rst/model/block/message_passing/en_equivariant_network_block.rst +++ b/docs/source/_rst/model/block/message_passing/en_equivariant_network_block.rst @@ -2,7 +2,7 @@ E(n) Equivariant Network Block ================================== .. currentmodule:: pina.model.block.message_passing.en_equivariant_network_block -.. autoclass:: EnEquivariantNetworkBlock +.. autoclass:: pina._src.model.block.message_passing.en_equivariant_network_block.EnEquivariantNetworkBlock :members: :show-inheritance: :noindex: \ No newline at end of file diff --git a/docs/source/_rst/model/block/message_passing/equivariant_graph_neural_operator_block.rst b/docs/source/_rst/model/block/message_passing/equivariant_graph_neural_operator_block.rst index 8d047f84e..b61c4f430 100644 --- a/docs/source/_rst/model/block/message_passing/equivariant_graph_neural_operator_block.rst +++ b/docs/source/_rst/model/block/message_passing/equivariant_graph_neural_operator_block.rst @@ -2,6 +2,6 @@ EquivariantGraphNeuralOperatorBlock ===================================== .. currentmodule:: pina.model.block.message_passing.equivariant_graph_neural_operator_block -.. autoclass:: EquivariantGraphNeuralOperatorBlock +.. autoclass:: pina._src.model.block.message_passing.equivariant_graph_neural_operator_block.EquivariantGraphNeuralOperatorBlock :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/block/message_passing/interaction_network_block.rst b/docs/source/_rst/model/block/message_passing/interaction_network_block.rst index ffac307e2..a4c86e562 100644 --- a/docs/source/_rst/model/block/message_passing/interaction_network_block.rst +++ b/docs/source/_rst/model/block/message_passing/interaction_network_block.rst @@ -2,7 +2,7 @@ Interaction Network Block ================================== .. currentmodule:: pina.model.block.message_passing.interaction_network_block -.. autoclass:: InteractionNetworkBlock +.. autoclass:: pina._src.model.block.message_passing.interaction_network_block.InteractionNetworkBlock :members: :show-inheritance: :noindex: \ No newline at end of file diff --git a/docs/source/_rst/model/block/message_passing/radial_field_network_block.rst b/docs/source/_rst/model/block/message_passing/radial_field_network_block.rst index e05203f33..bb66ee770 100644 --- a/docs/source/_rst/model/block/message_passing/radial_field_network_block.rst +++ b/docs/source/_rst/model/block/message_passing/radial_field_network_block.rst @@ -2,7 +2,7 @@ Radial Field Network Block ================================== .. currentmodule:: pina.model.block.message_passing.radial_field_network_block -.. autoclass:: RadialFieldNetworkBlock +.. autoclass:: pina._src.model.block.message_passing.radial_field_network_block.RadialFieldNetworkBlock :members: :show-inheritance: :noindex: \ No newline at end of file diff --git a/docs/source/_rst/model/block/orthogonal.rst b/docs/source/_rst/model/block/orthogonal.rst index 21d12998a..a9fc727fb 100644 --- a/docs/source/_rst/model/block/orthogonal.rst +++ b/docs/source/_rst/model/block/orthogonal.rst @@ -2,6 +2,6 @@ Orthogonal Block ====================== .. currentmodule:: pina.model.block.orthogonal -.. autoclass:: OrthogonalBlock +.. autoclass:: pina._src.model.block.orthogonal.OrthogonalBlock :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/block/pbc_embedding.rst b/docs/source/_rst/model/block/pbc_embedding.rst index f469644af..e79ae9514 100644 --- a/docs/source/_rst/model/block/pbc_embedding.rst +++ b/docs/source/_rst/model/block/pbc_embedding.rst @@ -2,7 +2,7 @@ Periodic Boundary Condition Embedding ======================================= .. currentmodule:: pina.model.block.embedding -.. autoclass:: PeriodicBoundaryEmbedding +.. autoclass:: pina._src.model.block.embedding.PeriodicBoundaryEmbedding :members: :show-inheritance: diff --git a/docs/source/_rst/model/block/pirate_network_block.rst b/docs/source/_rst/model/block/pirate_network_block.rst index 5d0428a68..f534d3cb0 100644 --- a/docs/source/_rst/model/block/pirate_network_block.rst +++ b/docs/source/_rst/model/block/pirate_network_block.rst @@ -2,7 +2,7 @@ PirateNet Block ======================================= .. currentmodule:: pina.model.block.pirate_network_block -.. autoclass:: PirateNetBlock +.. autoclass:: pina._src.model.block.pirate_network_block.PirateNetBlock :members: :show-inheritance: diff --git a/docs/source/_rst/model/block/pod_block.rst b/docs/source/_rst/model/block/pod_block.rst index 4b66e2c97..98fadbb1e 100644 --- a/docs/source/_rst/model/block/pod_block.rst +++ b/docs/source/_rst/model/block/pod_block.rst @@ -2,6 +2,6 @@ Proper Orthogonal Decomposition Block ============================================ .. currentmodule:: pina.model.block.pod_block -.. autoclass:: PODBlock +.. autoclass:: pina._src.model.block.pod_block.PODBlock :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/block/rbf_block.rst b/docs/source/_rst/model/block/rbf_block.rst index 545f14d08..b8997d21b 100644 --- a/docs/source/_rst/model/block/rbf_block.rst +++ b/docs/source/_rst/model/block/rbf_block.rst @@ -2,6 +2,6 @@ Radias Basis Function Block ============================= .. currentmodule:: pina.model.block.rbf_block -.. autoclass:: RBFBlock +.. autoclass:: pina._src.model.block.rbf_block.RBFBlock :members: :show-inheritance: diff --git a/docs/source/_rst/model/block/residual.rst b/docs/source/_rst/model/block/residual.rst index 69741c74c..d0e478563 100644 --- a/docs/source/_rst/model/block/residual.rst +++ b/docs/source/_rst/model/block/residual.rst @@ -2,6 +2,6 @@ Residual Block =================== .. currentmodule:: pina.model.block.residual -.. autoclass:: ResidualBlock +.. autoclass:: pina._src.model.block.residual.ResidualBlock :members: :show-inheritance: diff --git a/docs/source/_rst/model/block/spectral.rst b/docs/source/_rst/model/block/spectral.rst index 3c80f3dd8..1ee0e1d19 100644 --- a/docs/source/_rst/model/block/spectral.rst +++ b/docs/source/_rst/model/block/spectral.rst @@ -2,14 +2,14 @@ Spectral Convolution Block ============================ .. currentmodule:: pina.model.block.spectral -.. autoclass:: SpectralConvBlock1D +.. autoclass:: pina._src.model.block.spectral.SpectralConvBlock1D :members: :show-inheritance: -.. autoclass:: SpectralConvBlock2D +.. autoclass:: pina._src.model.block.spectral.SpectralConvBlock2D :members: :show-inheritance: -.. autoclass:: SpectralConvBlock3D +.. autoclass:: pina._src.model.block.spectral.SpectralConvBlock3D :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/deeponet.rst b/docs/source/_rst/model/deeponet.rst index 0ca08242d..eef25dcae 100644 --- a/docs/source/_rst/model/deeponet.rst +++ b/docs/source/_rst/model/deeponet.rst @@ -2,6 +2,6 @@ DeepONet =========== .. currentmodule:: pina.model.deeponet -.. autoclass:: DeepONet +.. autoclass:: pina._src.model.deeponet.DeepONet :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/equivariant_graph_neural_operator.rst b/docs/source/_rst/model/equivariant_graph_neural_operator.rst index a11edcc00..e100f5c1e 100644 --- a/docs/source/_rst/model/equivariant_graph_neural_operator.rst +++ b/docs/source/_rst/model/equivariant_graph_neural_operator.rst @@ -2,6 +2,6 @@ EquivariantGraphNeuralOperator ================================= .. currentmodule:: pina.model.equivariant_graph_neural_operator -.. autoclass:: EquivariantGraphNeuralOperator +.. autoclass:: pina._src.model.equivariant_graph_neural_operator.EquivariantGraphNeuralOperator :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/feed_forward.rst b/docs/source/_rst/model/feed_forward.rst index 2dea8e550..be75ed70b 100644 --- a/docs/source/_rst/model/feed_forward.rst +++ b/docs/source/_rst/model/feed_forward.rst @@ -2,6 +2,6 @@ FeedForward ====================== .. currentmodule:: pina.model.feed_forward -.. autoclass:: FeedForward +.. autoclass:: pina._src.model.feed_forward.FeedForward :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/fourier_integral_kernel.rst b/docs/source/_rst/model/fourier_integral_kernel.rst index b1fb484fe..dba63c429 100644 --- a/docs/source/_rst/model/fourier_integral_kernel.rst +++ b/docs/source/_rst/model/fourier_integral_kernel.rst @@ -2,6 +2,6 @@ FourierIntegralKernel ========================= .. currentmodule:: pina.model.fourier_neural_operator -.. autoclass:: FourierIntegralKernel +.. autoclass:: pina._src.model.fourier_neural_operator.FourierIntegralKernel :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/fourier_neural_operator.rst b/docs/source/_rst/model/fourier_neural_operator.rst index e77494fd0..14cb52667 100644 --- a/docs/source/_rst/model/fourier_neural_operator.rst +++ b/docs/source/_rst/model/fourier_neural_operator.rst @@ -2,6 +2,6 @@ FNO =========== .. currentmodule:: pina.model.fourier_neural_operator -.. autoclass:: FNO +.. autoclass:: pina._src.model.fourier_neural_operator.FNO :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/graph_neural_operator.rst b/docs/source/_rst/model/graph_neural_operator.rst index fbb8600e5..7f7b7ed6b 100644 --- a/docs/source/_rst/model/graph_neural_operator.rst +++ b/docs/source/_rst/model/graph_neural_operator.rst @@ -2,6 +2,6 @@ GraphNeuralOperator ======================= .. currentmodule:: pina.model.graph_neural_operator -.. autoclass:: GraphNeuralOperator +.. autoclass:: pina._src.model.graph_neural_operator.GraphNeuralOperator :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/graph_neural_operator_integral_kernel.rst b/docs/source/_rst/model/graph_neural_operator_integral_kernel.rst index cf15a31a5..45f78c366 100644 --- a/docs/source/_rst/model/graph_neural_operator_integral_kernel.rst +++ b/docs/source/_rst/model/graph_neural_operator_integral_kernel.rst @@ -2,6 +2,6 @@ GraphNeuralKernel ======================= .. currentmodule:: pina.model.graph_neural_operator -.. autoclass:: GraphNeuralKernel +.. autoclass:: pina._src.model.graph_neural_operator.GraphNeuralKernel :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/kernel_neural_operator.rst b/docs/source/_rst/model/kernel_neural_operator.rst index d693afac5..75a39b223 100644 --- a/docs/source/_rst/model/kernel_neural_operator.rst +++ b/docs/source/_rst/model/kernel_neural_operator.rst @@ -2,6 +2,6 @@ KernelNeuralOperator ======================= .. currentmodule:: pina.model.kernel_neural_operator -.. autoclass:: KernelNeuralOperator +.. autoclass:: pina._src.model.kernel_neural_operator.KernelNeuralOperator :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/low_rank_neural_operator.rst b/docs/source/_rst/model/low_rank_neural_operator.rst index 22fe7cc93..e0362d144 100644 --- a/docs/source/_rst/model/low_rank_neural_operator.rst +++ b/docs/source/_rst/model/low_rank_neural_operator.rst @@ -2,6 +2,6 @@ Low Rank Neural Operator ============================== .. currentmodule:: pina.model.low_rank_neural_operator -.. autoclass:: LowRankNeuralOperator +.. autoclass:: pina._src.model.low_rank_neural_operator.LowRankNeuralOperator :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/mionet.rst b/docs/source/_rst/model/mionet.rst index fe6281710..1888d911e 100644 --- a/docs/source/_rst/model/mionet.rst +++ b/docs/source/_rst/model/mionet.rst @@ -2,6 +2,6 @@ MIONet =========== .. currentmodule:: pina.model.deeponet -.. autoclass:: MIONet +.. autoclass:: pina._src.model.deeponet.MIONet :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/multi_feed_forward.rst b/docs/source/_rst/model/multi_feed_forward.rst index aa79580ee..458173ced 100644 --- a/docs/source/_rst/model/multi_feed_forward.rst +++ b/docs/source/_rst/model/multi_feed_forward.rst @@ -2,6 +2,6 @@ MultiFeedForward ================== .. currentmodule:: pina.model.multi_feed_forward -.. autoclass:: MultiFeedForward +.. autoclass:: pina._src.model.multi_feed_forward.MultiFeedForward :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/pirate_network.rst b/docs/source/_rst/model/pirate_network.rst index 5b374c247..a60449a6c 100644 --- a/docs/source/_rst/model/pirate_network.rst +++ b/docs/source/_rst/model/pirate_network.rst @@ -2,6 +2,6 @@ PirateNet ======================= .. currentmodule:: pina.model.pirate_network -.. autoclass:: PirateNet +.. autoclass:: pina._src.model.pirate_network.PirateNet :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/residual_feed_forward.rst b/docs/source/_rst/model/residual_feed_forward.rst index 66d83a42c..d8ce08152 100644 --- a/docs/source/_rst/model/residual_feed_forward.rst +++ b/docs/source/_rst/model/residual_feed_forward.rst @@ -2,6 +2,6 @@ ResidualFeedForward ====================== .. currentmodule:: pina.model.feed_forward -.. autoclass:: ResidualFeedForward +.. autoclass:: pina._src.model.feed_forward.ResidualFeedForward :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/sindy.rst b/docs/source/_rst/model/sindy.rst index bd507603b..f07ca6d30 100644 --- a/docs/source/_rst/model/sindy.rst +++ b/docs/source/_rst/model/sindy.rst @@ -2,6 +2,6 @@ SINDy ======================= .. currentmodule:: pina.model.sindy -.. autoclass:: SINDy +.. autoclass:: pina._src.model.sindy.SINDy :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/spline.rst b/docs/source/_rst/model/spline.rst index aa7450b70..278a95d3b 100644 --- a/docs/source/_rst/model/spline.rst +++ b/docs/source/_rst/model/spline.rst @@ -2,6 +2,6 @@ Spline ======== .. currentmodule:: pina.model.spline -.. autoclass:: Spline +.. autoclass:: pina._src.model.spline.Spline :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/model/spline_surface.rst b/docs/source/_rst/model/spline_surface.rst index 6bbf137d8..9b204cd22 100644 --- a/docs/source/_rst/model/spline_surface.rst +++ b/docs/source/_rst/model/spline_surface.rst @@ -2,6 +2,6 @@ Spline Surface ================ .. currentmodule:: pina.model.spline_surface -.. autoclass:: SplineSurface +.. autoclass:: pina._src.model.spline_surface.SplineSurface :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/operator.rst b/docs/source/_rst/operator.rst index 42746a6f8..fe0ad0398 100644 --- a/docs/source/_rst/operator.rst +++ b/docs/source/_rst/operator.rst @@ -3,6 +3,7 @@ Operators .. currentmodule:: pina.operator -.. automodule:: pina.operator + +.. automodule:: pina._src.core.operator :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/optim/optimizer_interface.rst b/docs/source/_rst/optim/optimizer_interface.rst index 88c18e8f5..afd62f6a0 100644 --- a/docs/source/_rst/optim/optimizer_interface.rst +++ b/docs/source/_rst/optim/optimizer_interface.rst @@ -2,6 +2,6 @@ Optimizer ============ .. currentmodule:: pina.optim.optimizer_interface -.. autoclass:: Optimizer +.. autoclass:: pina._src.optim.optimizer_interface.Optimizer :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/optim/scheduler_interface.rst b/docs/source/_rst/optim/scheduler_interface.rst index ab8ee292e..0795c34e3 100644 --- a/docs/source/_rst/optim/scheduler_interface.rst +++ b/docs/source/_rst/optim/scheduler_interface.rst @@ -2,6 +2,6 @@ Scheduler ============= .. currentmodule:: pina.optim.scheduler_interface -.. autoclass:: Scheduler +.. autoclass:: pina._src.optim.scheduler_interface.Scheduler :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/optim/torch_optimizer.rst b/docs/source/_rst/optim/torch_optimizer.rst index 3e6c9d912..67ab59164 100644 --- a/docs/source/_rst/optim/torch_optimizer.rst +++ b/docs/source/_rst/optim/torch_optimizer.rst @@ -2,6 +2,6 @@ TorchOptimizer =============== .. currentmodule:: pina.optim.torch_optimizer -.. autoclass:: TorchOptimizer +.. autoclass:: pina._src.optim.torch_optimizer.TorchOptimizer :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/optim/torch_scheduler.rst b/docs/source/_rst/optim/torch_scheduler.rst index 5c3e4df36..272ba631f 100644 --- a/docs/source/_rst/optim/torch_scheduler.rst +++ b/docs/source/_rst/optim/torch_scheduler.rst @@ -2,6 +2,6 @@ TorchScheduler =============== .. currentmodule:: pina.optim.torch_scheduler -.. autoclass:: TorchScheduler +.. autoclass:: pina._src.optim.torch_scheduler.TorchScheduler :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/problem/abstract_problem.rst b/docs/source/_rst/problem/abstract_problem.rst index 143909e1b..ae5e5f26e 100644 --- a/docs/source/_rst/problem/abstract_problem.rst +++ b/docs/source/_rst/problem/abstract_problem.rst @@ -2,8 +2,8 @@ AbstractProblem =============== .. currentmodule:: pina.problem.abstract_problem -.. automodule:: pina.problem.abstract_problem +.. automodule:: pina._src.problem.abstract_problem -.. autoclass:: AbstractProblem +.. autoclass:: pina._src.problem.abstract_problem.AbstractProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/inverse_problem.rst b/docs/source/_rst/problem/inverse_problem.rst index 5ce306ffc..4b5de05cb 100644 --- a/docs/source/_rst/problem/inverse_problem.rst +++ b/docs/source/_rst/problem/inverse_problem.rst @@ -2,8 +2,8 @@ InverseProblem ============== .. currentmodule:: pina.problem.inverse_problem -.. automodule:: pina.problem.inverse_problem +.. automodule:: pina._src.problem.inverse_problem -.. autoclass:: InverseProblem +.. autoclass:: pina._src.problem.inverse_problem.InverseProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/parametric_problem.rst b/docs/source/_rst/problem/parametric_problem.rst index 8f217fbbe..1a5e83490 100644 --- a/docs/source/_rst/problem/parametric_problem.rst +++ b/docs/source/_rst/problem/parametric_problem.rst @@ -2,8 +2,8 @@ ParametricProblem ==================== .. currentmodule:: pina.problem.parametric_problem -.. automodule:: pina.problem.parametric_problem +.. automodule:: pina._src.problem.parametric_problem -.. autoclass:: ParametricProblem +.. autoclass:: pina._src.problem.parametric_problem.ParametricProblem :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/problem/spatial_problem.rst b/docs/source/_rst/problem/spatial_problem.rst index 90ec6ec3c..757243ef1 100644 --- a/docs/source/_rst/problem/spatial_problem.rst +++ b/docs/source/_rst/problem/spatial_problem.rst @@ -2,8 +2,8 @@ SpatialProblem ============== .. currentmodule:: pina.problem.spatial_problem -.. automodule:: pina.problem.spatial_problem +.. automodule:: pina._src.problem.spatial_problem -.. autoclass:: SpatialProblem +.. autoclass:: pina._src.problem.spatial_problem.SpatialProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/time_dependent_problem.rst b/docs/source/_rst/problem/time_dependent_problem.rst index db94121c2..dda1e07f1 100644 --- a/docs/source/_rst/problem/time_dependent_problem.rst +++ b/docs/source/_rst/problem/time_dependent_problem.rst @@ -2,8 +2,8 @@ TimeDependentProblem ==================== .. currentmodule:: pina.problem.time_dependent_problem -.. automodule:: pina.problem.time_dependent_problem +.. automodule:: pina._src.problem.time_dependent_problem -.. autoclass:: TimeDependentProblem +.. autoclass:: pina._src.problem.time_dependent_problem.TimeDependentProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/zoo/acoustic_wave.rst b/docs/source/_rst/problem/zoo/acoustic_wave.rst index 4a9489667..34fd46895 100644 --- a/docs/source/_rst/problem/zoo/acoustic_wave.rst +++ b/docs/source/_rst/problem/zoo/acoustic_wave.rst @@ -2,8 +2,8 @@ AcousticWaveProblem ===================== .. currentmodule:: pina.problem.zoo.acoustic_wave -.. automodule:: pina.problem.zoo.acoustic_wave +.. automodule:: pina._src.problem.zoo.acoustic_wave -.. autoclass:: AcousticWaveProblem +.. autoclass:: pina._src.problem.zoo.acoustic_wave.AcousticWaveProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/zoo/advection.rst b/docs/source/_rst/problem/zoo/advection.rst index b83cc9d99..07d0cd45d 100644 --- a/docs/source/_rst/problem/zoo/advection.rst +++ b/docs/source/_rst/problem/zoo/advection.rst @@ -2,8 +2,8 @@ AdvectionProblem ================== .. currentmodule:: pina.problem.zoo.advection -.. automodule:: pina.problem.zoo.advection +.. automodule:: pina._src.problem.zoo.advection -.. autoclass:: AdvectionProblem +.. autoclass:: pina._src.problem.zoo.advection.AdvectionProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/zoo/allen_cahn.rst b/docs/source/_rst/problem/zoo/allen_cahn.rst index ada3465d1..7be2104bf 100644 --- a/docs/source/_rst/problem/zoo/allen_cahn.rst +++ b/docs/source/_rst/problem/zoo/allen_cahn.rst @@ -2,8 +2,8 @@ AllenCahnProblem ================== .. currentmodule:: pina.problem.zoo.allen_cahn -.. automodule:: pina.problem.zoo.allen_cahn +.. automodule:: pina._src.problem.zoo.allen_cahn -.. autoclass:: AllenCahnProblem +.. autoclass:: pina._src.problem.zoo.allen_cahn.AllenCahnProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/zoo/diffusion_reaction.rst b/docs/source/_rst/problem/zoo/diffusion_reaction.rst index 0cad0fd67..d5269edd7 100644 --- a/docs/source/_rst/problem/zoo/diffusion_reaction.rst +++ b/docs/source/_rst/problem/zoo/diffusion_reaction.rst @@ -2,8 +2,8 @@ DiffusionReactionProblem ========================= .. currentmodule:: pina.problem.zoo.diffusion_reaction -.. automodule:: pina.problem.zoo.diffusion_reaction +.. automodule:: pina._src.problem.zoo.diffusion_reaction -.. autoclass:: DiffusionReactionProblem +.. autoclass:: pina._src.problem.zoo.diffusion_reaction.DiffusionReactionProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/zoo/helmholtz.rst b/docs/source/_rst/problem/zoo/helmholtz.rst index af4ec7dbc..06724f83b 100644 --- a/docs/source/_rst/problem/zoo/helmholtz.rst +++ b/docs/source/_rst/problem/zoo/helmholtz.rst @@ -2,8 +2,8 @@ HelmholtzProblem ================== .. currentmodule:: pina.problem.zoo.helmholtz -.. automodule:: pina.problem.zoo.helmholtz +.. automodule:: pina._src.problem.zoo.helmholtz -.. autoclass:: HelmholtzProblem +.. autoclass:: pina._src.problem.zoo.helmholtz.HelmholtzProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/zoo/inverse_poisson_2d_square.rst b/docs/source/_rst/problem/zoo/inverse_poisson_2d_square.rst index 727c17b47..d4885ff0c 100644 --- a/docs/source/_rst/problem/zoo/inverse_poisson_2d_square.rst +++ b/docs/source/_rst/problem/zoo/inverse_poisson_2d_square.rst @@ -2,8 +2,8 @@ InversePoisson2DSquareProblem ============================== .. currentmodule:: pina.problem.zoo.inverse_poisson_2d_square -.. automodule:: pina.problem.zoo.inverse_poisson_2d_square +.. automodule:: pina._src.problem.zoo.inverse_poisson_2d_square -.. autoclass:: InversePoisson2DSquareProblem +.. autoclass:: pina._src.problem.zoo.inverse_poisson_2d_square.InversePoisson2DSquareProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/zoo/poisson_2d_square.rst b/docs/source/_rst/problem/zoo/poisson_2d_square.rst index 718c33ccc..96b5e4397 100644 --- a/docs/source/_rst/problem/zoo/poisson_2d_square.rst +++ b/docs/source/_rst/problem/zoo/poisson_2d_square.rst @@ -2,8 +2,8 @@ Poisson2DSquareProblem ======================== .. currentmodule:: pina.problem.zoo.poisson_2d_square -.. automodule:: pina.problem.zoo.poisson_2d_square +.. automodule:: pina._src.problem.zoo.poisson_2d_square -.. autoclass:: Poisson2DSquareProblem +.. autoclass:: pina._src.problem.zoo.poisson_2d_square.Poisson2DSquareProblem :members: :show-inheritance: diff --git a/docs/source/_rst/problem/zoo/supervised_problem.rst b/docs/source/_rst/problem/zoo/supervised_problem.rst index aad7d5aa5..6bf368376 100644 --- a/docs/source/_rst/problem/zoo/supervised_problem.rst +++ b/docs/source/_rst/problem/zoo/supervised_problem.rst @@ -2,8 +2,8 @@ SupervisedProblem ================== .. currentmodule:: pina.problem.zoo.supervised_problem -.. automodule:: pina.problem.zoo.supervised_problem +.. automodule:: pina._src.problem.zoo.supervised_problem -.. autoclass:: SupervisedProblem +.. autoclass:: pina._src.problem.zoo.supervised_problem.SupervisedProblem :members: :show-inheritance: diff --git a/docs/source/_rst/solver/ensemble_solver/ensemble_pinn.rst b/docs/source/_rst/solver/ensemble_solver/ensemble_pinn.rst index 2e42dcf0d..1ce086f45 100644 --- a/docs/source/_rst/solver/ensemble_solver/ensemble_pinn.rst +++ b/docs/source/_rst/solver/ensemble_solver/ensemble_pinn.rst @@ -2,7 +2,7 @@ DeepEnsemblePINN ================== .. currentmodule:: pina.solver.ensemble_solver.ensemble_pinn -.. autoclass:: DeepEnsemblePINN +.. autoclass:: pina._src.solver.ensemble_solver.ensemble_pinn.DeepEnsemblePINN :show-inheritance: :members: diff --git a/docs/source/_rst/solver/ensemble_solver/ensemble_solver_interface.rst b/docs/source/_rst/solver/ensemble_solver/ensemble_solver_interface.rst index 664bb8c8f..637520dd1 100644 --- a/docs/source/_rst/solver/ensemble_solver/ensemble_solver_interface.rst +++ b/docs/source/_rst/solver/ensemble_solver/ensemble_solver_interface.rst @@ -2,7 +2,7 @@ DeepEnsembleSolverInterface ============================= .. currentmodule:: pina.solver.ensemble_solver.ensemble_solver_interface -.. autoclass:: DeepEnsembleSolverInterface +.. autoclass:: pina._src.solver.ensemble_solver.ensemble_solver_interface.DeepEnsembleSolverInterface :show-inheritance: :members: diff --git a/docs/source/_rst/solver/ensemble_solver/ensemble_supervised.rst b/docs/source/_rst/solver/ensemble_solver/ensemble_supervised.rst index 575b28594..9b3f51522 100644 --- a/docs/source/_rst/solver/ensemble_solver/ensemble_supervised.rst +++ b/docs/source/_rst/solver/ensemble_solver/ensemble_supervised.rst @@ -2,7 +2,7 @@ DeepEnsembleSupervisedSolver ============================= .. currentmodule:: pina.solver.ensemble_solver.ensemble_supervised -.. autoclass:: DeepEnsembleSupervisedSolver +.. autoclass:: pina._src.solver.ensemble_solver.ensemble_supervised.DeepEnsembleSupervisedSolver :show-inheritance: :members: diff --git a/docs/source/_rst/solver/garom.rst b/docs/source/_rst/solver/garom.rst index 0e5820f6f..901272bee 100644 --- a/docs/source/_rst/solver/garom.rst +++ b/docs/source/_rst/solver/garom.rst @@ -2,6 +2,6 @@ GAROM ====== .. currentmodule:: pina.solver.garom -.. autoclass:: GAROM +.. autoclass:: pina._src.solver.garom.GAROM :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/solver/multi_solver_interface.rst b/docs/source/_rst/solver/multi_solver_interface.rst index 7f68c83a4..676dffa5b 100644 --- a/docs/source/_rst/solver/multi_solver_interface.rst +++ b/docs/source/_rst/solver/multi_solver_interface.rst @@ -2,7 +2,7 @@ MultiSolverInterface ====================== .. currentmodule:: pina.solver.solver -.. autoclass:: MultiSolverInterface +.. autoclass:: pina._src.solver.solver.MultiSolverInterface :show-inheritance: :members: diff --git a/docs/source/_rst/solver/physics_informed_solver/causal_pinn.rst b/docs/source/_rst/solver/physics_informed_solver/causal_pinn.rst index 6fab9ef0e..2f6e2393c 100644 --- a/docs/source/_rst/solver/physics_informed_solver/causal_pinn.rst +++ b/docs/source/_rst/solver/physics_informed_solver/causal_pinn.rst @@ -2,6 +2,6 @@ CausalPINN ============== .. currentmodule:: pina.solver.physics_informed_solver.causal_pinn -.. autoclass:: CausalPINN +.. autoclass:: pina._src.solver.physics_informed_solver.causal_pinn.CausalPINN :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/solver/physics_informed_solver/competitive_pinn.rst b/docs/source/_rst/solver/physics_informed_solver/competitive_pinn.rst index 372cb0f3d..3a48d280a 100644 --- a/docs/source/_rst/solver/physics_informed_solver/competitive_pinn.rst +++ b/docs/source/_rst/solver/physics_informed_solver/competitive_pinn.rst @@ -2,6 +2,6 @@ CompetitivePINN ================= .. currentmodule:: pina.solver.physics_informed_solver.competitive_pinn -.. autoclass:: CompetitivePINN +.. autoclass:: pina._src.solver.physics_informed_solver.competitive_pinn.CompetitivePINN :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/solver/physics_informed_solver/gradient_pinn.rst b/docs/source/_rst/solver/physics_informed_solver/gradient_pinn.rst index 66a490013..7d5008cdb 100644 --- a/docs/source/_rst/solver/physics_informed_solver/gradient_pinn.rst +++ b/docs/source/_rst/solver/physics_informed_solver/gradient_pinn.rst @@ -2,6 +2,6 @@ GradientPINN ============== .. currentmodule:: pina.solver.physics_informed_solver.gradient_pinn -.. autoclass:: GradientPINN +.. autoclass:: pina._src.solver.physics_informed_solver.gradient_pinn.GradientPINN :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/solver/physics_informed_solver/pinn.rst b/docs/source/_rst/solver/physics_informed_solver/pinn.rst index fdc31253b..48b9e603d 100644 --- a/docs/source/_rst/solver/physics_informed_solver/pinn.rst +++ b/docs/source/_rst/solver/physics_informed_solver/pinn.rst @@ -2,6 +2,6 @@ PINN ====== .. currentmodule:: pina.solver.physics_informed_solver.pinn -.. autoclass:: PINN +.. autoclass:: pina._src.solver.physics_informed_solver.pinn.PINN :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/solver/physics_informed_solver/pinn_interface.rst b/docs/source/_rst/solver/physics_informed_solver/pinn_interface.rst index 2242cf8b4..6577d6e69 100644 --- a/docs/source/_rst/solver/physics_informed_solver/pinn_interface.rst +++ b/docs/source/_rst/solver/physics_informed_solver/pinn_interface.rst @@ -2,6 +2,6 @@ PINNInterface ================= .. currentmodule:: pina.solver.physics_informed_solver.pinn_interface -.. autoclass:: PINNInterface +.. autoclass:: pina._src.solver.physics_informed_solver.pinn_interface.PINNInterface :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/solver/physics_informed_solver/rba_pinn.rst b/docs/source/_rst/solver/physics_informed_solver/rba_pinn.rst index cf94b6df0..af449fcf9 100644 --- a/docs/source/_rst/solver/physics_informed_solver/rba_pinn.rst +++ b/docs/source/_rst/solver/physics_informed_solver/rba_pinn.rst @@ -2,6 +2,6 @@ RBAPINN ======== .. currentmodule:: pina.solver.physics_informed_solver.rba_pinn -.. autoclass:: RBAPINN +.. autoclass:: pina._src.solver.physics_informed_solver.rba_pinn.RBAPINN :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/solver/physics_informed_solver/self_adaptive_pinn.rst b/docs/source/_rst/solver/physics_informed_solver/self_adaptive_pinn.rst index 2290059bd..dc42385fe 100644 --- a/docs/source/_rst/solver/physics_informed_solver/self_adaptive_pinn.rst +++ b/docs/source/_rst/solver/physics_informed_solver/self_adaptive_pinn.rst @@ -2,6 +2,6 @@ SelfAdaptivePINN ================== .. currentmodule:: pina.solver.physics_informed_solver.self_adaptive_pinn -.. autoclass:: SelfAdaptivePINN +.. autoclass:: pina._src.solver.physics_informed_solver.self_adaptive_pinn.SelfAdaptivePINN :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/solver/single_solver_interface.rst b/docs/source/_rst/solver/single_solver_interface.rst index 5b85f11b5..b47b85033 100644 --- a/docs/source/_rst/solver/single_solver_interface.rst +++ b/docs/source/_rst/solver/single_solver_interface.rst @@ -2,7 +2,7 @@ SingleSolverInterface ====================== .. currentmodule:: pina.solver.solver -.. autoclass:: SingleSolverInterface +.. autoclass:: pina._src.solver.solver.SingleSolverInterface :show-inheritance: :members: diff --git a/docs/source/_rst/solver/solver_interface.rst b/docs/source/_rst/solver/solver_interface.rst index 9bb11783e..bdd0aa92e 100644 --- a/docs/source/_rst/solver/solver_interface.rst +++ b/docs/source/_rst/solver/solver_interface.rst @@ -2,7 +2,7 @@ SolverInterface ================= .. currentmodule:: pina.solver.solver -.. autoclass:: SolverInterface +.. autoclass:: pina._src.solver.solver.SolverInterface :show-inheritance: :members: diff --git a/docs/source/_rst/solver/supervised_solver/reduced_order_model.rst b/docs/source/_rst/solver/supervised_solver/reduced_order_model.rst index 878014c29..08704c0b2 100644 --- a/docs/source/_rst/solver/supervised_solver/reduced_order_model.rst +++ b/docs/source/_rst/solver/supervised_solver/reduced_order_model.rst @@ -2,6 +2,6 @@ ReducedOrderModelSolver ========================== .. currentmodule:: pina.solver.supervised_solver.reduced_order_model -.. autoclass:: ReducedOrderModelSolver +.. autoclass:: pina._src.solver.supervised_solver.reduced_order_model.ReducedOrderModelSolver :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/solver/supervised_solver/supervised.rst b/docs/source/_rst/solver/supervised_solver/supervised.rst index 60ffdf828..8c93c4400 100644 --- a/docs/source/_rst/solver/supervised_solver/supervised.rst +++ b/docs/source/_rst/solver/supervised_solver/supervised.rst @@ -2,6 +2,6 @@ SupervisedSolver =================== .. currentmodule:: pina.solver.supervised_solver.supervised -.. autoclass:: SupervisedSolver +.. autoclass:: pina._src.solver.supervised_solver.supervised.SupervisedSolver :members: :show-inheritance: \ No newline at end of file diff --git a/docs/source/_rst/solver/supervised_solver/supervised_solver_interface.rst b/docs/source/_rst/solver/supervised_solver/supervised_solver_interface.rst index 4903a18dd..565b977cc 100644 --- a/docs/source/_rst/solver/supervised_solver/supervised_solver_interface.rst +++ b/docs/source/_rst/solver/supervised_solver/supervised_solver_interface.rst @@ -2,7 +2,7 @@ SupervisedSolverInterface ========================== .. currentmodule:: pina.solver.supervised_solver.supervised_solver_interface -.. autoclass:: SupervisedSolverInterface +.. autoclass:: pina._src.solver.supervised_solver.supervised_solver_interface.SupervisedSolverInterface :show-inheritance: :members: diff --git a/docs/source/_rst/trainer.rst b/docs/source/_rst/trainer.rst index 2582b6da9..8e5a99a38 100644 --- a/docs/source/_rst/trainer.rst +++ b/docs/source/_rst/trainer.rst @@ -3,6 +3,6 @@ Trainer .. automodule:: pina.trainer -.. autoclass:: Trainer +.. autoclass:: pina._src.core.trainer.Trainer :members: :show-inheritance: \ No newline at end of file