Skip to content

Commit 5517522

Browse files
authored
ci: fix black file regex (bentoml#4048)
1 parent 4649b6f commit 5517522

File tree

7 files changed

+1896
-347
lines changed

7 files changed

+1896
-347
lines changed

.pre-commit-config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
ci:
22
autoupdate_schedule: monthly
3-
exclude: '.*\.(css|js|svg)$'
3+
exclude: '(.*\.(css|js|svg))|(.*_generated.*)$'
44
repos:
55
- repo: https://github.com/astral-sh/ruff-pre-commit
66
rev: 'v0.0.277'
@@ -11,7 +11,7 @@ repos:
1111
rev: 23.3.0
1212
hooks:
1313
- id: black-jupyter
14-
files: '/(src|tests|docs|examples|typings)/'
14+
files: '(src|tests|docs|examples|typings)/'
1515
- repo: https://github.com/pre-commit/pre-commit-hooks
1616
rev: v4.4.0
1717
hooks:

examples/tutorial/init_model.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,7 @@
55
pipe = transformers.pipeline("text-classification")
66

77
bentoml.transformers.save_model(
8-
"text-classification-pipe",
9-
pipe,
10-
signatures={
11-
"__call__": {"batchable": True} # Enable dynamic batching for model
12-
}
8+
"text-classification-pipe",
9+
pipe,
10+
signatures={"__call__": {"batchable": True}}, # Enable dynamic batching for model
1311
)

examples/tutorial/service.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
svc = bentoml.Service("text-classification-service", runners=[model_runner])
66

7+
78
@svc.api(input=bentoml.io.Text(), output=bentoml.io.JSON())
89
async def classify(text: str) -> str:
910
results = await model_runner.async_run([text])

src/bentoml/bentos.py

Lines changed: 29 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -509,22 +509,22 @@ def serve(
509509
port = t.cast(int, BentoMLContainer.http.port.get())
510510

511511
res = HTTPServer(
512-
bento = bento,
513-
reload = reload,
514-
production = production,
515-
env = env,
516-
host = host,
517-
port = port,
518-
working_dir = working_dir,
519-
api_workers = api_workers,
520-
backlog = backlog,
521-
ssl_certfile = ssl_certfile,
522-
ssl_keyfile = ssl_keyfile,
523-
ssl_keyfile_password = ssl_keyfile_password,
524-
ssl_version = ssl_version,
525-
ssl_cert_reqs = ssl_cert_reqs,
526-
ssl_ca_certs = ssl_ca_certs,
527-
ssl_ciphers = ssl_ciphers,
512+
bento=bento,
513+
reload=reload,
514+
production=production,
515+
env=env,
516+
host=host,
517+
port=port,
518+
working_dir=working_dir,
519+
api_workers=api_workers,
520+
backlog=backlog,
521+
ssl_certfile=ssl_certfile,
522+
ssl_keyfile=ssl_keyfile,
523+
ssl_keyfile_password=ssl_keyfile_password,
524+
ssl_version=ssl_version,
525+
ssl_cert_reqs=ssl_cert_reqs,
526+
ssl_ca_certs=ssl_ca_certs,
527+
ssl_ciphers=ssl_ciphers,
528528
)
529529
elif server_type == "grpc":
530530
from .server import GrpcServer
@@ -535,19 +535,19 @@ def serve(
535535
port = t.cast(int, BentoMLContainer.grpc.port.get())
536536

537537
res = GrpcServer(
538-
bento = bento,
539-
reload = reload,
540-
production = production,
541-
env = env,
542-
host = host,
543-
port = port,
544-
working_dir = working_dir,
545-
api_workers = api_workers,
546-
backlog = backlog,
547-
enable_reflection = enable_reflection,
548-
enable_channelz = enable_channelz,
549-
max_concurrent_streams = max_concurrent_streams,
550-
grpc_protocol_version = grpc_protocol_version,
538+
bento=bento,
539+
reload=reload,
540+
production=production,
541+
env=env,
542+
host=host,
543+
port=port,
544+
working_dir=working_dir,
545+
api_workers=api_workers,
546+
backlog=backlog,
547+
enable_reflection=enable_reflection,
548+
enable_channelz=enable_channelz,
549+
max_concurrent_streams=max_concurrent_streams,
550+
grpc_protocol_version=grpc_protocol_version,
551551
)
552552
else:
553553
raise BadInput(f"Unknown server type: '{server_type}'")

0 commit comments

Comments
 (0)