?

Author: asukaminato0721

pyrefly/lib/alt/function.rs

@@ -368,14 +368,16 @@ impl<'a, Ans: LookupAnswer> AnswersSolver<'a, Ans> {
                 } else {
                     let metadata = self
                         .merge_overload_metadata_with_implementation(&defs, def.metadata().clone());
-                    let sigs = self.normalize_async_generator_overloads(
-                        self.extract_signatures(
-                            metadata.kind.function_name().as_ref(),
-                            defs,
-                            errors,
-                        ),
-                        def.ty.callable_signatures()[0],
+                    let sigs = self.extract_signatures(
+                        metadata.kind.function_name().as_ref(),
+                        defs,
+                        errors,
                     );
+                    let sigs = if let [impl_sig] = def.ty.callable_signatures().as_slice() {
+                        self.normalize_async_generator_overloads(sigs, impl_sig)
+                    } else {
+                        sigs
+                    };
                     self.check_signature_consistency(&sigs, &def, errors);
                     Type::Overload(Overload {
                         signatures: sigs.mapped(|(_, sig)| sig),

pyrefly/lib/test/lsp/lsp_interaction/pytorch_benchmark.rs

@@ -48,7 +48,7 @@ fn test_pytorch_error_propagation_latency() {
     };
     // Use all available cores for realistic benchmarking
     let mut interaction =
-        LspInteraction::new_with_args(args, NoTelemetry, Some(ThreadCount::AllThreads));
+        LspInteraction::new_with_args(args, NoTelemetry, Some(ThreadCount::AllThreads), None);
     interaction.set_root(pytorch_root.clone());
 
     interaction