Fix linting issues

pytorch · Jun 6, 2024 · 3c21cc4 · 3c21cc4
1 parent e434755
commit 3c21cc4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/userbenchmark/dynamo/dynamobench/common.py b/userbenchmark/dynamo/dynamobench/common.py
@@ -3976,7 +3976,9 @@ def run(runner, args, original_dir=None):
             try:
                 from torchao_backend import setup_baseline, torchao_optimize_ctx
             except ImportError:
-                from userbenchmark.dynamo.dynamobench.torchao_backend import setup_baseline, torchao_optimize_ctx
+                from userbenchmark.dynamo.dynamobench.torchao_backend import (
+                    setup_baseline, torchao_optimize_ctx
+                )
 
             setup_baseline()
             baseline_ctx = functools.partial(