ts.torch_handler.unit_tests.test_utils 包

子模块

ts.torch_handler.unit_tests.test_utils.mock_context 模块

用于在不加载所有 Torchserve 的情况下添加模型上下文的模拟

class ts.torch_handler.unit_tests.test_utils.mock_context.MockContext(model_pt_file='model.pt', model_dir='ts/torch_handler/unit_tests/models/tmp', model_file='model.py', gpu_id='0', model_name='mnist', model_yaml_config_file=None)

基类：object

模拟类，用于复制传递到模型初始化中的上下文

get_request_header(idx, exp)

模块内容