add missing yield

Lightning-AI · Oct 5, 2022 · 6cc8e44 · 6cc8e44
1 parent 8441726
commit 6cc8e44
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/lightning_lite/accelerators/cuda.py b/src/lightning_lite/accelerators/cuda.py
@@ -93,6 +93,8 @@ def _patch_cuda_is_available() -> Generator:
             yield
         finally:
             torch.cuda.is_available = orig_check
+    else:
+        yield
 
 
 @lru_cache(1)