Merge pull request #280 from brosaplanella/issue-279-refactor-problem

Refactor problems
pybop-team · Apr 14, 2024 · 36b1b11 · 36b1b11
2 parents 040a8b1 + f58c73f
commit 36b1b11
Show file tree

Hide file tree

Showing 9 changed files with 410 additions and 402 deletions.
diff --git a/examples/standalone/problem.py b/examples/standalone/problem.py
@@ -1,6 +1,6 @@
 import numpy as np
 
-from pybop._problem import BaseProblem
+from pybop import BaseProblem
 
 
 class StandaloneProblem(BaseProblem):

diff --git a/pybop/__init__.py b/pybop/__init__.py
@@ -31,7 +31,9 @@
 # Cost class
 # Problem class
 #
-from ._problem import BaseProblem, FittingProblem, DesignProblem
+from .problems.base_problem import BaseProblem
+from .problems.fitting_problem import FittingProblem
+from .problems.design_problem import DesignProblem
 
 #
 # Cost function class