diff --git a/python/morpheus/morpheus/stages/preprocess/preprocess_nlp_stage.py b/python/morpheus/morpheus/stages/preprocess/preprocess_nlp_stage.py index 25a1284337..0ee30c100b 100644 --- a/python/morpheus/morpheus/stages/preprocess/preprocess_nlp_stage.py +++ b/python/morpheus/morpheus/stages/preprocess/preprocess_nlp_stage.py @@ -203,7 +203,6 @@ def process_control_message(message: ControlMessage, del text_series - # We need the C++ impl of TensorMemory until #1646 is resolved message.tensors(count=tokenized.input_ids.shape[0], tensors={ "input_ids": tokenized.input_ids,