Fix import error

Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Nov 15, 2024 · 102c51c · 102c51c
1 parent 04be192
commit 102c51c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/project/algorithms/llm_finetuning_test.py b/project/algorithms/llm_finetuning_test.py
@@ -13,7 +13,7 @@
 from tensor_regression.to_array import to_ndarray
 from torch.utils.data import DataLoader
 
-from project.algorithms.llm_finetuning_example import (
+from project.algorithms.llm_finetuning import (
     DatasetConfig,
     LLMFinetuningExample,
     TokenizerConfig,