Update rag_demo/rag/query_expansion.py

rag_demo/rag/query_expansion.py

@@ -3,9 +3,9 @@ from typing import Any
 
 from huggingface_hub import InferenceClient
 
-from rag_demo.rag.base.query import Query
-from rag_demo.rag.base.template_factory import RAGStep
-from rag_demo.rag.prompt_templates import QueryExpansionTemplate
+from .base.query import Query
+from .base.template_factory import RAGStep
+from .prompt_templates import QueryExpansionTemplate
 
 
 class QueryExpansion(RAGStep):