diff --git a/shark/iree_utils/compile_utils.py b/shark/iree_utils/compile_utils.py
index 9d6fc658f6..8ddb530b38 100644
--- 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(
             device_num=device_num, extra_args=extra_args
         )
diff --git a/shark/iree_utils/metal_utils.py b/shark/iree_utils/metal_utils.py
index 0bd27d71dc..dbc400edf4 100644
--- a/shark/iree_utils/metal_utils.py
+++ b/shark/iree_utils/metal_utils.py
@@ -91,6 +91,7 @@ def get_metal_triple_flag(device_name="", device_num=0, extra_args=[]):
     print(f"Target : {metal_device}")
     return None
 
+
 def get_iree_metal_args(device_num=0, extra_args=[]):
     # res_metal_flag = ["--iree-flow-demote-i64-to-i32"]
 
@@ -101,7 +102,7 @@ def get_iree_metal_args(device_num=0, extra_args=[]):
             print(f"Using target triple {arg} from command line args")
             meatal_triple_flag = arg
             break
-    
+
     if metal_triple_flag is None:
         meatal_triple_flag = get_metal_triple_flag(
             device_num=device_num, extra_args=extra_args