Importing ExitStack on pytorch

1

Pytorch is installed on python2 but got the error when importing ExitStack .

from contextlib import ExitStack

I am getting the following error:

File “tools/train.py”, line 14, in
import adda
File “/media/user/DATA/New_CODE/Working/adda/adda/init.py”, line 1, in
from adda import adversary
File “/media/user/DATA/New_CODE/Working/adda/adda/adversary.py”, line 2, in
from contextlib import ExitStack
ImportError: cannot import name ExitStack

2

Hardly a pytorch question. Google has a good answer.