[BugFix] Fix sampling in NonTensorSpec (#2172) #1830

Job	Run time
unittests-meltingpot / linux-job	9m 37s
unittests-pettingzoo / linux-job	9m 5s
unittests-atari-dqn (3.9, 12.1) / linux-job	12m 20s
unittests-brax (3.11, 12.1) / linux-job	8m 33s
unittests-d4rl (3.9, 12.1) / linux-job	8m 34s
unittests-envpool (3.9, 12.1) / linux-job	8m 39s
unittests-gendgrl (3.9, 12.1) / linux-job	7m 31s
unittests-gym (3.9, 12.1) / linux-job	16m 36s
unittests-jumanji (3.9, 12.1) / linux-job	34m 48s
unittests-minari (3.9, 12.1) / linux-job	5m 31s
unittests-openx (3.9, 12.1) / linux-job	9m 30s
unittests-robohive (3.9, 12.1) / linux-job	47m 31s
unittests-sklearn (3.9, 12.1) / linux-job	5m 59s
unittests-roboset (3.9, 12.1) / linux-job	10m 5s
unittests-smacv2 (3.9, 12.1) / linux-job	19m 51s
unittests-vd4rl (3.9, 12.1) / linux-job	9m 30s
unittests-vmas (3.9, 12.1) / linux-job	11m 47s
	3h 55m 27s

Provide feedback