Update retriever.py
Browse files- retriever.py +1 -1
retriever.py
CHANGED
@@ -30,7 +30,7 @@ class GuestInfoRetrieverTool(Tool):
|
|
30 |
|
31 |
def load_guest_dataset():
|
32 |
# Load the dataset
|
33 |
-
guest_dataset = datasets.load_dataset("
|
34 |
|
35 |
# Convert dataset entries into Document objects
|
36 |
docs = [
|
|
|
30 |
|
31 |
def load_guest_dataset():
|
32 |
# Load the dataset
|
33 |
+
guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
|
34 |
|
35 |
# Convert dataset entries into Document objects
|
36 |
docs = [
|