[BEAM-13133] Loosen partitioning requirement for sample (#15818)
[BEAM-13133] Loosen partitioning requirement for sample (#15818)

* Loosen partitioning requirement for sample

* Add back index requirement for weights case
lukecwik/incubator-beam,chamikaramj/beam,chamikaramj/beam,apache/beam,lukecwik/incubator-beam,robertwb/incubator-beam,apache/beam,robertwb/incubator-beam,robertwb/incubator-beam,lukecwik/incubator-beam,lukecwik/incubator-beam,robertwb/incubator-beam,robertwb/incubator-beam,robertwb/incubator-beam,chamikaramj/beam,apache/beam,chamikaramj/beam,apache/beam,apache/beam,apache/beam,chamikaramj/beam,apache/beam,chamikaramj/beam,apache/beam,chamikaramj/beam,robertwb/incubator-beam,lukecwik/incubator-beam,lukecwik/incubator-beam,apache/beam,lukecwik/incubator-beam,robertwb/incubator-beam,robertwb/incubator-beam,chamikaramj/beam,apache/beam,lukecwik/incubator-beam,chamikaramj/beam,lukecwik/incubator-beam,lukecwik/incubator-beam,robertwb/incubator-beam,chamikaramj/beam,apache/beam