optimizer and scheduler classes

pina/__init__.py

@@ -6,13 +6,17 @@ __all__ = [
     "Condition",
     "SamplePointDataset",
     "SamplePointLoader",
+    "TorchOptimizer",
+    "TorchScheduler",
 ]
 
 from .meta import *
-#from .label_tensor import LabelTensor
+from .label_tensor import LabelTensor
 from .solvers.solver import SolverInterface
 from .trainer import Trainer
 from .plotter import Plotter
 from .condition import Condition
 from .dataset import SamplePointDataset
 from .dataset import SamplePointLoader
+from .optimizer import TorchOptimizer
+from .scheduler import TorchScheduler