Temporarily remove feature_dim range, until feature_dim can be dynamically passed to benchmarker.
Temporarily remove feature_dim range, until feature_dim can be dynamically passed to benchmarker.

google-research/graphworld,google-research/graphworld