Skip to content

Commit

Permalink
Clean out Pickle5 compat usage
Browse files Browse the repository at this point in the history
  • Loading branch information
mdekstrand committed Nov 10, 2023
1 parent cb607b2 commit f3eec5d
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 14 deletions.
10 changes: 1 addition & 9 deletions tests/test_als_explicit.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,8 @@
import logging
import pickle

from numpy.lib.npyio import _save_dispatcher

# get a usable pickle disassembler
if pickle.HIGHEST_PROTOCOL >= 5:
from pickletools import dis as pickle_dis
else:
try:
from pickle5.pickletools import dis as pickle_dis
except ImportError:
pass
from pickletools import dis as pickle_dis

from lenskit.algorithms import als
from lenskit import util
Expand Down
5 changes: 0 additions & 5 deletions tests/test_batch_recommend.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,6 @@
from lenskit import batch, topn
import lenskit.crossfold as xf

try:
import pickle5
except ImportError:
pickle5 = None

MLB = namedtuple('MLB', ['ratings', 'algo'])
_log = logging.getLogger(__name__)

Expand Down

0 comments on commit f3eec5d

Please sign in to comment.