Check the controllble dialogue data loads.
train_output, valid_output, test_output = testing_utils.display_data(
{"task": "self_feeding:all"}
)
// check valid data
self.assertIn("i am spending time with my 4 sisters", train_output)
self.assertIn("193777 episodes with a total of 193777 examples", train_output)
After Change
Check the controllable dialogue data loads.
pp = ParlaiParser(True, False)
opt = pp.parse_kwargs(task="self_feeding:all", datatype="train:ordered")
teacher = AllTeacher(opt)
assert teacher.num_examples() == 193777
assert teacher.num_episodes() == 193777
assert "some cheetah chasing to stay in shape" in teacher.act()["text"]
opt["datatype"] = "valid"
teacher = AllTeacher(opt)
assert teacher.num_examples() == 3500
opt["datatype"] = "test"
