Issues with data_generator.py

[ioannagiorgi]

I keep getting KeyError: 0
self.state_idx = range(demos[0]['demoX'].shape[-1])
self._dU = demos[0]['demoU'].shape[-1]

and
states = np.vstack(demos[i]['demoX'] for i in self.train_idx) # hardcoded here to solve the memory issue
ValueError: need at least one array to concatenate

Thanks

[FRANKJIANG20]

I have the same problem with you.Maybe it has something to do with the python3

[ZlSong417]

@ioannagiorgi @FRANKJIANG20 hi，I have the same problem with you.Have you solved the problem now?