From 146c6ca0dc0192c45cbc7c9e6a0d0458868890be Mon Sep 17 00:00:00 2001 From: Yodogawa Mikio <53687275+software-samurai@users.noreply.github.com> Date: Tue, 14 Mar 2023 16:31:13 +0100 Subject: [PATCH] Use SupportsFloat in step_api_compatibility (#368) Co-authored-by: Mark Towers --- gymnasium/utils/step_api_compatibility.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gymnasium/utils/step_api_compatibility.py b/gymnasium/utils/step_api_compatibility.py index 7131637d4..d0b0d71c2 100644 --- a/gymnasium/utils/step_api_compatibility.py +++ b/gymnasium/utils/step_api_compatibility.py @@ -1,5 +1,5 @@ """Contains methods for step compatibility, from old-to-new and new-to-old API.""" -from typing import Tuple, Union +from typing import SupportsFloat, Tuple, Union import numpy as np @@ -8,14 +8,14 @@ from gymnasium.core import ObsType DoneStepType = Tuple[ Union[ObsType, np.ndarray], - Union[float, np.ndarray], + Union[SupportsFloat, np.ndarray], Union[bool, np.ndarray], Union[dict, list], ] TerminatedTruncatedStepType = Tuple[ Union[ObsType, np.ndarray], - Union[float, np.ndarray], + Union[SupportsFloat, np.ndarray], Union[bool, np.ndarray], Union[bool, np.ndarray], Union[dict, list],