linting

Author: hinriksnaer

helion/autotuner/base_search.py

@@ -337,16 +337,17 @@ def create_precompile_future(
             A ``PrecompileFuture`` that resolves to True on success or False on
             failure/timeout when called.
         """
-        ctx = self.benchmark_provider._precompile_context()
+        # TODO(hinriksnaer): migrate create_precompile_future to BenchmarkProvider
+        bp = self.benchmark_provider
+        assert isinstance(bp, LocalBenchmarkProvider)
+        ctx = bp._precompile_context()
         if not self.settings.autotune_precompile:
             return PrecompileFuture.skip(ctx, config, True)
         mode = self.settings.autotune_precompile
         if mode not in {"fork", "spawn"}:
             raise exc.InvalidAPIUsage("autotune_precompile must be 'fork' or 'spawn'")
-        if len(self.benchmark_provider.mutated_arg_indices) > 0:
-            args = _clone_args(
-                self.args, idx_to_clone=self.benchmark_provider.mutated_arg_indices
-            )
+        if len(bp.mutated_arg_indices) > 0:
+            args = _clone_args(self.args, idx_to_clone=bp.mutated_arg_indices)
         else:
             args = self.args
 
@@ -355,8 +356,8 @@ def create_precompile_future(
             config=config,
             fn=fn,
             args=args,
-            result_path=self.benchmark_provider._next_precompile_result_path(),
-            args_path=self.benchmark_provider._precompile_args_path,
+            result_path=bp._next_precompile_result_path(),
+            args_path=bp._precompile_args_path,
         )
 
     # TODO(hinriksnaer): migrate _benchmark to BenchmarkProvider

helion/autotuner/external.py

@@ -297,5 +297,5 @@ def autotune(
             **settings_kw,
         ),
         args,
-        **search_kw,
+        **search_kw,  # pyrefly: ignore[bad-argument-type]
     ).autotune()