fix: explicit assign keys for multipart

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
bentoml · Sep 27, 2022 · aeb4868 · aeb4868
1 parent 10a1a33
commit aeb4868
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bentoml/_internal/io_descriptors/multipart.py b/bentoml/_internal/io_descriptors/multipart.py
@@ -248,7 +248,7 @@ async def from_proto(self, field: pb.Multipart) -> dict[str, t.Any]:
             ) from None
         message = field.fields
         self.validate_input_mapping(message)
-        to_populate = zip(self._inputs.values(), message.values())
+        to_populate = {self._inputs[k]: message[k] for k in self._inputs}
         reqs = await asyncio.gather(
             *tuple(
                 descriptor.from_proto(
@@ -259,10 +259,10 @@ async def from_proto(self, field: pb.Multipart) -> dict[str, t.Any]:
                         ),
                     )
                 )
-                for descriptor, part in to_populate
+                for descriptor, part in to_populate.items()
             )
         )
-        return dict(zip(self._inputs, reqs))
+        return dict(zip(self._inputs.keys(), reqs))
 
     async def to_proto(self, obj: dict[str, t.Any]) -> pb.Multipart:
         self.validate_input_mapping(obj)