diff --git a/examples/community_lm/community_lm_rag.ipynb b/examples/community_lm/community_lm_rag.ipynb index e342899..0b5cc4f 100644 --- a/examples/community_lm/community_lm_rag.ipynb +++ b/examples/community_lm/community_lm_rag.ipynb @@ -32,8 +32,8 @@ "from llments.datastore.pyserini_datastore import PyseriniDatastore\n", "import pandas as pd\n", "import numpy as np\n", - "from community_lm.community_lm_constants import politician_feelings, groups_feelings, anes_df\n", - "from examples.community_lm.community_lm_utils import generate_community_opinion, compute_group_stance, generate_community_opinion_rag\n", + "from community_lm_constants import politician_feelings, groups_feelings, anes_df\n", + "from community_lm_utils import generate_community_opinion, compute_group_stance, generate_community_opinion_rag\n", "\n", "device = 'cuda' # change to 'mps' if you have a mac, or 'cuda:0' if you have an NVIDIA GPU " ]