Update collaboration.py

collaboration.py

@@ -1,17 +1,17 @@
 from datasets import Dataset
 from typing import Dict, List
 
-COLLAB_DATA: List[Dict] = []
+_COLLAB: List[Dict] = []
 
-def add_collaboration_note(doc_title: str, note: str) -> Dict:
+def add_collaboration_note(document: str, note: str) -> Dict:
     record = {
-        "document": doc_title,
+        "document": document,
         "note": note
     }
-    COLLAB_DATA.append(record)
+    _COLLAB.append(record)
     return record
 
-def get_collaboration_dataset():
-    if not COLLAB_DATA:
+def get_collaboration_dataset() -> Dataset:
+    if not _COLLAB:
         return Dataset.from_dict({"document": [], "note": []})
-    return Dataset.from_list(COLLAB_DATA)
+    return Dataset.from_list(_COLLAB)