diff --git a/tritonparse/reproducer/multi_file_analyzer.py b/tritonparse/reproducer/multi_file_analyzer.py index ba94b9e..88bbceb 100644 --- a/tritonparse/reproducer/multi_file_analyzer.py +++ b/tritonparse/reproducer/multi_file_analyzer.py @@ -242,9 +242,11 @@ def _analyze_file( for resolved_path, imported_names in imports_by_file.items(): already_pending = any( - pending_item[0] == resolved_path - if isinstance(pending_item, tuple) - else pending_item == resolved_path + ( + pending_item[0] == resolved_path + if isinstance(pending_item, tuple) + else pending_item == resolved_path + ) for pending_item in self.pending_files ) if not already_pending: diff --git a/tritonparse/reproducer/tests/artifacts/triton_fused_kernel.py b/tritonparse/reproducer/tests/artifacts/triton_fused_kernel.py index 121a8f0..baf2f4b 100644 --- a/tritonparse/reproducer/tests/artifacts/triton_fused_kernel.py +++ b/tritonparse/reproducer/tests/artifacts/triton_fused_kernel.py @@ -14,7 +14,6 @@ import torch import triton import triton.language as tl - from tritonparse.reproducer.tests.artifacts.triton_preprocess import scale_kernel diff --git a/tritonparse/reproducer/tests/artifacts/triton_preprocess.py b/tritonparse/reproducer/tests/artifacts/triton_preprocess.py index 042ddeb..93d6565 100644 --- a/tritonparse/reproducer/tests/artifacts/triton_preprocess.py +++ b/tritonparse/reproducer/tests/artifacts/triton_preprocess.py @@ -4,7 +4,6 @@ import triton import triton.language as tl - from tritonparse.reproducer.tests.artifacts.triton_utils import add_values