[Feature] DT compatibility with compile #3666

Job	Run time
unittests-meltingpot / linux-job	14m 6s
unittests-pettingzoo / linux-job	9m 19s
unittests-atari-dqn (3.9, 12.1) / linux-job	12m 34s
unittests-brax (3.11, 12.1) / linux-job	10m 53s
unittests-d4rl (3.9, 12.1) / linux-job	8m 54s
unittests-envpool (3.9, 12.1) / linux-job	9m 46s
unittests-gendgrl (3.9, 12.1) / linux-job	8m 0s
unittests-gym (3.9, 12.1) / linux-job	10m 51s
unittests-jumanji (3.9, 12.1) / linux-job	44m 32s
unittests-minari (3.9, 12.1) / linux-job	6m 35s
unittests-open_spiel (3.9, 12.1) / linux-job	8m 1s
unittests-openx (3.9, 12.1) / linux-job	24m 40s
unittests-robohive (3.9, 12.1) / linux-job	49m 18s
unittests-roboset (3.9, 12.1) / linux-job	11m 51s
unittests-sklearn (3.9, 12.1) / linux-job	6m 46s
unittests-smacv2 (3.9, 12.1) / linux-job	12m 10s
unittests-unity_mlagents (3.10.12, 12.1) / linux-job	8m 30s
unittests-vd4rl (3.9, 12.1) / linux-job	6m 33s
unittests-vmas (3.9, 12.1) / linux-job	12m 51s
	4h 36m 10s

Provide feedback