From 6aca092bfc44520e56b607aed8b06ef5d1f1c2d4 Mon Sep 17 00:00:00 2001
From: Aaron <29749331+aarnphm@users.noreply.github.com>
Date: Sat, 27 May 2023 23:42:06 -0700
Subject: [PATCH] infra: prepare for 0.0.8 release

Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
---
 src/openllm/__init__.py                | 7 +++----
 src/openllm/models/chatglm/__init__.py | 6 ++----
 src/openllm/models/falcon/__init__.py  | 6 ++----
 src/openllm/utils/__init__.py          | 7 +++----
 4 files changed, 10 insertions(+), 16 deletions(-)

diff --git a/src/openllm/__init__.py b/src/openllm/__init__.py
index 93e1fa30..d0bb9465 100644
--- a/src/openllm/__init__.py
+++ b/src/openllm/__init__.py
@@ -117,6 +117,7 @@ if t.TYPE_CHECKING:
     from . import client as client
     from . import exceptions as exceptions
     from . import models as models
+
     # Specific types import
     from ._configuration import LLMConfig as LLMConfig
     from ._llm import LLM as LLM
@@ -172,8 +173,7 @@ if t.TYPE_CHECKING:
         from .utils.dummy_flax_objects import *
     else:
         from .models.auto import MODEL_FLAX_MAPPING as MODEL_FLAX_MAPPING
-        from .models.auto import \
-            MODEL_FLAX_MAPPING_NAMES as MODEL_FLAX_MAPPING_NAMES
+        from .models.auto import MODEL_FLAX_MAPPING_NAMES as MODEL_FLAX_MAPPING_NAMES
         from .models.auto import AutoFlaxLLM as AutoFlaxLLM
         from .models.flan_t5 import FlaxFlanT5 as FlaxFlanT5
 
@@ -184,8 +184,7 @@ if t.TYPE_CHECKING:
         from .utils.dummy_tf_objects import *
     else:
         from .models.auto import MODEL_TF_MAPPING as MODEL_TF_MAPPING
-        from .models.auto import \
-            MODEL_TF_MAPPING_NAMES as MODEL_TF_MAPPING_NAMES
+        from .models.auto import MODEL_TF_MAPPING_NAMES as MODEL_TF_MAPPING_NAMES
         from .models.auto import AutoTFLLM as AutoTFLLM
         from .models.flan_t5 import TFFlanT5 as TFFlanT5
 
diff --git a/src/openllm/models/chatglm/__init__.py b/src/openllm/models/chatglm/__init__.py
index a7fc678b..504aa1c4 100644
--- a/src/openllm/models/chatglm/__init__.py
+++ b/src/openllm/models/chatglm/__init__.py
@@ -31,10 +31,8 @@ else:
     _import_structure["modeling_chatglm"] = ["ChatGLM"]
 
 if t.TYPE_CHECKING:
-    from .configuration_chatglm import \
-        DEFAULT_PROMPT_TEMPLATE as DEFAULT_PROMPT_TEMPLATE
-    from .configuration_chatglm import \
-        START_CHATGLM_COMMAND_DOCSTRING as START_CHATGLM_COMMAND_DOCSTRING
+    from .configuration_chatglm import DEFAULT_PROMPT_TEMPLATE as DEFAULT_PROMPT_TEMPLATE
+    from .configuration_chatglm import START_CHATGLM_COMMAND_DOCSTRING as START_CHATGLM_COMMAND_DOCSTRING
     from .configuration_chatglm import ChatGLMConfig as ChatGLMConfig
 
     try:
diff --git a/src/openllm/models/falcon/__init__.py b/src/openllm/models/falcon/__init__.py
index f7c2481b..9f53b4a5 100644
--- a/src/openllm/models/falcon/__init__.py
+++ b/src/openllm/models/falcon/__init__.py
@@ -31,10 +31,8 @@ else:
     _import_structure["modeling_falcon"] = ["Falcon"]
 
 if t.TYPE_CHECKING:
-    from .configuration_falcon import \
-        DEFAULT_PROMPT_TEMPLATE as DEFAULT_PROMPT_TEMPLATE
-    from .configuration_falcon import \
-        START_FALCON_COMMAND_DOCSTRING as START_FALCON_COMMAND_DOCSTRING
+    from .configuration_falcon import DEFAULT_PROMPT_TEMPLATE as DEFAULT_PROMPT_TEMPLATE
+    from .configuration_falcon import START_FALCON_COMMAND_DOCSTRING as START_FALCON_COMMAND_DOCSTRING
     from .configuration_falcon import FalconConfig as FalconConfig
 
     try:
diff --git a/src/openllm/utils/__init__.py b/src/openllm/utils/__init__.py
index 83808378..d7752322 100644
--- a/src/openllm/utils/__init__.py
+++ b/src/openllm/utils/__init__.py
@@ -29,15 +29,14 @@ import typing as t
 import bentoml
 import inflection
 from bentoml._internal.types import LazyType as LazyType
+
 # NOTE: The following exports useful utils from bentoml
 from bentoml._internal.utils import LazyLoader as LazyLoader
 from bentoml._internal.utils import bentoml_cattr as bentoml_cattr
-from bentoml._internal.utils import \
-    copy_file_to_fs_folder as copy_file_to_fs_folder
+from bentoml._internal.utils import copy_file_to_fs_folder as copy_file_to_fs_folder
 from bentoml._internal.utils import pkg as pkg
 from bentoml._internal.utils import reserve_free_port as reserve_free_port
-from bentoml._internal.utils import \
-    resolve_user_filepath as resolve_user_filepath
+from bentoml._internal.utils import resolve_user_filepath as resolve_user_filepath
 
 from . import analytics as analytics
 from . import codegen as codegen