ValueError: Missing type hint for return value in function train_test_split

[gilad-rubin]

For an unknown reason - I'm getting this error even though I've defined a return values in the file
https://github.com/DAGWorks-Inc/tuning-playground/blob/main/ml_project/phase4/split.py

[skrawcz]

That's odd -- otherwise you don't need the typing import. You can just do dict as the type annotation.

[skrawcz]

Also to rule out the sklearn function somehow interfering:

from sklearn import model_selection

then in your function do

model_selection.train_test_split(...)