pin pickle protocol to 4 (Python 3.4+) and rebuild cache

biocommons · Jun 23, 2020 · c8a2a54 · c8a2a54
1 parent 1469be3
commit c8a2a54
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/hgvs/utils/PersistentDict.py b/hgvs/utils/PersistentDict.py
@@ -1,8 +1,8 @@
 # -*- coding: utf-8 -*-
-from __future__ import absolute_import, division, print_function, unicode_literals
 
 import pickle
 
+protocol = 4
 
 class PersistentDict(dict):
     ''' Persistent dictionary 
@@ -23,7 +23,7 @@ def sync(self):
         if self.flag == 'r':
             return
         with open(self.filename, 'wb') as f:
-            pickle.dump(dict(self), f, -1)
+            pickle.dump(dict(self), f, protocol)
 
     def close(self):
         self.sync()

diff --git a/tests/data/cache-py3.hdp b/tests/data/cache-py3.hdp
diff --git a/tests/test_hgvs_dataproviders_uta.py b/tests/test_hgvs_dataproviders_uta.py
@@ -64,7 +64,7 @@ def test_get_tx_exons_invalid_alt_aln_method(self):
 
     def test_get_tx_for_gene(self):
         tig = self.hdp.get_tx_for_gene("VHL")
-        self.assertEqual(14, len(tig))
+        self.assertEqual(16, len(tig))
 
     def test_get_tx_for_gene_invalid_gene(self):
         tig = self.hdp.get_tx_for_gene("GENE")