Skip to content

Commit

Permalink
Renamed vars
Browse files Browse the repository at this point in the history
Signed-off-by: Adrian Boguszewski <adrian.boguszewski@intel.com>
  • Loading branch information
adrianboguszewski committed Feb 9, 2024
1 parent ff5ed2f commit cf0da5d
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions tests/integration/model/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ def models() -> list[str]:
return get_available_models()


def frameworks() -> list[ExportType]:
def export_types() -> list[ExportType]:
"""Return all available export frameworks."""
return list(ExportType)

Expand Down Expand Up @@ -121,19 +121,19 @@ def test_predict(self, model_name: str, dataset_path: Path, project_path: Path)
)

@pytest.mark.parametrize("model_name", models())
@pytest.mark.parametrize("export_framework", frameworks())
@pytest.mark.parametrize("export_framework", export_types())
def test_export(
self,
model_name: str,
export_framework: ExportType,
export_type: ExportType,
dataset_path: Path,
project_path: Path,
) -> None:
"""Export model from checkpoint.
Args:
model_name (str): Name of the model.
export_framework (ExportType): Framework to export to.
export_type (ExportType): Framework to export to.
dataset_path (Path): Root to dataset from fixture.
project_path (Path): Path to temporary project folder from fixture.
"""
Expand All @@ -158,7 +158,7 @@ def test_export(
model=model,
datamodule=dataset,
ckpt_path=f"{project_path}/{model_name}/dummy/weights/last.ckpt",
export_type=export_framework,
export_type=export_type,
input_size=input_size,
)

Expand Down

0 comments on commit cf0da5d

Please sign in to comment.