We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3a1f1b8 commit 63194f5Copy full SHA for 63194f5
1 file changed
harp/io.py
@@ -219,9 +219,9 @@ def to_buffer(
219
if nrows == 0:
220
return np.empty(0, dtype=np.uint8)
221
222
- if "MessageType" in data.columns:
223
- msgtype = data["MessageType"].cat.codes
224
- payload = data[data.columns.drop("MessageType")].values
+ if MessageType.__name__ in data.columns:
+ msgtype = data[MessageType.__name__].cat.codes
+ payload = data[data.columns.drop(MessageType.__name__)].values
225
elif message_type is not None:
226
msgtype = message_type
227
payload = data.values
0 commit comments