[python] Fixes batch header key issue (deepjavalibrary#1006)

Lokiiiiii · Aug 9, 2023 · a7b7c2d · a7b7c2d
1 parent 9885cd2
commit a7b7c2d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/engines/python/setup/djl_python/inputs.py b/engines/python/setup/djl_python/inputs.py
@@ -96,12 +96,11 @@ def get_batches(self) -> list:
         batch = []
         for i in range(batch_size):
             item = Input()
-            item.properties = {}
             prefix = f"batch_{i}."
             length = len(prefix)
             for key, value in self.properties.items():
                 if key.startswith(prefix):
-                    key = key[:length]
+                    key = key[length:]
                     item.properties[key] = value
 
             batch.append(item)