refactor missed imports

Lightning-Universe · May 5, 2024 · 9820f58 · 9820f58
1 parent fbc8f2f
commit 9820f58
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/src/pl_bolts/callbacks/knn_online.py b/src/pl_bolts/callbacks/knn_online.py
@@ -2,7 +2,7 @@
 
 import torch
 from lightning import Callback, LightningModule, Trainer
-from lightning.accelerators import Accelerator
+from lightning.pytorch.accelerators import Accelerator
 from pl_bolts.utils.stability import under_review
 from torch import Tensor
 from torch.nn import functional as F  # noqa: N812

diff --git a/src/pl_bolts/utils/_dependency.py b/src/pl_bolts/utils/_dependency.py
@@ -2,10 +2,7 @@
 import os
 from typing import Any, Callable
 
-from lightning.fabric.utilities.core.imports import (
-    ModuleAvailableCache,
-    RequirementCache,
-)
+from lightning_utilities.core.imports import ModuleAvailableCache, RequirementCache
 
 
 # ToDo: replace with utils wrapper after 0.10 is released