Allow sample_predicate as key as well