diff --git a/test/components/builders/test_chat_prompt_builder.py b/test/components/builders/test_chat_prompt_builder.py index a8fb8bc5b..1c39922df 100644 --- a/test/components/builders/test_chat_prompt_builder.py +++ b/test/components/builders/test_chat_prompt_builder.py @@ -5,7 +5,7 @@ import pytest from haystack.components.builders.chat_prompt_builder import ChatPromptBuilder from haystack import component from haystack.core.pipeline.pipeline import Pipeline -from haystack.dataclasses.chat_message import ChatMessage, ChatRole +from haystack.dataclasses.chat_message import ChatMessage from haystack.dataclasses.document import Document diff --git a/test/components/converters/test_csv_to_document.py b/test/components/converters/test_csv_to_document.py index cfb955041..1b4a273ec 100644 --- a/test/components/converters/test_csv_to_document.py +++ b/test/components/converters/test_csv_to_document.py @@ -2,9 +2,6 @@ # # SPDX-License-Identifier: Apache-2.0 import logging -from unittest.mock import patch -import pandas as pd -from pathlib import Path import os import pytest diff --git a/test/core/pipeline/test_component_checks.py b/test/core/pipeline/test_component_checks.py index d6dd9ffff..b98c3150e 100644 --- a/test/core/pipeline/test_component_checks.py +++ b/test/core/pipeline/test_component_checks.py @@ -4,7 +4,23 @@ import pytest -from haystack.core.pipeline.component_checks import * +from haystack.core.pipeline.component_checks import ( + all_predecessors_executed, + all_socket_predecessors_executed, + any_predecessors_provided_input, + any_socket_input_received, + any_socket_value_from_predecessor_received, + are_all_lazy_variadic_sockets_resolved, + are_all_sockets_ready, + can_component_run, + can_not_receive_inputs_from_pipeline, + has_any_trigger, + has_lazy_variadic_socket_received_all_inputs, + has_socket_received_all_inputs, + has_user_input, + is_any_greedy_socket_ready, + is_socket_lazy_variadic, +) from haystack.core.pipeline.component_checks import _NO_OUTPUT_PRODUCED from haystack.core.component.types import InputSocket, OutputSocket, Variadic, GreedyVariadic diff --git a/test/core/test_serialization.py b/test/core/test_serialization.py index fd83122c3..45fbb7c64 100644 --- a/test/core/test_serialization.py +++ b/test/core/test_serialization.py @@ -1,8 +1,6 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 -import datetime - import sys from unittest.mock import Mock diff --git a/test/evaluation/test_eval_run_result.py b/test/evaluation/test_eval_run_result.py index d59ef9b63..055b1d718 100644 --- a/test/evaluation/test_eval_run_result.py +++ b/test/evaluation/test_eval_run_result.py @@ -1,7 +1,6 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 -import json from haystack.evaluation import EvaluationRunResult import pytest