Merge pull request #97 from aidh-ms/issue/91-rrt-probe-error

aidh-ms · Dec 5, 2023 · 6e736ab · 6e736ab
2 parents c3b5083 + b4bf51c
commit 6e736ab
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pyAKI/kdigo.py b/pyAKI/kdigo.py
@@ -151,7 +151,9 @@ def process_stay(self, stay_id: str) -> pd.DataFrame:
         logger.debug("Processing stay with id: %s", stay_id)
 
         datasets: list[Dataset] = [
-            Dataset(dtype, data.loc[stay_id]) for dtype, data in self._data  # type: ignore
+            Dataset(dtype, data.loc[stay_id])  # type: ignore
+            for dtype, data in self._data
+            if stay_id in data.index
         ]
 
         for probe in self._probes: