Skip to content

Commit 0e46fc5

Browse files
GiovanniCanalidario-coscia
authored andcommitted
remove inverse diffusion-reaction
1 parent 36279cf commit 0e46fc5

File tree

4 files changed

+6
-136
lines changed

4 files changed

+6
-136
lines changed

pina/problem/zoo/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
"Poisson2DSquareProblem",
99
"DiffusionReactionProblem",
1010
"InversePoisson2DSquareProblem",
11-
"InverseDiffusionReactionProblem",
1211
]
1312

1413
from .supervised_problem import SupervisedProblem
@@ -18,4 +17,3 @@
1817
from .poisson_2d_square import Poisson2DSquareProblem
1918
from .diffusion_reaction import DiffusionReactionProblem
2019
from .inverse_poisson_2d_square import InversePoisson2DSquareProblem
21-
from .inverse_diffusion_reaction import InverseDiffusionReactionProblem

pina/problem/zoo/inverse_diffusion_reaction.py

Lines changed: 0 additions & 110 deletions
This file was deleted.

tests/test_problem_zoo/test_inverse_diffusion_reaction.py

Lines changed: 0 additions & 13 deletions
This file was deleted.

tests/test_solver/test_causal_pinn.py

Lines changed: 6 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,7 @@
66
from pina.solver import CausalPINN
77
from pina.trainer import Trainer
88
from pina.model import FeedForward
9-
from pina.problem.zoo import (
10-
DiffusionReactionProblem,
11-
InverseDiffusionReactionProblem,
12-
)
9+
from pina.problem.zoo import DiffusionReactionProblem
1310
from pina.condition import (
1411
InputTargetCondition,
1512
InputEquationCondition,
@@ -31,8 +28,6 @@ class DummySpatialProblem(SpatialProblem):
3128
# define problems
3229
problem = DiffusionReactionProblem()
3330
problem.discretise_domain(50)
34-
inverse_problem = InverseDiffusionReactionProblem()
35-
inverse_problem.discretise_domain(50)
3631

3732
# add input-output condition to test supervised learning
3833
input_pts = torch.rand(50, len(problem.input_variables))
@@ -45,7 +40,7 @@ class DummySpatialProblem(SpatialProblem):
4540
model = FeedForward(len(problem.input_variables), len(problem.output_variables))
4641

4742

48-
@pytest.mark.parametrize("problem", [problem, inverse_problem])
43+
@pytest.mark.parametrize("problem", [problem])
4944
@pytest.mark.parametrize("eps", [100, 100.1])
5045
def test_constructor(problem, eps):
5146
with pytest.raises(ValueError):
@@ -59,7 +54,7 @@ def test_constructor(problem, eps):
5954
)
6055

6156

62-
@pytest.mark.parametrize("problem", [problem, inverse_problem])
57+
@pytest.mark.parametrize("problem", [problem])
6358
@pytest.mark.parametrize("batch_size", [None, 1, 5, 20])
6459
@pytest.mark.parametrize("compile", [True, False])
6560
def test_solver_train(problem, batch_size, compile):
@@ -79,7 +74,7 @@ def test_solver_train(problem, batch_size, compile):
7974
assert isinstance(solver.model, OptimizedModule)
8075

8176

82-
@pytest.mark.parametrize("problem", [problem, inverse_problem])
77+
@pytest.mark.parametrize("problem", [problem])
8378
@pytest.mark.parametrize("batch_size", [None, 1, 5, 20])
8479
@pytest.mark.parametrize("compile", [True, False])
8580
def test_solver_validation(problem, batch_size, compile):
@@ -99,7 +94,7 @@ def test_solver_validation(problem, batch_size, compile):
9994
assert isinstance(solver.model, OptimizedModule)
10095

10196

102-
@pytest.mark.parametrize("problem", [problem, inverse_problem])
97+
@pytest.mark.parametrize("problem", [problem])
10398
@pytest.mark.parametrize("batch_size", [None, 1, 5, 20])
10499
@pytest.mark.parametrize("compile", [True, False])
105100
def test_solver_test(problem, batch_size, compile):
@@ -119,7 +114,7 @@ def test_solver_test(problem, batch_size, compile):
119114
assert isinstance(solver.model, OptimizedModule)
120115

121116

122-
@pytest.mark.parametrize("problem", [problem, inverse_problem])
117+
@pytest.mark.parametrize("problem", [problem])
123118
def test_train_load_restore(problem):
124119
dir = "tests/test_solver/tmp"
125120
problem = problem

0 commit comments

Comments
 (0)