fix

Signed-off-by: Cruz Zhao <[email protected]>

Author: zxpdemonio

mooncake-integration/store/store_py.cpp

@@ -227,10 +227,9 @@ class MooncakeStorePyWrapper {
             py::gil_scoped_acquire acquire_gil;
             auto torch = torch_module();
 
-	    int i = 0;
-            for (const auto &buffer_handle : buffer_handles) {
+            for (const auto &key : keys) {
 		uint64_t total_length = 0;
-                auto get_result = store_->get_allocated_internal(keys[i++], total_length);
+                auto get_result = store_->get_allocated_internal(key, total_length);
                 if (!get_result) {
                     results_list.append(py::none());
                     continue;