Hi Raymond,
Thanks for your message and the helpful direction. I went back and re-ran the notebook from the top, especially focusing on Cell 9 like you suggested. I double-checked the outputs of split_dataset() and confirmed that it’s returning non-empty lists — but the test still expects [1, 2] and [0, 3, 4], while my version gives [1, 2, 4] for that case.
I tried a few different ways to match what the grader seems to want, but no luck so far. I haven’t attached the notebook yet, but I can definitely share it if that helps. Let me know if you have any other ideas — I really appreciate your time.
— Mark