[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
3c6cae3863
commit
9b8e49262c
1 changed files with 6 additions and 1 deletions
|
|
@ -321,7 +321,12 @@ def test_no_light_gpu_init_retry_on_an_accelerator_host(monkeypatch):
|
|||
attempts = []
|
||||
|
||||
class _Blocker:
|
||||
def find_spec(self, name, path = None, target = None):
|
||||
def find_spec(
|
||||
self,
|
||||
name,
|
||||
path = None,
|
||||
target = None,
|
||||
):
|
||||
if name == "unsloth_zoo":
|
||||
attempts.append(os.environ.get("UNSLOTH_ZOO_DISABLE_GPU_INIT"))
|
||||
raise RuntimeError("CUDA Setup failed despite GPU being available")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue