you probably want to get rid of the test argument, and then just pass `subsampled` in once on line 111 and remove the saving of `n_test_observations` in the json?