Skip to content

Commit bfea3e9

Browse files
committed
Fix README.md with 23.08
1 parent d82974e commit bfea3e9

File tree

3 files changed

+11
-8
lines changed

3 files changed

+11
-8
lines changed

README.md

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -80,14 +80,11 @@ if __name__ == "__main__":
8080
### With Triton
8181

8282
```bash
83-
docker run --rm \
84-
-v ${PWD}:/models \
85-
nvcr.io/nvidia/tritonserver:22.01-pyt-python-py3 \
86-
tritonserver --model-repo=/models
83+
./bin/run_triton_tritony_sample.sh
8784
```
8885

8986
```bash
90-
pytest -m -s tests/test_tritony.py
87+
pytest -s --cov-report term-missing --cov=tritony tests/
9188
```
9289

9390
### Example with image_client.py

model_repository/sample_optional/config.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,12 @@ input [
2020
data_type: TYPE_FP32
2121
optional: true
2222
dims: [ -1 ]
23+
},
24+
{
25+
name: "optional_model_string"
26+
data_type: TYPE_STRING
27+
optional: true
28+
dims: [ -1 ]
2329
}
2430
]
2531

tests/test_connect.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,10 @@ def test_basics(protocol_and_port):
2525

2626
sample = np.random.rand(1, 100).astype(np.float32)
2727
result = client(sample)
28-
print(f"Result: {np.isclose(result, sample).all()}")
28+
assert np.isclose(result, sample).all()
2929

3030
result = client({"model_in": sample})
31-
print(f"Dict Result: {np.isclose(result, sample).all()}")
31+
assert np.isclose(result, sample).all()
3232

3333

3434
def test_batching(protocol_and_port):
@@ -40,7 +40,7 @@ def test_batching(protocol_and_port):
4040
sample = np.random.rand(100, 100).astype(np.float32)
4141
# client automatically makes sub batches with (50, 2, 100)
4242
result = client(sample)
43-
print(f"Result: {np.isclose(result, sample).all()}")
43+
assert np.isclose(result, sample).all()
4444

4545

4646
def test_exception(protocol_and_port):

0 commit comments

Comments
 (0)