From 73a4adb6d4161ea222ed1ba56413571892da73d6 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 9 Apr 2024 01:04:50 +0200 Subject: Fix imports --- rag/generator/prompt.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) (limited to 'rag/generator/prompt.py') diff --git a/rag/generator/prompt.py b/rag/generator/prompt.py index ed372c9..059592a 100644 --- a/rag/generator/prompt.py +++ b/rag/generator/prompt.py @@ -2,10 +2,7 @@ from dataclasses import dataclass from typing import List -try: - from rag.retriever.vector import Document -except ModuleNotFoundError: - from retriever.vector import Document +from rag.retriever.vector import Document @dataclass -- cgit v1.2.3-70-g09d2