Lint fixes

acaklovic-nv · Oct 31, 2023 · c7cf46c · c7cf46c
1 parent 8424d5f
commit c7cf46c
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 5 deletions.
diff --git a/morpheus/llm/task_handlers/simple_task_handler.py b/morpheus/llm/task_handlers/simple_task_handler.py
@@ -28,7 +28,7 @@ class SimpleTaskHandler(LLMTaskHandler):
     Parameters
     ----------
     output_columns : list[str], optional
-        The list of columns to copy from the `LLMContext` instance to the DataFrame. If `None`, defaults to 
+        The list of columns to copy from the `LLMContext` instance to the DataFrame. If `None`, defaults to
         `["response"]`.
     """
 

diff --git a/tests/llm/task_handlers/test_simple_task_handler.py b/tests/llm/task_handlers/test_simple_task_handler.py
@@ -15,8 +15,6 @@
 
 import pytest
 
-import cudf
-
 from _utils.dataset_manager import DatasetManager
 from _utils.llm import execute_task_handler
 from morpheus._lib.messages import MessageMeta as MessageMetaCpp

diff --git a/tests/llm/task_handlers/test_simple_task_handler_pipe.py b/tests/llm/task_handlers/test_simple_task_handler_pipe.py
@@ -15,8 +15,6 @@
 
 import pytest
 
-import cudf
-
 from _utils import assert_results
 from _utils.dataset_manager import DatasetManager
 from morpheus.config import Config