Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

Marginally faster display data #3544

Merged
merged 3 commits into from
Mar 31, 2021
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions parlai/scripts/display_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
"""

from parlai.core.params import ParlaiParser
from parlai.agents.repeat_label.repeat_label import RepeatLabelAgent
from parlai.agents.fixed_response.fixed_response import FixedResponseAgent
from parlai.core.worlds import create_task
from parlai.utils.strings import colorize
from parlai.core.script import ParlaiScript, register_script
Expand Down Expand Up @@ -66,7 +66,8 @@ def display_data(opt):

# create repeat label agent and assign it to the specified task
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit; update comment

opt.log()
agent = RepeatLabelAgent(opt)
opt['fixed_response'] = None
agent = FixedResponseAgent(opt)
world = create_task(opt, agent)

# Show some example dialogs.
Expand Down