added return statement

neulab · Mar 28, 2024 · 256e396 · 256e396
1 parent 60ab96f
commit 256e396
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/examples/rag/rag_example.ipynb b/examples/rag/rag_example.ipynb
@@ -87,7 +87,7 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "5it [00:00, 37786.52it/s]\n"
+            "5it [00:00, 39199.10it/s]\n"
           ]
         },
         {
@@ -101,7 +101,7 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "100%|██████████| 1/1 [00:01<00:00,  1.83s/it]"
+            "100%|██████████| 1/1 [00:01<00:00,  1.73s/it]"
           ]
         },
         {
@@ -152,12 +152,12 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 8,
+      "execution_count": 7,
       "id": "456ecd6c",
       "metadata": {},
       "outputs": [],
       "source": [
-        "rag_LM.generate(condition=None)"
+        "lm_response = rag_LM.generate(condition=None)"
       ]
     },
     {

diff --git a/llments/lm/rag.py b/llments/lm/rag.py
@@ -30,7 +30,7 @@ def generate(self, condition: str | None, do_sample: bool = False, max_length: i
         Returns:
             str: Sampled output sequences from the language model.
         """
-        pass
+        return super().generate(condition, do_sample, max_length, temperature, num_return_sequences)