83a4e59ca670d4cc1ccb34e0b9f80ce5bf22dd49,tests/main_test.py,TestMain,test_docnn_with_torchscript_export,#TestMain#,87

Before Change

        )
        expected = ExportConfig:\n    accelerate: []\n    batch_padding_control: None\n    export_caffe2_path: /tmp/model.caffe2.predictor\n    export_lite_path: None\n    export_onnx_path: /tmp/model.onnx\n    export_torchscript_path: /tmp/new_docnn.pt1\n    inference_interface: None\n    seq_padding_control: None\n    torchscript_quantize: False
        assert not result.exception, result.exception
        assert expected in result.output

        // export the trained model with output path
        result = self.runner.invoke(
            main,

After Change

                os.path.join(self.config_base_path, "export_options.json"),
            ],
        )
        expected = [
            "accelerate: []",
            "batch_padding_control: None",
            "export_caffe2_path: /tmp/model.caffe2.predictor",
            "export_lite_path: None",
            "export_onnx_path: /tmp/model.onnx",
            "export_torchscript_path: /tmp/new_docnn.pt1",
            "seq_padding_control: None",
            "target: unknown",
            "torchscript_quantize: False",
        ]
        assert not result.exception, result.exception
        for item in expected:
            assert item in result.output

        // export the trained model with output path
        result = self.runner.invoke(
            main,
            args=[
                "--config-json",

Instances