• Tal Sh's avatar
    Fix RepeatedData + PrefetchDataZMQ (#747) · 7388f508
    Tal Sh authored
    Doing:
    
    ```
    ds = RepeatedData(ds, -1)
    ds = PrefetchDataZMQ(ds, num_workers)
    ds.reset_state()
    ```
    
    Would blow up, since `RepeatedData` raises a ValueError, and `PrefetchDataZMQ` only checks for `NotImplementedError`.
    
    Not the cleanest fix but 🤷‍♂️
    7388f508
common.py 25.7 KB