fixed linting

prompt2model/dataset_transformer/prompt_based.py

@@ -204,7 +204,6 @@ def transform_data(
         Returns:
             A tuple containing two lists: inputs and outputs.
         """
-
         task_explanation = self.generate_task_explanation(prompt_spec)
         self.plan = self.generate_plan(task_explanation, dataset, prompt_spec)
         logger.info(f"Plan created. Plan: {self.plan}")

prompt2model/utils/api_tools.py

@@ -98,7 +98,6 @@ def generate_one_completion(
             An OpenAI-like response object if there were no errors in generation.
             In case of API-specific error, Exception object is captured and returned.
         """
-        breakpoint()
         num_prompt_tokens = count_tokens_from_string(prompt)
         if self.max_tokens:
             max_tokens = self.max_tokens - num_prompt_tokens - token_buffer

tests/dataset_transformer_test.py

@@ -30,11 +30,13 @@
 )
 @patch(
     "prompt2model.utils.APIAgent.generate_one_completion",
-    side_effect=[TASK_EXPLANATION,PLAN_RESPONSE],
+    side_effect=[TASK_EXPLANATION, PLAN_RESPONSE],
 )
 def test_transform_data(mock_batch_completion, mock_one_completion):
     """Test transform_data."""
-    dataset_transformer = PromptBasedDatasetTransformer(num_points_to_transform=1, max_allowed_failed_transforms=0) # noqa: E501
+    dataset_transformer = PromptBasedDatasetTransformer(
+        num_points_to_transform=1, max_allowed_failed_transforms=0
+    )  # noqa: E501
     prompt_spec = MockPromptSpec(
         TaskType.TEXT_GENERATION,
         instruction="instruction",
@@ -56,11 +58,8 @@ def test_transform_data(mock_batch_completion, mock_one_completion):
     inputs, outputs = dataset_transformer.transform_data(
         prompt_spec=prompt_spec,
         dataset=dataset["train"],
-
-    )
-    breakpoint()
-    assert (
-        inputs
-        == ["context: Albany, the state capital, is the sixth-largest city in the State of New York.\nquestion: What is the capital of New York?"]  # noqa E501
     )
+    assert inputs == [
+        "context: Albany, the state capital, is the sixth-largest city in the State of New York.\nquestion: What is the capital of New York?"  # noqa E501
+    ]  # noqa E501
     assert outputs == ["Albany"]