Skip to content

Commit 5fdd7b0

Browse files
jbkyang-nvimc-nv
authored andcommitted
fix formatting and L0 copyright
1 parent f15f0a2 commit 5fdd7b0

File tree

2 files changed

+28
-18
lines changed

2 files changed

+28
-18
lines changed

qa/L0_tf_parameters/tf_parameter_test.py

+24-17
Original file line numberDiff line numberDiff line change
@@ -35,33 +35,40 @@
3535

3636

3737
class TFParameterTest(tu.TestResultCollector):
38+
3839
def setUp(self):
39-
self._client = tritonhttpclient.InferenceServerClient("localhost:8000", verbose=True)
40+
self._client = tritonhttpclient.InferenceServerClient("localhost:8000",
41+
verbose=True)
4042

4143
def _infer_helper(self):
42-
# The model has a single variable which is added to the input. Since the
43-
# variable is initialized to zero the input and output must match.
44-
model_name = 'graphdef_variable'
45-
input = np.array([10], dtype=np.int32)
44+
# The model has a single variable which is added to the input. Since the
45+
# variable is initialized to zero the input and output must match.
46+
model_name = 'graphdef_variable'
47+
input = np.array([10], dtype=np.int32)
4648

47-
inputs = []
48-
inputs.append(tritonhttpclient.InferInput('INPUT', input.shape, 'INT32'))
49-
inputs[-1].set_data_from_numpy(input)
49+
inputs = []
50+
inputs.append(tritonhttpclient.InferInput('INPUT', input.shape,
51+
'INT32'))
52+
inputs[-1].set_data_from_numpy(input)
5053

51-
outputs = []
52-
outputs.append(tritonhttpclient.InferRequestedOutput('OUTPUT'))
54+
outputs = []
55+
outputs.append(tritonhttpclient.InferRequestedOutput('OUTPUT'))
5356

54-
results = self._client.infer(model_name=model_name, inputs=inputs, outputs=outputs)
55-
output = results.as_numpy('OUTPUT')
56-
np.testing.assert_array_equal(output, input)
57+
results = self._client.infer(model_name=model_name,
58+
inputs=inputs,
59+
outputs=outputs)
60+
output = results.as_numpy('OUTPUT')
61+
np.testing.assert_array_equal(output, input)
5762

5863
def test_tf_variable(self):
59-
self._infer_helper()
64+
self._infer_helper()
6065

6166
def test_tf_variable_error(self):
62-
with self.assertRaises(tritonclient.utils.InferenceServerException) as e:
63-
self._infer_helper()
64-
self.assertIn("Attempting to use uninitialized value VARIABLE", e.exception.message())
67+
with self.assertRaises(
68+
tritonclient.utils.InferenceServerException) as e:
69+
self._infer_helper()
70+
self.assertIn("Attempting to use uninitialized value VARIABLE",
71+
e.exception.message())
6572

6673

6774
if __name__ == '__main__':

qa/common/check_copyright.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,10 @@
4949
'qa/L0_model_config/noautofill_platform',
5050
'qa/L0_model_config/autofill_noplatform',
5151
'qa/L0_model_config/autofill_noplatform_success',
52-
'qa/L0_model_config/special_cases', 'qa/L0_perf_nomodel/baseline',
52+
'qa/L0_model_config/special_cases',
53+
'qa/L0_model_config/cli_messages/cli_override/expected',
54+
'qa/L0_model_config/cli_messages/cli_deprecation/expected',
55+
'qa/L0_perf_nomodel/baseline',
5356
'qa/L0_perf_nomodel/legacy_baseline', 'qa/L0_warmup/raw_mug_data',
5457
'qa/L0_java_resnet/expected_output_data', 'TRITON_VERSION')
5558

0 commit comments

Comments
 (0)