fix imports

lenskit · Aug 5, 2024 · ec64658 · ec64658
1 parent 6abc677
commit ec64658
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/lenskit/lenskit/splitting/users.py b/lenskit/lenskit/splitting/users.py
@@ -14,8 +14,7 @@
 from seedbank import numpy_rng
 
 from lenskit.data.dataset import Dataset, MatrixDataset
-from lenskit.data.vocab import EntityId
-from lenskit.types import RandomSeed
+from lenskit.types import EntityId, RandomSeed
 
 from .holdout import HoldoutMethod
 from .split import TTSplit

diff --git a/lenskit/tests/test_knn_item_item.py b/lenskit/tests/test_knn_item_item.py
@@ -23,8 +23,7 @@
 from lenskit.algorithms.basic import Fallback
 from lenskit.algorithms.bias import Bias
 from lenskit.algorithms.ranking import TopN
-from lenskit.data.dataset import from_interactions_df
-from lenskit.data.vocab import EntityId, Vocabulary
+from lenskit.data import Vocabulary, from_interactions_df
 from lenskit.diagnostics import ConfigWarning, DataWarning
 from lenskit.util import clone
 from lenskit.util.test import ml_ds, ml_ratings  # noqa: F401

diff --git a/lenskit/tests/test_pipeline.py b/lenskit/tests/test_pipeline.py
@@ -12,8 +12,7 @@
 
 from pytest import fail, raises
 
-from lenskit.data.dataset import Dataset
-from lenskit.data.vocab import EntityId, Vocabulary
+from lenskit.data import Dataset, Vocabulary
 from lenskit.pipeline import InputNode, Node, Pipeline
 from lenskit.pipeline.components import TrainableComponent