Iterator error after pip install

Hello,

Thanks so much for an awesome package!😃

I recently re-pip installed plbolts on a google colab notebook. My code worked before. I’m trying to train a DQN.

On importing the DQN (from pl_bolts.models.rl import DQN), I get the following error:

/usr/local/lib/python3.7/dist-packages/torch/utils/data/_typing.py in _dp_init_subclass(sub_cls, *args, **kwargs)
    369                      return_hint.__origin__ == collections.abc.Iterator)):
    370                 raise TypeError("Expected 'Iterator' as the return annotation for `__iter__` of {}"
--> 371                                 ", but found {}".format(sub_cls.__name__, _type_repr(hints['return'])))
    372             data_type = return_hint.__args__[0]
    373             if not issubtype(data_type, sub_cls.type.param):

TypeError: Expected 'Iterator' as the return annotation for `__iter__` of ExperienceSourceDataset, but found typing.Iterable

Help would be appreciated!

Thank you:-)

[gurlina]

I found that installing the latest version from the Master branch fixes the import problem (no guarantees, though, that it wouldn’t break something else…):

pip install git+https://github.com/PytorchLightning/lightning-bolts.git@master --upgrade

[mariakesa]

Yep, I was also working in Colab. The funny thing is that some time before the code worked.