adding possibility to pass empty list in the solver

2023-11-09 17:29:16 +01:00
parent 09d013e8fc
commit 934ae409ff
1 changed files with 1 additions and 1 deletions
--- a/pina/solvers/solver.py
+++ b/pina/solvers/solver.py
@@ -69,7 +69,7 @@ class SolverInterface(pytorch_lightning.LightningModule, metaclass=ABCMeta):
                             f' {len_optimizer_kwargs} dicitionaries')

        # extra features handling
-        if extra_features is None:
+        if extra_features is None or (len(extra_features)==0):
            extra_features = [None] * len_model
        else:
            # if we only have a list of extra features