fix bug

pyAKI/probes.py

@@ -305,7 +305,7 @@ def probe(self, df: pd.DataFrame = None, **kwargs) -> pd.DataFrame:
         df = df.copy()
         df[self.RESNAME] = 0
         df.loc[(df[self._column] - baseline_values) >= 0.3, self.RESNAME] = 1
-        df.loc[(df[self._column] - baseline_values) >= 4, self.RESNAME] = 3
+        df.loc[df[self._column] >= 4, self.RESNAME] = 3
 
         df.loc[df[self._column] == 0, self.RESNAME] = None
         df[self.RESNAME] = df[self.RESNAME].ffill().fillna(0)

test_machine_sec.py

@@ -0,0 +1,24 @@
+import pyAKI.kdigo
+import pandas as pd
+
+if __name__ == "__main__":
+    # Load the test data
+    data = pd.read_excel("test_aki_combined_prep.xlsx")
+
+    urine_output = data["urineoutput"]
+    creatinine = data["creat"]
+    weight = data["weight"]
+    crrt = data["crrt_status"]
+    # prep weights
+    weight = weight.drop_duplicates(subset=["weight"], keep="first")
+
+    ana = pyAKI.kdigo.Analyser(
+        [
+            pyAKI.utils.Dataset(pyAKI.utils.DatasetType.URINEOUTPUT, urine_output),
+            pyAKI.utils.Dataset(pyAKI.utils.DatasetType.CREATININE, creatinine),
+            pyAKI.utils.Dataset(pyAKI.utils.DatasetType.DEMOGRAPHICS, weight),
+            pyAKI.utils.Dataset(pyAKI.utils.DatasetType.CRRT, crrt),
+        ]
+    )
+    ana.process_stays().to_csv("tests/data/test_machine_aki_second.csv")
+    ana.process_stays().to_excel("tests/data/test_machine_aki_second.xlsx")