Fix sample dataset indexing

[zzachw]

          We should probably add a check. But, it's mostly so we don't break compatibility with legacy code like split_by_visit() in https://github.com/sunlabuiuc/PyHealth/blob/master/pyhealth/datasets/splitter.py

Originally posted by @jhnwu3 in #335 (comment)