Merge pull request #2158 from ubergonmx/patch-1

Correct variable used in pickle dump in `mind_utils.ipynb`
recommenders-team · Aug 28, 2024 · d61fcd5 · d61fcd5
2 parents d0a4af3 + 587dc01
commit d61fcd5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/AUTHORS.md b/AUTHORS.md
@@ -52,6 +52,8 @@ To contributors: please add your name to the list when you submit a patch to the
 * **[Aaron He](https://github.com/AaronHeee)**
    * Reco utils of NCF
    * Deep dive notebook demonstrating the use of NCF
+* **[Aaron Palpallatoc](https://github.com/ubergonmx)**
+   * Corrected variable in pickle dump in `mind_utils.ipynb` notebook
 * **[Abir Chakraborty](https://github.com/aeroabir)**
    * Self-Attentive Sequential Recommendation (SASRec)
    * Sequential Recommendation Via Personalized Transformer (SSEPT)

diff --git a/examples/01_prepare_data/mind_utils.ipynb b/examples/01_prepare_data/mind_utils.ipynb
@@ -306,7 +306,7 @@
                 "    pickle.dump(word_dict, f)\n",
                 "    \n",
                 "with open(os.path.join(output_path, 'word_dict_all.pkl'), 'wb') as f:\n",
-                "    pickle.dump(word_dict, f)"
+                "    pickle.dump(word_dict_all, f)"
             ]
         },
         {