From b57ab9a3f2306ecc47caad42060b355438527273 Mon Sep 17 00:00:00 2001 From: "Wu, Zhenyu" Date: Fri, 10 Oct 2025 16:05:07 +0000 Subject: [PATCH] Fix #165125: Type "str" is not assignable to return type "None" (#165128) Fixes #165125 Pull Request resolved: https://github.com/pytorch/pytorch/pull/165128 Approved by: https://github.com/malfet --- torch/utils/cpp_extension.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torch/utils/cpp_extension.py b/torch/utils/cpp_extension.py index 764ce87d7692..b0b4399f0a52 100644 --- a/torch/utils/cpp_extension.py +++ b/torch/utils/cpp_extension.py @@ -11,6 +11,7 @@ import setuptools import subprocess import sys import sysconfig +import types import collections from pathlib import Path import errno @@ -2095,7 +2096,7 @@ def _jit_compile(name, with_sycl: Optional[bool], is_python_module, is_standalone, - keep_intermediates=True) -> None: + keep_intermediates=True) -> Union[types.ModuleType, str]: if is_python_module and is_standalone: raise ValueError("`is_python_module` and `is_standalone` are mutually exclusive.")