From b5ffdcf7967e8a25c4a8900c49701165a91e9426 Mon Sep 17 00:00:00 2001
From: Konstantin Dobler <konstantin.j.dobler@gmail.com>
Date: Tue, 7 Nov 2023 00:21:03 +0100
Subject: [PATCH] Fix import of FOCUS

---
 focus_example.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/focus_example.py b/focus_example.py
index e9a87da..55a8490 100644
--- a/focus_example.py
+++ b/focus_example.py
@@ -5,7 +5,7 @@
     PreTrainedTokenizer,
 )
 
-from focus import FOCUS
+from deepfocus import FOCUS
 
 source_tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
 source_model: PreTrainedModel = AutoModelForMaskedLM.from_pretrained("xlm-roberta-base")