Use local dict instead of modifying locals()

What's Changed

• Use local dict instead of modifying locals() by @martinhoyer in #248

New Contributors

• @martinhoyer made their first contribution in #248