From 004cbc00c07dc57f77839659c0b897b246e07060 Mon Sep 17 00:00:00 2001
From: FilippoOlivo <filippo@filippoolivo.com>
Date: Thu, 6 Feb 2025 17:02:28 +0100
Subject: [PATCH] problem.input_pts takes into consideration also input_points
 coming from InputOutputCondition

---
 pina/problem/abstract_problem.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/pina/problem/abstract_problem.py b/pina/problem/abstract_problem.py
index 2273c52..3a30e33 100644
--- a/pina/problem/abstract_problem.py
+++ b/pina/problem/abstract_problem.py
@@ -57,9 +57,11 @@ class AbstractProblem(metaclass=ABCMeta):
     @property
     def input_pts(self):
         to_return = {}
-        for k, v in self.collector.data_collections.items():
-            if 'input_points' in v.keys():
-                to_return[k] = v['input_points']
+        for cond_name, cond in self.conditions.items():
+            if hasattr(cond, "input_points"):
+                to_return[cond_name] = cond.input_points
+            elif hasattr(cond, "domain"):
+                to_return[cond_name] = self.discretised_domains[cond.domain]
         return to_return
 
     def __deepcopy__(self, memo):