From b4c4f0022f6c94d4e15925c0cd0c84ee174f0e38 Mon Sep 17 00:00:00 2001 From: Jerin Philip Date: Tue, 15 Feb 2022 15:26:26 +0000 Subject: [PATCH] Apply formatting --- bindings/python/repository.py | 4 +++- bindings/python/test_all.py | 17 +++++------------ 2 files changed, 8 insertions(+), 13 deletions(-) diff --git a/bindings/python/repository.py b/bindings/python/repository.py index f8d6e9f10..a39ff89bf 100644 --- a/bindings/python/repository.py +++ b/bindings/python/repository.py @@ -171,7 +171,9 @@ def modelConfigPath(self, name: str, code: str) -> PathLike: ) def models(self, name: str, filter_downloaded: bool = True) -> t.List[str]: - return self.repositories.get(name, self.default_repository).models(filter_downloaded) + return self.repositories.get(name, self.default_repository).models( + filter_downloaded + ) def model(self, name: str, model_identifier: str) -> t.Any: return self.repositories.get(name, self.default_repository).model( diff --git a/bindings/python/test_all.py b/bindings/python/test_all.py index c214c3654..b91ec575a 100644 --- a/bindings/python/test_all.py +++ b/bindings/python/test_all.py @@ -1,12 +1,10 @@ import pytest - -from bergamot import REPOSITORY -from bergamot import ResponseOptions, Service, ServiceConfig, VectorString +from bergamot import REPOSITORY, ResponseOptions, Service, ServiceConfig, VectorString def test_basic(): - keys = ['browsermt'] - config = ServiceConfig(numWorkers=1, logLevel='critical') + keys = ["browsermt"] + config = ServiceConfig(numWorkers=1, logLevel="critical") service = Service(config) for repository in keys: models = REPOSITORY.models(repository, filter_downloaded=False) @@ -16,9 +14,7 @@ def test_basic(): for modelId in models: configPath = REPOSITORY.modelConfigPath(repository, modelId) model = service.modelFromConfigPath(configPath) - options = ResponseOptions( - alignment=True, qualityScores=True, HTML=False - ) + options = ResponseOptions(alignment=True, qualityScores=True, HTML=False) print(repository, modelId) source = "1 2 3 4 5 6 7 8 9" responses = service.translate(model, VectorString([source]), options) @@ -28,8 +24,5 @@ def test_basic(): print() - - - -if __name__ == '__main__': +if __name__ == "__main__": test_basic()