tf_agents.specs.array_spec.is_discrete

View source on GitHub