folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove inner_size (#44)

Browse Source
This commit is contained in:
Francesco Andreuzzi
2022-12-05 10:50:28 +01:00
committed by GitHub
parent b6c88bb756
commit 399c72fc0e
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_deeponet.py
View File

@@ -17,6 +17,12 @@ def test_constructor():
onet = DeepONet(trunk_net=trunk, branch_net=branch,
output_variables=output_vars)
def test_constructor_fails_when_invalid_inner_layer_size():
branch = FFN(input_variables=['a', 'c'], output_variables=20)
trunk = FFN(input_variables=['b'], output_variables=19)
with pytest.raises(ValueError):
DeepONet(trunk_net=trunk, branch_net=branch, output_variables=output_vars)
def test_forward():
branch = FFN(input_variables=['a', 'c'], output_variables=10)
trunk = FFN(input_variables=['b'], output_variables=10)