Farama-Foundation/Gymnasium

[Proposal] Allow to specify dtype for Discrete

已關閉

#1,118 建立於 2024年7月13日

在 GitHub 查看貢獻者指南
 (4 則留言) (0 個反應) (0 位負責人)Python (567 個分叉)batch import
enhancementgood first issue

倉庫指標

星標
 (4,712 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

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

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

貢獻者指南

研究方向
在gymnasium/spaces/discrete.py中找到Discrete類的定義。添加一個預設為numpy.int64的dtype參數。更新 init 以儲存dtype，並修改sample方法以使用該dtype。通過保持預設行為確保向後相容。
技術棧
python
領域
apimachine learning
議題類型
功能
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
半天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
Pythonnumpy
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
70