Implement keymap equivalently to the reference

Implement keymap equivalently to the reference: a task in repo2rlenv-equivalence-tests (Harbor dataset). Implement the function keymap in /workspace/task module.py so that it is behaviorally equivalent to reference keymap (which is already provided in task module.py ).