fixed for black lint formating

nod-ai · Jun 26, 2023 · bae93bd · bae93bd
1 parent f52feb2
commit bae93bd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/shark/iree_utils/compile_utils.py b/shark/iree_utils/compile_utils.py
@@ -51,7 +51,7 @@ def get_iree_device_args(device, extra_args=[]):
         )
     if device_uri[0] == "metal":
         from shark.iree_utils.metal_utils import get_iree_metal_args
-        
+
         return get_iree_metal_args(extra_args=extra_args)
     if device_uri[0] == "rocm":
         from shark.iree_utils.gpu_utils import get_iree_rocm_args

diff --git a/shark/iree_utils/metal_utils.py b/shark/iree_utils/metal_utils.py
@@ -100,7 +100,9 @@ def get_iree_metal_args(device_num=0, extra_args=[]):
         metal_triple_flag = get_metal_triple_flag(extra_args=extra_args)
 
     if metal_triple_flag is not None:
-        vulkan_target_env = get_vulkan_target_env_flag("-iree-vulkan-target-triple=m1-moltenvk-macos")
+        vulkan_target_env = get_vulkan_target_env_flag(
+            "-iree-vulkan-target-triple=m1-moltenvk-macos"
+        )
         res_metal_flag.append(vulkan_target_env)
     return res_metal_flag