Skip to content

Commit 63194f5

Browse files
committed
Use message type enum name for symmetry
1 parent 3a1f1b8 commit 63194f5

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

harp/io.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -219,9 +219,9 @@ def to_buffer(
219219
if nrows == 0:
220220
return np.empty(0, dtype=np.uint8)
221221

222-
if "MessageType" in data.columns:
223-
msgtype = data["MessageType"].cat.codes
224-
payload = data[data.columns.drop("MessageType")].values
222+
if MessageType.__name__ in data.columns:
223+
msgtype = data[MessageType.__name__].cat.codes
224+
payload = data[data.columns.drop(MessageType.__name__)].values
225225
elif message_type is not None:
226226
msgtype = message_type
227227
payload = data.values

0 commit comments

Comments
 (0)