[BUG] StringIO module not found

[lokimonoton]

Traceback (most recent call last):
  File "main.py", line 4, in <module>
    from trainer import Trainer
  File "/home/ubuntu/workspace/BEGAN-tensorflow/trainer.py", line 4, in <module>
    import StringIO
ModuleNotFoundError: No module named 'StringIO'

resolved
open trainer.py and add from io before import StringIO like this :
from io import StringIO