From 40a9e8464c67c2032ba5718a6bc1763690a28be4 Mon Sep 17 00:00:00 2001
From: David Perl <david.perl@psi.ch>
Date: Tue, 11 Aug 2026 10:09:01 +0200
Subject: [PATCH] fix: kwarg name for mlbox klass

---
 src/aare/daq/mlbox.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/aare/daq/mlbox.py b/src/aare/daq/mlbox.py
index 33621bc3..60916a4e 100644
--- a/src/aare/daq/mlbox.py
+++ b/src/aare/daq/mlbox.py
@@ -359,7 +359,7 @@ class MlBox:
                 logger.debug(f"Failed to parse prediction: {e}", exc_info=True)
                 continue
 
-            base_key = MLOutputModel.get_class_str(cls=cls)
+            base_key = MLOutputModel.get_class_str(klass=cls)
             if base_key not in out.boxes:
                 out.boxes[base_key] = MLBoxModel(
                     cls=cls,