gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-16 03:38:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

redo black (#2272)

Browse Source
This commit is contained in:
J K Terry
2021-07-29 15:39:42 -04:00
committed by GitHub
parent ff8c269abb
commit 78d2b512d8
104 changed files with 1350 additions and 418 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
gym/vector/utils/numpy_utils.py
View File

@@ -46,7 +46,10 @@ def concatenate(items, out, space):
elif isinstance(space, Space):
return concatenate_custom(items, out, space)
else:
raise ValueError("Space of type `{0}` is not a valid `gym.Space` " "instance.".format(type(space)))
raise ValueError(
"Space of type `{0}` is not a valid `gym.Space` "
"instance.".format(type(space))
)
def concatenate_base(items, out, space):
@@ -54,12 +57,18 @@ def concatenate_base(items, out, space):
def concatenate_tuple(items, out, space):
return tuple(concatenate([item[i] for item in items], out[i], subspace) for (i, subspace) in enumerate(space.spaces))
return tuple(
concatenate([item[i] for item in items], out[i], subspace)
for (i, subspace) in enumerate(space.spaces)
)
def concatenate_dict(items, out, space):
return OrderedDict(
[(key, concatenate([item[key] for item in items], out[key], subspace)) for (key, subspace) in space.spaces.items()]
[
(key, concatenate([item[key] for item in items], out[key], subspace))
for (key, subspace) in space.spaces.items()
]
)
@@ -109,7 +118,10 @@ def create_empty_array(space, n=1, fn=np.zeros):
elif isinstance(space, Space):
return create_empty_array_custom(space, n=n, fn=fn)
else:
raise ValueError("Space of type `{0}` is not a valid `gym.Space` " "instance.".format(type(space)))
raise ValueError(
"Space of type `{0}` is not a valid `gym.Space` "
"instance.".format(type(space))
)
def create_empty_array_base(space, n=1, fn=np.zeros):
@@ -122,7 +134,12 @@ def create_empty_array_tuple(space, n=1, fn=np.zeros):
def create_empty_array_dict(space, n=1, fn=np.zeros):
return OrderedDict([(key, create_empty_array(subspace, n=n, fn=fn)) for (key, subspace) in space.spaces.items()])
return OrderedDict(
[
(key, create_empty_array(subspace, n=n, fn=fn))
for (key, subspace) in space.spaces.items()
]
)
def create_empty_array_custom(space, n=1, fn=np.zeros):