diff --git a/pina/__init__.py b/pina/__init__.py index 835232b..0fe9375 100644 --- a/pina/__init__.py +++ b/pina/__init__.py @@ -6,8 +6,6 @@ __all__ = [ "Condition", "SamplePointDataset", "SamplePointLoader", - "TorchOptimizer", - "TorchScheduler", ] from .meta import * @@ -15,8 +13,6 @@ from .label_tensor import LabelTensor from .solvers.solver import SolverInterface from .trainer import Trainer from .plotter import Plotter -from .optimizer import TorchOptimizer -from .scheduler import TorchScheduler from .condition.condition import Condition from .data import SamplePointDataset from .data import SamplePointLoader