From 9820f586ff7c282df12afab2fb9485a18625115c Mon Sep 17 00:00:00 2001
From: James Bishop <jbishop@turing.ac.uk>
Date: Sun, 5 May 2024 14:31:49 +0100
Subject: [PATCH] refactor missed imports

---
 src/pl_bolts/callbacks/knn_online.py | 2 +-
 src/pl_bolts/utils/_dependency.py    | 5 +----
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/src/pl_bolts/callbacks/knn_online.py b/src/pl_bolts/callbacks/knn_online.py
index 0d0cb8896..9c392a39e 100644
--- 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
index 32571c478..81ce2d912 100644
--- 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