Gymnasium icon indicating copy to clipboard operation
Gymnasium copied to clipboard

Published 20 hours ago • verified publisher icon Farama-Foundation

[Proposal] Allow to specify dtype for Discrete

Open younik opened this issue 7 months ago • 1 comments

Proposal

Add dtype argument to spaces.Discrete (similar to MultiDiscrete and Box).

Motivation

Currently the dtype is fixed to numpy.int64. However, often Discrete spaces are much smaller, resulting in a waste of memory.

Pitch

No response

Alternatives

No response

Additional context

No response

Checklist

  • [X] I have checked that there is no similar issue in the repo

younik avatar Jul 13 '24 17:07 younik