Update base.py
Merge pull request #6 from DreamerAG/patch-1

base.py