Skip to content

Commit 77cc0fc

Browse files
committed
chore: fix optionals
1 parent 041d824 commit 77cc0fc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

cuda_pathfinder/cuda/pathfinder/_headers/find_nvidia_headers.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def _joined_isfile(dirpath: str, basename: str) -> bool:
2121
return os.path.isfile(os.path.join(dirpath, basename))
2222

2323

24-
def _find_under_site_packages(sub_dir: str, h_basename: str) -> Optional[str]:
24+
def _find_under_site_packages(sub_dir: str, h_basename: str) -> str | None:
2525
# Installed from a wheel
2626
hdr_dir: str # help mypy
2727
for hdr_dir in find_sub_dirs_all_sitepackages(tuple(sub_dir.split("/"))):
@@ -52,7 +52,7 @@ def _find_based_on_ctk_layout(libname: str, h_basename: str, anchor_point: str)
5252
return None
5353

5454

55-
def _find_based_on_conda_layout(libname: str, h_basename: str, ctk_layout: bool) -> Optional[str]:
55+
def _find_based_on_conda_layout(libname: str, h_basename: str, ctk_layout: bool) -> str | None:
5656
conda_prefix = os.environ.get("CONDA_PREFIX")
5757
if not conda_prefix:
5858
return None
@@ -134,7 +134,7 @@ def find_nvidia_header_directory(libname: str) -> str | None:
134134
raise RuntimeError(f"UNKNOWN {libname=}")
135135

136136
candidate_dirs = supported_nvidia_headers.SUPPORTED_SITE_PACKAGE_HEADER_DIRS_NON_CTK.get(libname, [])
137-
hdr_dir: Optional[str] # help mypy
137+
hdr_dir: str | None # help mypy
138138
for cdir in candidate_dirs:
139139
if hdr_dir := _find_under_site_packages(cdir, h_basename):
140140
return _abs_norm(hdr_dir)

0 commit comments

Comments
 (0)