Missing sotopia environment in td3.py

likenneth / dialogue_action_token

Dialogue Action Tokens: Steering Language Models in Goal-Directed Dialogue with a Multi-Turn Planner

MIT License

17 stars 1 forks source link

Missing sotopia environment in td3.py #1

Closed Bohemianc closed 1 month ago

Bohemianc commented 4 months ago

Hello,

Nice work! The td3.py file seems to be missing the sotopia environment. Is there a need to modify the code, or was the file not included in the upload?

likenneth commented 4 months ago

Currently we haven't added the option to experiemnt on Sotopia in td3.py. Technically speaking, it should be easy as all we need to do is swap in the environment class defined in envs/sot_env.py.