mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-19 21:42:02 +00:00
934b2acbb7
* Warn if seed doesn't return a list * Add preliminary BenchmarkRun support * Add experimental benchmark registration * Flesh out interface * Add preliminary BenchmarkRun support * Warn if seed doesn't return a list * Add experimental benchmark registration * Flesh out interface * Make benchmarkrun upload recursive * Add evaluation episodes * Add benchmark scoring * Tweak reward locations * Tweak scoring * Clear default metadata in Wrapper * Improve scoring * Expose registry; fix test * Add initial_reset_timestamp * Add back algorithm; fix tests
47 lines
1.6 KiB
Python
Executable File
47 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
# This script assumes you have set an OPENAI_GYM_API_KEY environment
|
|
# variable. You can find your API key in the web interface:
|
|
# https://gym.openai.com/settings/profile.
|
|
import argparse
|
|
import logging
|
|
import os
|
|
import sys
|
|
|
|
import gym
|
|
|
|
# In modules, use `logger = logging.getLogger(__name__)`
|
|
logger = logging.getLogger()
|
|
|
|
class Uploader(object):
|
|
def __init__(self, training_dir, algorithm_id, benchmark_run_id, writeup):
|
|
self.training_dir = training_dir
|
|
self.algorithm_id = algorithm_id
|
|
self.benchmark_run_id = benchmark_run_id
|
|
self.writeup = writeup
|
|
|
|
def run(self):
|
|
gym.upload(self.training_dir, algorithm_id=self.algorithm_id, benchmark_run_id=self.benchmark_run_id, writeup=self.writeup)
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description=None)
|
|
parser.add_argument('-t', '--training-dir', required=True, help='What directory to upload.')
|
|
parser.add_argument('-a', '--algorithm_id', help='Set the algorithm id.')
|
|
parser.add_argument('-b', '--benchmark-run-id', help='Set the algorithm id.')
|
|
parser.add_argument('-w', '--writeup', help='Writeup to attach.')
|
|
parser.add_argument('-v', '--verbose', action='count', dest='verbosity', default=0, help='Set verbosity.')
|
|
args = parser.parse_args()
|
|
|
|
if args.verbosity == 0:
|
|
logger.setLevel(logging.INFO)
|
|
elif args.verbosity >= 1:
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
runner = Uploader(training_dir=args.training_dir, algorithm_id=args.algorithm_id, benchmark_run_id=args.benchmark_run_id, writeup=args.writeup)
|
|
runner.run()
|
|
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|