jannerm / trajectory-transformer

Code for the paper "Offline Reinforcement Learning as One Big Sequence Modeling Problem"
https://trajectory-transformer.github.io
MIT License
455 stars 63 forks source link

Can't run train script #1

Closed pyrito closed 2 years ago

pyrito commented 2 years ago

Hi I keep getting an error from this line: from tap import Tap

What is this module? I can't seem to find it properly and it's not working well.

pyrito commented 2 years ago

I think I figured it out. Sorry!

jannerm commented 2 years ago

Glad you figured it out! In case anyone else runs into this issue: tap is the import name for typed-argument-parser. It's included in the environment file here.