File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed
Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -113,7 +113,7 @@ def _find_lib_dir_using_cuda_home(libname: str) -> Optional[str]:
113113
114114
115115def _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 (
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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_" )
You can’t perform that action at this time.
0 commit comments