Sam

I made it work. Please see the #17

I took a hint from your issue and modified the code like this, ``` all_zero = (~padding_mask).all(dim=0) ind_first_done = padding_mask.long().argmax(0) + 1 ind_first_done = (~all_zero) * ind_first_done + all_zero *...

> > > I meet the same problem with the following command: > > > git clone https://github.com/rail-berkeley/d4rl.git > > > cd d4rl > > > pip install -e ....