Run code formatting

Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
caikit · Jun 24, 2024 · 9dacb91 · 9dacb91
1 parent 147c94a
commit 9dacb91
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/examples/runtime/run_train_and_inference.py b/examples/runtime/run_train_and_inference.py
@@ -30,7 +30,7 @@
 
 # pylint: disable=no-name-in-module,import-error
 try:
-    # Third Party
+    # First Party
     from generated import (
         computervisionservice_pb2_grpc,
         computervisiontrainingservice_pb2_grpc,
@@ -41,7 +41,7 @@
 # The location of these imported message types depends on the version of Caikit
 # that we are using.
 try:
-    # Third Party
+    # First Party
     from generated.caikit_data_model.caikit_computer_vision import (
         flatchannel_pb2,
         flatimage_pb2,
@@ -58,7 +58,7 @@
     IS_LEGACY = False
 except ModuleNotFoundError:
     # older versions of Caikit / py to proto create a flat proto structure
-    # Third Party
+    # First Party
     from generated import objectdetectiontaskrequest_pb2
     from generated import (
         objectdetectiontasktransformersobjectdetectortrainrequest_pb2 as odt_request_pb2,

diff --git a/tests/modules/text_to_image/test_sdxl.py b/tests/modules/text_to_image/test_sdxl.py
@@ -13,10 +13,13 @@
 # limitations under the License.
 
 # Standard
-import os
 from tempfile import TemporaryDirectory
-import caikit_computer_vision
+import os
+
+# Local
 from caikit_computer_vision.modules.text_to_image import SDXLStub
+import caikit_computer_vision
+
 
 def test_sdxl_stub():
     """Ensure that the stubs for load / save / run work as expected."""