From 9fcd8ec9bfff87ddd8527a0ab68bfb58761a9227 Mon Sep 17 00:00:00 2001 From: Rashmika Reddy Vookanti Date: Thu, 18 Jan 2024 11:33:14 -0800 Subject: [PATCH] Updated test_main.py --- tests/test_main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_main.py b/tests/test_main.py index b325c53..a1eed5f 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1,11 +1,11 @@ from pathlib import Path -from autora.doc.pipelines.main import eval, evaluate_documentation, generate, import_data -from autora.doc.runtime.prompts import InstructionPrompts, SystemPrompts - import jsonlines import pytest +from autora.doc.pipelines.main import eval, evaluate_documentation, generate, import_data +from autora.doc.runtime.prompts import InstructionPrompts, SystemPrompts + # dummy HF model for testing TEST_HF_MODEL = "hf-internal-testing/tiny-random-FalconForCausalLM"