Python load Examples

Example #1

File: suite_bsuite_test.py Project: Nitty12/MADRL

 def testBsuiteEnvRegistered(self):
     env = suite_bsuite.load('deep_sea/0', record=False)
     self.assertIsInstance(env, py_environment.PyEnvironment)

Example #2

File: suite_bsuite_test.py Project: Nitty12/MADRL

 def testGinConfig(self):
     gin.parse_config_file(
         test_utils.test_src_dir_path(
             'environments/configs/suite_bsuite.gin'))
     env = suite_bsuite.load()
     self.assertIsInstance(env, py_environment.PyEnvironment)

Example #3

File: suite_bsuite_test.py Project: Nitty12/MADRL

 def testBsuiteEnvRegisteredWithRecord(self):
     env = suite_bsuite.load('deep_sea/0',
                             record=True,
                             save_path=None,
                             logging_mode='terminal')
     self.assertIsInstance(env, py_environment.PyEnvironment)