From 939353f517127a5c83b440eca17f5619429687b4 Mon Sep 17 00:00:00 2001
From: Dario Coscia <dariocoscia@dhcp-176.eduroam.sissa.it>
Date: Tue, 19 Sep 2023 15:21:45 +0200
Subject: [PATCH] adding trainer solver property decorator

---
 pina/trainer.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pina/trainer.py b/pina/trainer.py
index 3250614..be220d2 100644
--- a/pina/trainer.py
+++ b/pina/trainer.py
@@ -36,3 +36,9 @@ class Trainer(pl.Trainer):
     def train(self, **kwargs): # TODO add kwargs and lightining capabilities
         return super().fit(self._model, self._loader, **kwargs)
     
+    @property
+    def solver(self):
+        """
+        Returning trainer solver.
+        """
+        return self._model
\ No newline at end of file