[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
pyg-team · Jul 10, 2023 · b14c947 · b14c947
1 parent a69eefa
commit b14c947
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/benchmark/inference/inference_benchmark.py b/benchmark/inference/inference_benchmark.py
@@ -49,8 +49,8 @@ def run(args: argparse.ArgumentParser):
         except ImportError:
             raise RuntimeError('XPU device requires IPEX to be installed')
 
-    if ((args.device == 'cuda' and not torch.cuda.is_available()) or
-        (args.device == 'xpu' and not torch.xpu.is_available())):
+    if ((args.device == 'cuda' and not torch.cuda.is_available())
+            or (args.device == 'xpu' and not torch.xpu.is_available())):
         raise RuntimeError(f'{args.device.upper()} is not available')
 
     if args.device == 'cuda' and args.full_batch:
@@ -195,7 +195,7 @@ def run(args: argparse.ArgumentParser):
                             profile = xpu_profiler(args.export_chrome_trace)
                         elif args.profile:
                             profile = torch_profile(args.export_chrome_trace,
-                                csv_data, args.write_csv)
+                                                    csv_data, args.write_csv)
                         else:
                             profile = nullcontext()
                         itt = emit_itt(

diff --git a/benchmark/utils/utils.py b/benchmark/utils/utils.py
@@ -19,6 +19,7 @@
 try:
     from torch.autograd.profiler import emit_itt
 except ImportError:
+
     @contextmanager
     def emit_itt(*args, **kwargs):
         yield