Commit c29b78b8 authored by Benjamin Beyret's avatar Benjamin Beyret
Browse files

add inference to Gym env

parent 818f29a9
......@@ -154,6 +154,9 @@ v0.6.1)
## Version History
- v1.0.3
- Adds inference mode to Gym environment
- v1.0.2
- Adds custom resolution for docker training as well
- fix version checker
......
......@@ -33,6 +33,7 @@ class AnimalAIEnv(gym.Env):
arenas_configurations=None,
greyscale=False,
retro=True,
inference=False,
resolution=None):
"""
Environment initialization
......@@ -51,9 +52,11 @@ class AnimalAIEnv(gym.Env):
docker_training=docker_training,
n_arenas=n_arenas,
arenas_configurations=arenas_configurations,
inference=inference,
resolution=resolution)
# self.name = self._env.academy_name
self.vector_obs = None
self.inference = inference
self.resolution = resolution
self._current_state = None
self._n_agents = None
......
......@@ -2,7 +2,7 @@ from setuptools import setup
setup(
name='animalai',
version='1.0.2',
version='1.0.3',
description='Animal AI competition interface',
url='https://github.com/beyretb/AnimalAI-Olympics',
author='Benjamin Beyret',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment