diff --git a/src/serve.py b/src/serve.py index 37a7906..16ac4ea 100644 --- a/src/serve.py +++ b/src/serve.py @@ -10,7 +10,6 @@ from image_prediction.utils.banner import show_banner def main(): - def predict(pdf): # Keras service_estimator.predict stalls when service_estimator was loaded in different process; # therefore, we re-load the model (part of the pipeline) every time we process a new document. diff --git a/test/conftest.py b/test/conftest.py index 65298b0..807551b 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -17,7 +17,7 @@ pytest_plugins = [ "test.fixtures.parameters", "test.fixtures.pdf", "test.fixtures.target", - "test.unit_tests.image_stitching_test" + "test.unit_tests.image_stitching_test", ] diff --git a/test/exploration_tests/funcy_test.py b/test/exploration_tests/funcy_test.py index 30c2cef..fc32538 100644 --- a/test/exploration_tests/funcy_test.py +++ b/test/exploration_tests/funcy_test.py @@ -3,7 +3,7 @@ from funcy import rcompose, chunks def test_rcompose(): - f = rcompose(lambda x: x ** 2, str, lambda x: x * 2) + f = rcompose(lambda x: x**2, str, lambda x: x * 2) assert f(3) == "99"