* [ML] Adding ability to override number of sample lines in file data viz * tiny tweak * updating tests