From d75147e0080783e778cbf32da407b9cda6ddac69 Mon Sep 17 00:00:00 2001 From: Yuchen Zhang Date: Thu, 4 Apr 2024 16:13:42 -0700 Subject: [PATCH] fix ci --- tests/_utils/stages/conv_msg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/_utils/stages/conv_msg.py b/tests/_utils/stages/conv_msg.py index 089539f8ba..950ceeaa20 100755 --- a/tests/_utils/stages/conv_msg.py +++ b/tests/_utils/stages/conv_msg.py @@ -37,7 +37,7 @@ @register_stage("unittest-conv-msg", ignore_args=["expected_data"]) class ConvMsg(SinglePortStage): """ - Simple test stage to convert a MultiMessage to a MultiResponseProbsMessage, + Simple test stage to convert a MultiMessage to a MultiResponseProbsMessage, or a ControlMessage to a ControlMessage with probs tensor. Basically a cheap replacement for running an inference stage.