Skip to content

Commit

Permalink
add test for multimodal inference
Browse files Browse the repository at this point in the history
  • Loading branch information
Малахов Алексей Павлович committed Aug 26, 2024
1 parent e1497bd commit b0a5d58
Showing 1 changed file with 21 additions and 21 deletions.
42 changes: 21 additions & 21 deletions tests/cli/test_multimodal_inference.py
Original file line number Diff line number Diff line change
@@ -1,26 +1,26 @@
# from pathlib import Path
from pathlib import Path

# import pytest
# from typer.testing import CliRunner
import pytest
from typer.testing import CliRunner

# from tests.constants import FIXTURES_PATH
# from turbo_alignment.cli import app
# from turbo_alignment.settings.pipelines.inference.multimodal import (
# MultimodalInferenceExperimentSettings,
# )
from tests.constants import FIXTURES_PATH
from turbo_alignment.cli import app
from turbo_alignment.settings.pipelines.inference.multimodal import (
MultimodalInferenceExperimentSettings,
)

# runner = CliRunner()
runner = CliRunner()


# @pytest.mark.parametrize(
# 'config_path',
# [
# FIXTURES_PATH / 'configs/inference/multimodal/llama_llava_clip_pickle.json',
# ],
# )
# def test_multimodal_inference_mlp_with_preprocessing(config_path: Path):
# result = runner.invoke(
# app, ['inference_multimodal', '--inference_settings_path', str(config_path)], catch_exceptions=False
# )
# assert result.exit_code == 0
# assert MultimodalInferenceExperimentSettings.parse_file(config_path).save_path.is_dir()
@pytest.mark.parametrize(
'config_path',
[
FIXTURES_PATH / 'configs/inference/multimodal/llama_llava_clip_pickle.json',
],
)
def test_multimodal_inference_mlp_with_preprocessing(config_path: Path):
result = runner.invoke(
app, ['inference_multimodal', '--inference_settings_path', str(config_path)], catch_exceptions=False
)
assert result.exit_code == 0
assert MultimodalInferenceExperimentSettings.parse_file(config_path).save_path.is_dir()

0 comments on commit b0a5d58

Please sign in to comment.