diff options
| author | Max Ryabinin <mryabinin0@gmail.com> | 2022-08-24 18:45:17 +0300 |
|---|---|---|
| committer | Max Ryabinin <mryabinin0@gmail.com> | 2022-08-24 18:45:17 +0300 |
| commit | 92a3363096e10ad6a5c4e944af898bd1186d806a (patch) | |
| tree | 4d21c6cb5cef45e992cf2cc36caf89af3c81632e /bitsandbytes/cuda_setup | |
| parent | 9fc0ab415c564d278b673c694de9b884ea1121d2 (diff) | |
Replace print_stderr with warnings.warn
Diffstat (limited to 'bitsandbytes/cuda_setup')
| -rw-r--r-- | bitsandbytes/cuda_setup/paths.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/bitsandbytes/cuda_setup/paths.py b/bitsandbytes/cuda_setup/paths.py index 610665f..6f6425f 100644 --- a/bitsandbytes/cuda_setup/paths.py +++ b/bitsandbytes/cuda_setup/paths.py @@ -2,7 +2,6 @@ from pathlib import Path from typing import Set, Union from warnings import warn -from ..utils import print_stderr from .env_vars import get_potentially_lib_path_containing_env_vars CUDA_RUNTIME_LIB: str = "libcudart.so" @@ -18,7 +17,7 @@ def remove_non_existent_dirs(candidate_paths: Set[Path]) -> Set[Path]: } if non_existent_directories: - print_stderr( + warn( "WARNING: The following directories listed in your path were found to " f"be non-existent: {non_existent_directories}" ) |