fix CI

nv-morpheus · Jul 26, 2024 · 66013af · 66013af
1 parent 8cd1a7d
commit 66013af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/examples/log_parsing/test_postprocessing.py b/tests/examples/log_parsing/test_postprocessing.py
@@ -107,7 +107,7 @@ def test_undefined_variable_error(caplog: pytest.LogCaptureFixture,
                                                              model_config_path=model_config_file)
 
     post_proc_message = build_post_proc_message(dataset_cudf, log_test_data_dir)
-    post_proc_message.payload().get_tensor('input_ids')[0] = 27716.0
+    post_proc_message.tensors().get_tensor('input_ids')[0] = 27716.0
 
     expected_log_re = re.compile(r"^Ignoring unexecpected subword token:.*")