fixed key error loading device detector

SENERGY-Platform · Apr 24, 2024 · 402543d · 402543d
1 parent 990c294
commit 402543d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/algo/operator.py b/algo/operator.py
@@ -73,7 +73,7 @@ def init(self, *args, **kwargs):
         self.device_detectors = {} 
 
     def get_device_detectors(self, input_ids):
-        device_detector = self.device_detectors[input_ids]
+        device_detector = self.device_detectors.get(input_ids)
         if device_detector:
             return device_detector