From bc8adad3b6ca52156a9bac859ccb0008fdb6fd09 Mon Sep 17 00:00:00 2001
From: guarin <guarin@lightly.ai>
Date: Mon, 11 Dec 2023 18:10:11 +0000
Subject: [PATCH] Add backwards compatible import

---
 lightly/loss/memory_bank.py | 2 ++
 1 file changed, 2 insertions(+)
 create mode 100644 lightly/loss/memory_bank.py

diff --git a/lightly/loss/memory_bank.py b/lightly/loss/memory_bank.py
new file mode 100644
index 000000000..3c01a7082
--- /dev/null
+++ b/lightly/loss/memory_bank.py
@@ -0,0 +1,2 @@
+# For backwards compatibility as memory_bank module was previously in loss module.
+from lightly.models.modules.memory_bank import MemoryBankModule