Skip to content

Commit 894c6b5

Browse files
committed
Standardize on os.environ.get under cuda_pathfinder/
1 parent cc7d84d commit 894c6b5

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

cuda_pathfinder/cuda/pathfinder/_dynamic_libs/find_nvidia_dynamic_lib.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ def _find_lib_dir_using_cuda_home(libname: str) -> Optional[str]:
113113

114114

115115
def _find_lib_dir_using_conda_prefix(libname: str) -> Optional[str]:
116-
conda_prefix = os.getenv("CONDA_PREFIX")
116+
conda_prefix = os.environ.get("CONDA_PREFIX")
117117
if not conda_prefix:
118118
return None
119119
return _find_lib_dir_using_anchor_point(

cuda_pathfinder/cuda/pathfinder/_headers/find_nvidia_headers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def _find_ctk_header_directory(libname: str) -> Optional[str]:
9090
if _joined_isfile(hdr_dir, h_basename):
9191
return hdr_dir
9292

93-
conda_prefix = os.getenv("CONDA_PREFIX")
93+
conda_prefix = os.environ.get("CONDA_PREFIX")
9494
if conda_prefix: # noqa: SIM102
9595
if result := _find_based_on_conda_layout(libname, h_basename, conda_prefix):
9696
return result

cuda_pathfinder/tests/test_find_nvidia_headers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def test_find_libname_nvshmem(info_summary_append):
5959
if have_nvidia_nvshmem_package():
6060
hdr_dir_parts = hdr_dir.split(os.path.sep)
6161
assert "site-packages" in hdr_dir_parts
62-
elif conda_prefix := os.getenv("CONDA_PREFIX"):
62+
elif conda_prefix := os.environ.get("CONDA_PREFIX"):
6363
assert hdr_dir.startswith(conda_prefix)
6464
else:
6565
assert hdr_dir.startswith("/usr/include/nvshmem_")

0 commit comments

Comments
 (0)