Fixed README imports for LitQA

Future-House · Feb 24, 2025 · 15e36d3 · 15e36d3
1 parent 925b47b
commit 15e36d3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/litqa/README.md b/packages/litqa/README.md
@@ -38,12 +38,12 @@ Here is an example of how to use them:
 ```python
 import os
 
-from aviary.env import TaskDataset
 from ldp.agent import SimpleAgent
 from ldp.alg import Evaluator, EvaluatorConfig, MeanMetricsCallback
-
 from paperqa import Settings
-from litqa import TASK_DATASET_NAME
+
+from aviary.env import TaskDataset
+from aviary.envs.litqa.task import TASK_DATASET_NAME
 
 
 async def evaluate(folder_of_litqa_v2_papers: str | os.PathLike) -> None: