From 9a10d4e4649b0cc92a63af0b7dc64e65db496a0d Mon Sep 17 00:00:00 2001 From: Michael Ekstrand Date: Mon, 29 May 2023 16:53:26 -0600 Subject: [PATCH] fix np.object --- lenskit/datasets.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/lenskit/datasets.py b/lenskit/datasets.py index babb96fa0..82d606fa0 100644 --- a/lenskit/datasets.py +++ b/lenskit/datasets.py @@ -99,8 +99,8 @@ def movies(self): fn = self.path / 'movies.csv' movies = pd.read_csv(fn, dtype={ 'movieId': np.int32, - 'title': np.object, - 'genres': np.object + 'title': object, + 'genres': object, }) movies.rename(columns={'movieId': 'item'}, inplace=True) movies.set_index('item', inplace=True) @@ -159,8 +159,8 @@ def tags(self): tags = pd.read_csv(fn, dtype={ 'movieId': np.int32, 'userId': np.int32, - 'tag': np.object, - 'timestamp': np.int32 + 'tag': object, + 'timestamp': np.int32, }) tags.rename(columns={'userId': 'user', 'movieId': 'item'}, inplace=True) _log.debug('loaded %s, takes %d bytes', fn, tags.memory_usage().sum()) @@ -192,7 +192,7 @@ def tag_genome(self): genome = pd.read_csv(fn, dtype={ 'movieId': np.int32, 'tagId': np.int32, - 'relevance': np.float64 + 'relevance': np.float64, }) genome.rename(columns={'userId': 'user', 'movieId': 'item'}, inplace=True) genome = genome.join(tags, on='tagId')