Skip to content

Commit 0842486

Browse files
committed
fix TT split type error
1 parent 81010c9 commit 0842486

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/lenskit/tuning/_search.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from ray.tune.search.optuna import OptunaSearch
2424

2525
from lenskit.config import TuneSettings, lenskit_config
26-
from lenskit.data import Dataset, GenericKey, ItemListCollection
26+
from lenskit.data import Dataset, ItemListCollection
2727
from lenskit.logging import get_logger
2828
from lenskit.parallel import get_parallel_config
2929
from lenskit.parallel.ray import ensure_cluster
@@ -58,7 +58,7 @@ class PipelineTuner:
5858
random_seed: np.random.SeedSequence
5959
iterative: bool
6060

61-
data: TTSplit[GenericKey]
61+
data: TTSplit
6262
harness: Any
6363
tuner: ray.tune.Tuner
6464
"""

0 commit comments

Comments
 (0)