diff --git a/lib/model/generic_transformer.py b/lib/model/generic_transformer.py index 88a1717..c8f3fc8 100644 --- a/lib/model/generic_transformer.py +++ b/lib/model/generic_transformer.py @@ -22,7 +22,6 @@ def respond(self, docs: Union[List[schemas.Message], schemas.Message]) -> List[s """ docs = self._ensure_list(docs) self._log_docs(docs) - docs_to_process, texts_to_vectorize = self._separate_cached_docs(docs) if texts_to_vectorize: diff --git a/lib/queue/queue.py b/lib/queue/queue.py index 6a11cb0..f54a286 100644 --- a/lib/queue/queue.py +++ b/lib/queue/queue.py @@ -1,4 +1,3 @@ -import pdb import json from typing import List, Dict, Tuple import os diff --git a/lib/queue/worker.py b/lib/queue/worker.py index edc9914..4435988 100644 --- a/lib/queue/worker.py +++ b/lib/queue/worker.py @@ -1,4 +1,3 @@ -import pdb import os import time from concurrent.futures import ThreadPoolExecutor, TimeoutError diff --git a/lib/telemetry.py b/lib/telemetry.py index 40916fe..f557a47 100644 --- a/lib/telemetry.py +++ b/lib/telemetry.py @@ -1,4 +1,3 @@ -import pdb import os from opentelemetry import metrics from opentelemetry.sdk.resources import SERVICE_NAME, Resource diff --git a/test/lib/queue/test_queue.py b/test/lib/queue/test_queue.py index d30b4e9..c531722 100644 --- a/test/lib/queue/test_queue.py +++ b/test/lib/queue/test_queue.py @@ -31,7 +31,7 @@ class TestQueueWorker(unittest.TestCase): @patch('lib.helpers.get_environment_setting', return_value='us-west-1') @patch('lib.telemetry.OpenTelemetryExporter.log_execution_time') def setUp(self, mock_log_execution_time, mock_get_env_setting, mock_boto_resource): - self.model = AudioModel(None) + self.model = AudioModel() self.model.model_name = "audio" self.mock_model = MagicMock() self.queue_name_input = Queue.get_input_queue_name() @@ -234,7 +234,7 @@ def test_error_capturing_in_get_response(self, mock_cache_set, mock_cache_get): mock_cache_get.return_value = None mock_cache_set.return_value = True message_data = { - "body": {"id": 1, "callback_url": "http://example.com", "text": "This is a test"}, + "body": {"id": 1, "callback_url": "http://example.com", "text": "This is a testzzz"}, "model_name": "audio" } message = schemas.parse_message(message_data) diff --git a/test/lib/test_telemetry.py b/test/lib/test_telemetry.py index 6e577a4..48e9b3e 100644 --- a/test/lib/test_telemetry.py +++ b/test/lib/test_telemetry.py @@ -1,4 +1,3 @@ -import pdb import os import pytest from unittest.mock import patch, MagicMock