From 35d07473575756e0af4a4c0c83c6c0b6c718b4af Mon Sep 17 00:00:00 2001
From: Roy Stegeman <roystegeman@live.nl>
Date: Thu, 14 Nov 2024 15:03:45 +0000
Subject: [PATCH] fix pytest

---
 src/yadism/runner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/yadism/runner.py b/src/yadism/runner.py
index d69f747b..2498b8d7 100644
--- a/src/yadism/runner.py
+++ b/src/yadism/runner.py
@@ -216,7 +216,7 @@ def replace_nans_with_0(self, out):
         # Loop through each observable in the dictionary
         for observable, points in out2.items():
             # Skip the keys that are not an observable
-            if observable in ['xgrid', 'polynomial_degree', 'is_log', 'pids', 'projectilePID']:
+            if observable not in observable_name.kinds:
                 continue
 
             # Loop over the kinematic points