Indicate that `env_keys` should be a dictionary:

```python
"""
Restore the environment by setting the keys in the env_keys dict again with their old value
"""
```

also, I see no reason to split the docstring across two lines, anything less than 120 chars is OK on 1 line according to our coding style