Skip to content

Commit

Permalink
Updated test_main.py
Browse files Browse the repository at this point in the history
  • Loading branch information
RashmikaReddy authored Jan 18, 2024
1 parent 3df0c8d commit 9fcd8ec
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/test_main.py
Original file line number Diff line number Diff line change
@@ -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"

Expand Down

0 comments on commit 9fcd8ec

Please sign in to comment.