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

edited utils to take list (#115)

Browse Source 
* enhanced difference domain
* refactored utils
* fixed typo
* added tests
---------

Co-authored-by: Dario Coscia <93731561+dario-coscia@users.noreply.github.com>
This commit is contained in:
Kush
2023-06-19 18:47:52 +02:00
committed by Nicola Demo
parent aaf2bed732
commit 62ec69ccac
9 changed files with 73 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/solver.py
View File

@@ -20,7 +20,7 @@ class SolverInterface(pl.LightningModule, metaclass=ABCMeta):
super().__init__()
# check inheritance for pina problem
check_consistency(problem, AbstractProblem, 'pina problem')
check_consistency(problem, AbstractProblem)
# assigning class variables (check consistency inside Network class)
self._pina_model = Network(model=model, extra_features=extra_features)