diff --git a/test/fixtures/image_metadata_pair.py b/test/fixtures/image_metadata_pair.py index 4dc1acf..9b4b916 100644 --- a/test/fixtures/image_metadata_pair.py +++ b/test/fixtures/image_metadata_pair.py @@ -7,4 +7,4 @@ from image_prediction.image_extractor.extractor import ImageMetadataPair @pytest.fixture def image_metadata_pairs(images, metadata): - return list(starmap(ImageMetadataPair, zip(images, metadata))) \ No newline at end of file + return list(starmap(ImageMetadataPair, zip(images, metadata))) diff --git a/test/fixtures/input.py b/test/fixtures/input.py index b50274a..b02f414 100644 --- a/test/fixtures/input.py +++ b/test/fixtures/input.py @@ -4,4 +4,4 @@ import pytest @pytest.fixture def input_batch(batch_size, input_size): - return np.random.random_sample(size=(batch_size, *input_size)) \ No newline at end of file + return np.random.random_sample(size=(batch_size, *input_size)) diff --git a/test/fixtures/label.py b/test/fixtures/label.py index 2b61e47..571ce23 100644 --- a/test/fixtures/label.py +++ b/test/fixtures/label.py @@ -22,4 +22,4 @@ def label_format(request): @pytest.fixture def classes(): - return ["A", "B", "C"] \ No newline at end of file + return ["A", "B", "C"] diff --git a/test/fixtures/model.py b/test/fixtures/model.py index dbeee5b..57968a3 100644 --- a/test/fixtures/model.py +++ b/test/fixtures/model.py @@ -124,4 +124,4 @@ def model_handle_mock(estimator_mock): def predict_proba(self, batch): return [None for _ in batch] - return ModelHandleMock() \ No newline at end of file + return ModelHandleMock() diff --git a/test/fixtures/model_store.py b/test/fixtures/model_store.py index bc71634..97e1139 100644 --- a/test/fixtures/model_store.py +++ b/test/fixtures/model_store.py @@ -58,4 +58,4 @@ def mlruns_dir(): @pytest.fixture def mlflow_reader(mlruns_dir): - return MlflowModelReader(mlruns_dir) \ No newline at end of file + return MlflowModelReader(mlruns_dir) diff --git a/test/fixtures/parameters.py b/test/fixtures/parameters.py index 7a08b9d..a837613 100644 --- a/test/fixtures/parameters.py +++ b/test/fixtures/parameters.py @@ -35,4 +35,4 @@ def width(request): @pytest.fixture(params=[0, 1, 2, 16, 32]) def batch_size(request): - return request.param \ No newline at end of file + return request.param diff --git a/test/fixtures/pdf.py b/test/fixtures/pdf.py index 1cc4f77..7353917 100644 --- a/test/fixtures/pdf.py +++ b/test/fixtures/pdf.py @@ -20,4 +20,4 @@ def pdf(image_metadata_pairs): @pytest.fixture def real_pdf(): with open(os.path.join(TEST_DATA_DIR, "f2dc689ca794fccb8cd38b95f2bf6ba9.pdf"), "rb") as f: - yield f.read() \ No newline at end of file + yield f.read() diff --git a/test/fixtures/target.py b/test/fixtures/target.py index c376579..bcc0e75 100644 --- a/test/fixtures/target.py +++ b/test/fixtures/target.py @@ -88,4 +88,4 @@ def expected_predictions_mapped_and_formatted(expected_predictions_mapped): @pytest.fixture def real_expected_service_response(): with open(os.path.join(TEST_DATA_DIR, "f2dc689ca794fccb8cd38b95f2bf6ba9_predictions.json"), "r") as f: - yield json.load(f) \ No newline at end of file + yield json.load(f) diff --git a/test/utils/label.py b/test/utils/label.py index 0e5babc..b3fe2d9 100644 --- a/test/utils/label.py +++ b/test/utils/label.py @@ -1,2 +1,2 @@ def map_labels(numeric_labels, classes): - return [classes[nl] for nl in numeric_labels] \ No newline at end of file + return [classes[nl] for nl in numeric_labels] diff --git a/test/utils/metadata.py b/test/utils/metadata.py index 92bd58d..cf67451 100644 --- a/test/utils/metadata.py +++ b/test/utils/metadata.py @@ -8,4 +8,4 @@ def get_base_position_metadata(width, height, page_width, page_height): Info.PAGE_IDX: 0, Info.PAGE_WIDTH: page_width, Info.PAGE_HEIGHT: page_height, - } \ No newline at end of file + }