BUGFIX for issue 543

robinhood#543 Passes key_type to derive if key has "type" attribute.
pscottdevos · Dec 26, 2021 · e15375a · e15375a
1 parent 01b4c0a
commit e15375a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/faust/streams.py b/faust/streams.py
@@ -582,7 +582,12 @@ def get_key(withdrawal):
             suffix = f'-{name}-repartition'
             p = partitions if partitions else self.app.conf.topic_partitions
             channel = cast(ChannelT, self.channel).derive(
-                prefix=prefix, suffix=suffix, partitions=p, internal=True)
+                key_type=key.type if hasattr(key, 'type') else None,
+                prefix=prefix,
+                suffix=suffix,
+                partitions=p,
+                internal=True,
+            )
         format_key = self._format_key
 
         channel_it = aiter(channel)