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

redo black

Browse Source
This commit is contained in:
Justin Terry
2021-07-29 12:42:48 -04:00
parent d5004b7ec1
commit e9d2c41f2b
109 changed files with 459 additions and 1363 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
scripts/generate_json.py
View File

@@ -15,9 +15,7 @@ steps = ROLLOUT_STEPS
ROLLOUT_FILE = os.path.join(DATA_DIR, "rollout.json")
if not os.path.isfile(ROLLOUT_FILE):
logger.info(
"No rollout file found. Writing empty json file to {}".format(ROLLOUT_FILE)
)
logger.info("No rollout file found. Writing empty json file to {}".format(ROLLOUT_FILE))
with open(ROLLOUT_FILE, "w") as outfile:
json.dump({}, outfile, indent=2)
@@ -50,9 +48,7 @@ def update_rollout_dict(spec, rollout_dict):
except:
# If running the env generates an exception, don't write to the rollout file
logger.warn(
"Exception {} thrown while generating rollout for {}. Rollout not added.".format(
sys.exc_info()[0], spec.id
)
"Exception {} thrown while generating rollout for {}. Rollout not added.".format(sys.exc_info()[0], spec.id)
)
return False
@@ -78,9 +74,7 @@ def update_rollout_dict(spec, rollout_dict):
def add_new_rollouts(spec_ids, overwrite):
environments = [
spec for spec in envs.registry.all() if spec.entry_point is not None
]
environments = [spec for spec in envs.registry.all() if spec.entry_point is not None]
if spec_ids:
environments = [spec for spec in environments if spec.id in spec_ids]
assert len(environments) == len(spec_ids), "Some specs not found"
@@ -110,9 +104,7 @@ if __name__ == "__main__":
help="Overwrite " + "existing rollouts if hashes differ.",
)
parser.add_argument("-v", "--verbose", action="store_true")
parser.add_argument(
"specs", nargs="*", help="ids of env specs to check (default: all)"
)
parser.add_argument("specs", nargs="*", help="ids of env specs to check (default: all)")
args = parser.parse_args()
if args.verbose:
logger.set_level(logger.INFO)