disable compilation py>=3.14

2025-10-28 12:29:19 +01:00
parent 24d806b262
commit 64930c431f
3 changed files with 40 additions and 11 deletions
--- a/pina/solver/solver.py
+++ b/pina/solver/solver.py
@@ -174,11 +174,7 @@ class SolverInterface(lightning.pytorch.LightningModule, metaclass=ABCMeta):
        :return: The result of the parent class ``setup`` method.
        :rtype: Any
        """
-        if stage == "fit" and self.trainer.compile:
-            self._setup_compile()
-        if stage == "test" and (
-            self.trainer.compile and not self._is_compiled()
-        ):
+        if self.trainer.compile and not self._is_compiled():
            self._setup_compile()
        return super().setup(stage)