37f58cc580da8735ae5a259d4cb4ba0b8701e78e,tests/test_pytorch_data_teacher.py,TestPytorchDataTeacher,test_valid_pyt_batchsort,#TestPytorchDataTeacher#,144

Before Change

        defaults["datatype"] = "train:stream:ordered"
        defaults["pytorch_teacher_batch_sort"] = True

        f = io.StringIO()

        with redirect_stdout(f):
            // Get processed act from agent
            defaults["pytorch_teacher_task"] = "babi:task1k:1"

After Change

            defaults["pytorch_teacher_task"] = "babi:task1k:1"
            defaults["batch_sort_cache_type"] = "index"
            defaults["batchsize"] = 50
            defaults["model_file"] = os.path.join(tmpdir, "model")
            defaults["dict_file"] = os.path.join(tmpdir, "model.dict")
            bsrt_acts_ep1, bsrt_acts_ep2 = get_acts_epochs_1_and_2(defaults)

            defaults["pytorch_teacher_batch_sort"] = False
            defaults["model_file"] = os.path.join(tmpdir, "model")
            defaults["dict_file"] = os.path.join(tmpdir, "model.dict")
            no_bsrt_acts_ep1, no_bsrt_acts_ep2 = get_acts_epochs_1_and_2(defaults)

        check_equal_act_lists(bsrt_acts_ep1, no_bsrt_acts_ep1)
        check_equal_act_lists(bsrt_acts_ep2, no_bsrt_acts_ep2)

Instances