From dfba0af6b12711af00a4bfbc0e6fd1c6c5966756 Mon Sep 17 00:00:00 2001 From: Alex-Brooks Date: Wed, 17 Jan 2024 15:10:07 -0700 Subject: [PATCH] code formatting Signed-off-by: Alex-Brooks --- examples/runtime/dump_protos.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/examples/runtime/dump_protos.py b/examples/runtime/dump_protos.py index 07dc89c..d4b66e9 100644 --- a/examples/runtime/dump_protos.py +++ b/examples/runtime/dump_protos.py @@ -16,6 +16,7 @@ """ # Standard +from inspect import signature from pathlib import Path from shutil import rmtree import json @@ -23,7 +24,6 @@ import shutil import sys import tempfile -from inspect import signature # Third Party from common import PROTO_EXPORT_DIR, RUNTIME_CONFIG @@ -45,11 +45,13 @@ def export_protos(): # grpc service dumper kwargs depend on the version of caikit we are using grpc_service_dumper_kwargs = { "output_dir": PROTO_EXPORT_DIR, - "write_modules_file": True + "write_modules_file": True, } # Only keep things in the signature, e.g., old versions don't take write_modules_file expected_grpc_params = signature(dump_grpc_services).parameters - grpc_service_dumper_kwargs = {k:v for k, v in grpc_service_dumper_kwargs.items() if k in expected_grpc_params} + grpc_service_dumper_kwargs = { + k: v for k, v in grpc_service_dumper_kwargs.items() if k in expected_grpc_params + } dump_grpc_services(**grpc_service_dumper_kwargs) # Then export HTTP services... dump_http_services(output_dir=PROTO_EXPORT_DIR)