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

adding possibility to pass empty list in the solver

Browse Source
This commit is contained in:
Dario Coscia
2023-11-09 17:29:16 +01:00
committed by Nicola Demo
parent 09d013e8fc
commit 934ae409ff
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/solvers/solver.py
View File

@@ -69,7 +69,7 @@ class SolverInterface(pytorch_lightning.LightningModule, metaclass=ABCMeta):
f' {len_optimizer_kwargs} dicitionaries') f' {len_optimizer_kwargs} dicitionaries')
# extra features handling # extra features handling
if extra_features is None: if extra_features is None or (len(extra_features)==0):
extra_features = [None] * len_model extra_features = [None] * len_model
else: else:
# if we only have a list of extra features # if we only have a list of extra features