Skip to content

Commit

Permalink
Chore: pre-commit autoupdate
Browse files Browse the repository at this point in the history
  • Loading branch information
pre-commit-ci[bot] committed Jul 16, 2024
1 parent 693d46f commit 13408d5
Showing 1 changed file with 4 additions and 8 deletions.
12 changes: 4 additions & 8 deletions tests/test_inference.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
"""
Test module for inference functions.
"""Test module for inference functions.
This module contains test cases for the inference functions in the
osc_transformer_based_extractor.inference module.
Expand Down Expand Up @@ -28,8 +27,7 @@


def test_check_model_and_tokenizer_path():
"""
Test the check_model_and_tokenizer_path function.
"""Test the check_model_and_tokenizer_path function.
This test verifies that the function correctly identifies valid
and invalid model and tokenizer paths.
Expand All @@ -52,8 +50,7 @@ def test_check_model_and_tokenizer_path():


def test_check_question_context():
"""
Test the check_question_context function.
"""Test the check_question_context function.
This test verifies that the function correctly handles valid and
invalid question and context inputs.
Expand Down Expand Up @@ -88,8 +85,7 @@ def test_check_question_context():
)
@patch("osc_transformer_based_extractor.inference.AutoTokenizer.from_pretrained")
def test_get_inference(mock_tokenizer, mock_model):
"""
Test the get_inference function.
"""Test the get_inference function.
This test verifies that the get_inference function correctly performs
inference using the provided model and tokenizer mocks.
Expand Down

0 comments on commit 13408d5

Please sign in to comment.