fix(yapf): align weird new lines break [generated] [skip ci] (#284)

fix(yapf): align weird new lines break

Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>

openllm-python/src/openllm/cli/extension/build_base_container.py

@@ -34,8 +34,8 @@ if t.TYPE_CHECKING:
               help='Version strategy to use for tagging the image.')
 @click.option('--push/--no-push', help='Whether to push to remote repository', is_flag=True, default=False)
 @machine_option
-def cli(container_registry: tuple[LiteralContainerRegistry, ...] | None,
-        version_strategy: LiteralContainerVersionStrategy, push: bool, machine: bool) -> dict[str, str]:
+def cli(container_registry: tuple[LiteralContainerRegistry, ...] | None, version_strategy: LiteralContainerVersionStrategy, push: bool,
+        machine: bool) -> dict[str, str]:
   mapping = openllm.bundle.build_container(container_registry, version_strategy, push, machine)
   if machine: termui.echo(orjson.dumps(mapping, option=orjson.OPT_INDENT_2).decode(), fg='white')
   return mapping

openllm-python/src/openllm/cli/extension/dive_bentos.py

@@ -24,10 +24,7 @@ if t.TYPE_CHECKING:
 @machine_option
 @click.pass_context
 @inject
-def cli(ctx: click.Context,
-        bento: str,
-        machine: bool,
-        _bento_store: BentoStore = Provide[BentoMLContainer.bento_store]) -> str | None:
+def cli(ctx: click.Context, bento: str, machine: bool, _bento_store: BentoStore = Provide[BentoMLContainer.bento_store]) -> str | None:
   '''Dive into a BentoLLM. This is synonymous to cd $(b get <bento>:<tag> -o path).'''
   try:
     bentomodel = _bento_store.get(bento)

openllm-python/src/openllm/cli/extension/get_containerfile.py

@@ -19,9 +19,7 @@ from openllm_core.utils import bentoml_cattr
 if t.TYPE_CHECKING:
   from bentoml._internal.bento import BentoStore
 
-@click.command('get_containerfile',
-               context_settings=termui.CONTEXT_SETTINGS,
-               help='Return Containerfile of any given Bento.')
+@click.command('get_containerfile', context_settings=termui.CONTEXT_SETTINGS, help='Return Containerfile of any given Bento.')
 @click.argument('bento', type=str, shell_complete=bento_complete_envvar)
 @click.pass_context
 @inject

openllm-python/src/openllm/cli/extension/get_prompt.py

@@ -32,8 +32,8 @@ LiteralOutput = t.Literal['json', 'pretty', 'porcelain']
               callback=opt_callback,
               metavar='ARG=VALUE[,ARG=VALUE]')
 @click.pass_context
-def cli(ctx: click.Context, /, model_name: str, prompt: str, format: str | None, output: LiteralOutput, machine: bool,
-        _memoized: dict[str, t.Any], **_: t.Any) -> str | None:
+def cli(ctx: click.Context, /, model_name: str, prompt: str, format: str | None, output: LiteralOutput, machine: bool, _memoized: dict[str, t.Any],
+        **_: t.Any) -> str | None:
   '''Get the default prompt used by OpenLLM.'''
   module = openllm.utils.EnvVarMixin(model_name).module
   _memoized = {k: v[0] for k, v in _memoized.items() if v}
@@ -46,15 +46,11 @@ def cli(ctx: click.Context, /, model_name: str, prompt: str, format: str | None,
       if format is None:
         if not hasattr(module, 'PROMPT_MAPPING') or module.PROMPT_MAPPING is None:
           raise RuntimeError('Failed to find prompt mapping while DEFAULT_PROMPT_TEMPLATE is a function.')
-        raise click.BadOptionUsage(
-            'format',
-            f"{model_name} prompt requires passing '--format' (available format: {list(module.PROMPT_MAPPING)})")
+        raise click.BadOptionUsage('format', f"{model_name} prompt requires passing '--format' (available format: {list(module.PROMPT_MAPPING)})")
       if prompt_mapping is None:
-        raise click.BadArgumentUsage(
-            f'Failed to fine prompt mapping while the default prompt for {model_name} is a callable.') from None
+        raise click.BadArgumentUsage(f'Failed to fine prompt mapping while the default prompt for {model_name} is a callable.') from None
       if format not in prompt_mapping:
-        raise click.BadOptionUsage(
-            'format', f'Given format {format} is not valid for {model_name} (available format: {list(prompt_mapping)})')
+        raise click.BadOptionUsage('format', f'Given format {format} is not valid for {model_name} (available format: {list(prompt_mapping)})')
       _prompt_template = template(format)
     else:
       _prompt_template = template

openllm-python/src/openllm/cli/extension/list_bentos.py

@@ -19,26 +19,28 @@ def cli(ctx: click.Context, output: LiteralOutput) -> None:
   '''List available bentos built by OpenLLM.'''
   mapping = {
       k: [{
-          'tag':
-              str(b.tag),
-          'size':
-              human_readable_size(openllm.utils.calc_dir_size(b.path)),
+          'tag': str(b.tag),
+          'size': human_readable_size(openllm.utils.calc_dir_size(b.path)),
           'models': [{
               'tag': str(m.tag),
               'size': human_readable_size(openllm.utils.calc_dir_size(m.path))
-          } for m in (bentoml.models.get(_.tag) for _ in b.info.models)]
-      } for b in tuple(i for i in bentoml.list() if all(
-          k in i.info.labels for k in {'start_name', 'bundler'})) if b.info.labels['start_name'] == k
-         ] for k in tuple(inflection.dasherize(key) for key in openllm.CONFIG_MAPPING.keys())
+          }
+                     for m in (bentoml.models.get(_.tag)
+                               for _ in b.info.models)]
+      }
+          for b in tuple(i
+                         for i in bentoml.list()
+                         if all(k in i.info.labels
+                                for k in {'start_name', 'bundler'}))
+          if b.info.labels['start_name'] == k] for k in tuple(inflection.dasherize(key) for key in openllm.CONFIG_MAPPING.keys())
   }
   mapping = {k: v for k, v in mapping.items() if v}
   if output == 'pretty':
     import tabulate
     tabulate.PRESERVE_WHITESPACE = True
-    termui.echo(tabulate.tabulate(
-        [(k, i['tag'], i['size'], [_['tag'] for _ in i['models']]) for k, v in mapping.items() for i in v],
-        tablefmt='fancy_grid',
-        headers=['LLM', 'Tag', 'Size', 'Models']),
+    termui.echo(tabulate.tabulate([(k, i['tag'], i['size'], [_['tag'] for _ in i['models']]) for k, v in mapping.items() for i in v],
+                                  tablefmt='fancy_grid',
+                                  headers=['LLM', 'Tag', 'Size', 'Models']),
                 fg='white')
   else:
     termui.echo(orjson.dumps(mapping, option=orjson.OPT_INDENT_2).decode(), fg='white')

openllm-python/src/openllm/cli/extension/list_models.py

@@ -26,17 +26,14 @@ def cli(model_name: str | None, output: LiteralOutput) -> DictStrAny:
   models = tuple(inflection.dasherize(key) for key in openllm.CONFIG_MAPPING.keys())
   ids_in_local_store = {
       k: [
-          i for i in bentoml.models.list() if 'framework' in i.info.labels and
-          i.info.labels['framework'] == 'openllm' and 'model_name' in i.info.labels and i.info.labels['model_name'] == k
+          i for i in bentoml.models.list() if 'framework' in i.info.labels and i.info.labels['framework'] == 'openllm' and
+          'model_name' in i.info.labels and i.info.labels['model_name'] == k
       ] for k in models
   }
   if model_name is not None:
     ids_in_local_store = {
-        k: [
-            i
-            for i in v
-            if 'model_name' in i.info.labels and i.info.labels['model_name'] == inflection.dasherize(model_name)
-        ] for k, v in ids_in_local_store.items()
+        k: [i for i in v if 'model_name' in i.info.labels and i.info.labels['model_name'] == inflection.dasherize(model_name)]
+        for k, v in ids_in_local_store.items()
     }
   ids_in_local_store = {k: v for k, v in ids_in_local_store.items() if v}
   local_models = {

openllm-python/src/openllm/cli/extension/playground.py

@@ -34,12 +34,7 @@ def load_notebook_metadata() -> DictStrAny:
 
 @click.command('playground', context_settings=termui.CONTEXT_SETTINGS)
 @click.argument('output-dir', default=None, required=False)
-@click.option('--port',
-              envvar='JUPYTER_PORT',
-              show_envvar=True,
-              show_default=True,
-              default=8888,
-              help='Default port for Jupyter server')
+@click.option('--port', envvar='JUPYTER_PORT', show_envvar=True, show_default=True, default=8888, help='Default port for Jupyter server')
 @click.pass_context
 def cli(ctx: click.Context, output_dir: str | None, port: int) -> None:
   """OpenLLM Playground.
@@ -60,9 +55,7 @@ def cli(ctx: click.Context, output_dir: str | None, port: int) -> None:
   > This command requires Jupyter to be installed. Install it with 'pip install "openllm[playground]"'
   """
   if not is_jupyter_available() or not is_jupytext_available() or not is_notebook_available():
-    raise RuntimeError(
-        "Playground requires 'jupyter', 'jupytext', and 'notebook'. Install it with 'pip install \"openllm[playground]\"'"
-    )
+    raise RuntimeError("Playground requires 'jupyter', 'jupytext', and 'notebook'. Install it with 'pip install \"openllm[playground]\"'")
   metadata = load_notebook_metadata()
   _temp_dir = False
   if output_dir is None:
@@ -74,8 +67,7 @@ def cli(ctx: click.Context, output_dir: str | None, port: int) -> None:
   termui.echo('The playground notebooks will be saved to: ' + os.path.abspath(output_dir), fg='blue')
   for module in pkgutil.iter_modules(playground.__path__):
     if module.ispkg or os.path.exists(os.path.join(output_dir, module.name + '.ipynb')):
-      logger.debug('Skipping: %s (%s)', module.name,
-                   'File already exists' if not module.ispkg else f'{module.name} is a module')
+      logger.debug('Skipping: %s (%s)', module.name, 'File already exists' if not module.ispkg else f'{module.name} is a module')
       continue
     if not isinstance(module.module_finder, importlib.machinery.FileFinder): continue
     termui.echo('Generating notebook for: ' + module.name, fg='magenta')
@@ -84,10 +76,7 @@ def cli(ctx: click.Context, output_dir: str | None, port: int) -> None:
     f.cells.insert(0, markdown_cell)
     jupytext.write(f, os.path.join(output_dir, module.name + '.ipynb'), fmt='notebook')
   try:
-    subprocess.check_output([
-        sys.executable, '-m', 'jupyter', 'notebook', '--notebook-dir', output_dir, '--port',
-        str(port), '--no-browser', '--debug'
-    ])
+    subprocess.check_output([sys.executable, '-m', 'jupyter', 'notebook', '--notebook-dir', output_dir, '--port', str(port), '--no-browser', '--debug'])
   except subprocess.CalledProcessError as e:
     termui.echo(e.output, fg='red')
     raise click.ClickException(f'Failed to start a jupyter server:\n{e}') from None