diff --git a/qa/L0_tf_parameters/tf_parameter_test.py b/qa/L0_tf_parameters/tf_parameter_test.py index 5f4794bdef..68d46861f6 100644 --- a/qa/L0_tf_parameters/tf_parameter_test.py +++ b/qa/L0_tf_parameters/tf_parameter_test.py @@ -35,33 +35,40 @@ class TFParameterTest(tu.TestResultCollector): + def setUp(self): - self._client = tritonhttpclient.InferenceServerClient("localhost:8000", verbose=True) + self._client = tritonhttpclient.InferenceServerClient("localhost:8000", + verbose=True) def _infer_helper(self): - # The model has a single variable which is added to the input. Since the - # variable is initialized to zero the input and output must match. - model_name = 'graphdef_variable' - input = np.array([10], dtype=np.int32) + # The model has a single variable which is added to the input. Since the + # variable is initialized to zero the input and output must match. + model_name = 'graphdef_variable' + input = np.array([10], dtype=np.int32) - inputs = [] - inputs.append(tritonhttpclient.InferInput('INPUT', input.shape, 'INT32')) - inputs[-1].set_data_from_numpy(input) + inputs = [] + inputs.append(tritonhttpclient.InferInput('INPUT', input.shape, + 'INT32')) + inputs[-1].set_data_from_numpy(input) - outputs = [] - outputs.append(tritonhttpclient.InferRequestedOutput('OUTPUT')) + outputs = [] + outputs.append(tritonhttpclient.InferRequestedOutput('OUTPUT')) - results = self._client.infer(model_name=model_name, inputs=inputs, outputs=outputs) - output = results.as_numpy('OUTPUT') - np.testing.assert_array_equal(output, input) + results = self._client.infer(model_name=model_name, + inputs=inputs, + outputs=outputs) + output = results.as_numpy('OUTPUT') + np.testing.assert_array_equal(output, input) def test_tf_variable(self): - self._infer_helper() + self._infer_helper() def test_tf_variable_error(self): - with self.assertRaises(tritonclient.utils.InferenceServerException) as e: - self._infer_helper() - self.assertIn("Attempting to use uninitialized value VARIABLE", e.exception.message()) + with self.assertRaises( + tritonclient.utils.InferenceServerException) as e: + self._infer_helper() + self.assertIn("Attempting to use uninitialized value VARIABLE", + e.exception.message()) if __name__ == '__main__': diff --git a/qa/common/check_copyright.py b/qa/common/check_copyright.py index 78022b118b..c4c181a3a2 100755 --- a/qa/common/check_copyright.py +++ b/qa/common/check_copyright.py @@ -49,7 +49,10 @@ 'qa/L0_model_config/noautofill_platform', 'qa/L0_model_config/autofill_noplatform', 'qa/L0_model_config/autofill_noplatform_success', - 'qa/L0_model_config/special_cases', 'qa/L0_perf_nomodel/baseline', + 'qa/L0_model_config/special_cases', + 'qa/L0_model_config/cli_messages/cli_override/expected', + 'qa/L0_model_config/cli_messages/cli_deprecation/expected', + 'qa/L0_perf_nomodel/baseline', 'qa/L0_perf_nomodel/legacy_baseline', 'qa/L0_warmup/raw_mug_data', 'qa/L0_java_resnet/expected_output_data', 'TRITON_VERSION')