fix lint

pytorch · Sep 22, 2024 · 9ec69b1 · 9ec69b1
1 parent 4a26ee8
commit 9ec69b1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/py/torch_tensorrt/dynamo/lowering/_decompositions.py b/py/torch_tensorrt/dynamo/lowering/_decompositions.py
@@ -395,4 +395,8 @@ def get_decompositions(
         return {**CORE_ATEN_DECOMPOSITIONS_FILTERED, **TORCH_TRT_DECOMPOSITIONS}
     else:
         # changes made here due to torch2.6 changes https://github.com/pytorch/pytorch/pull/135080
-        return {**_decomp_table_to_post_autograd_aten(), **ENABLED_TORCH_DECOMPOSITIONS, **TORCH_TRT_DECOMPOSITIONS}
+        return {
+            **ENABLED_TORCH_DECOMPOSITIONS,
+            **_decomp_table_to_post_autograd_aten(),
+            **TORCH_TRT_DECOMPOSITIONS,
+        }