mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-01 06:07:08 +00:00
67212547ac
* remove six * remove __future__ imports * remove six from setup.py, python 2.7 from README.rst
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
from gym.envs.algorithmic import algorithmic_env
|
|
|
|
|
|
class ReversedAdditionEnv(algorithmic_env.GridAlgorithmicEnv):
|
|
def __init__(self, rows=2, base=3):
|
|
super(ReversedAdditionEnv, self).__init__(rows=rows, base=base, chars=False)
|
|
|
|
def target_from_input_data(self, input_strings):
|
|
curry = 0
|
|
target = []
|
|
for digits in input_strings:
|
|
total = sum(digits) + curry
|
|
target.append(total % self.base)
|
|
curry = total // self.base
|
|
|
|
if curry > 0:
|
|
target.append(curry)
|
|
return target
|
|
|
|
@property
|
|
def time_limit(self):
|
|
# Quirk preserved for the sake of consistency: add the length of the input
|
|
# rather than the length of the desired output (which may differ if there's
|
|
# an extra carried digit).
|
|
# TODO: It seems like this time limit is so strict as to make Addition3-v0
|
|
# unsolvable, since agents aren't even given enough time steps to look at
|
|
# all the digits. (The solutions on the scoreboard seem to only work by
|
|
# save-scumming.)
|
|
return self.input_width*2 + 4
|