Merge pull request nltk#2583 from CubieDev/hotfix-2582

Fix for Issue nltk#2582, name `save_file` is not defined
chantera · Sep 4, 2020 · 873b359 · 873b359
2 parents b8a9647 + 019cc03
commit 873b359
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/nltk/sentiment/sentiment_analyzer.py b/nltk/sentiment/sentiment_analyzer.py
@@ -190,6 +190,7 @@ def save_file(self, content, filename):
         """
         print("Saving", filename, file=sys.stderr)
         with open(filename, 'wb') as storage_file:
+            import pickle
             # The protocol=2 parameter is for python2 compatibility
             pickle.dump(content, storage_file, protocol=2)
 

diff --git a/nltk/sentiment/util.py b/nltk/sentiment/util.py
@@ -685,7 +685,7 @@ def demo_subjectivity(trainer, save_analyzer=False, n_instances=None, output=Non
     results = sentim_analyzer.evaluate(test_set)
 
     if save_analyzer == True:
-        save_file(sentim_analyzer, "sa_subjectivity.pickle")
+        sentim_analyzer.save_file(sentim_analyzer, "sa_subjectivity.pickle")
 
     if output:
         extr = [f.__name__ for f in sentim_analyzer.feat_extractors]