| 121 | |
| 122 | @pytest.fixture(autouse=True) |
| 123 | def apply_patches(): |
| 124 | patch("model.pb_utils.Tensor", new=MockTritonTensor).start() |
| 125 | patch("model.pb_utils.InferenceResponse", new=MockTritonResponse).start() |
| 126 | patch("model.pb_utils.TritonError", new=MockTritonError).start() |
| 127 | patch("model.pb_utils.InferenceRequest", new=MockTritonRequest).start() |
| 128 | patch("model.pb_utils.get_input_tensor_by_name", |
| 129 | new=mock_pb_utils_get_input_tensor_by_name_side_effect).start() |
| 130 | patch("model.pb_utils.TritonModelException", new=Exception).start() |
| 131 | |
| 132 | |
| 133 | @pytest.fixture |