Skip to content

Commit 6e91ff3

Browse files
authored
updated check for model repo to align with checks for other options
1 parent f0ff1e5 commit 6e91ff3

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

python/test/test_api.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -376,6 +376,10 @@ def test_stop(self):
376376

377377
server.stop()
378378

379+
def test_model_repository_not_specified(self):
380+
with self.assertRaises(tritonserver.InvalidArgumentError):
381+
tritonserver.Server(model_repository=None).start()
382+
379383

380384
class InferenceTests(unittest.TestCase):
381385
def setup_method(self, method):
@@ -452,7 +456,3 @@ def test_basic_inference(self):
452456
):
453457
fp16_output = numpy.from_dlpack(response.outputs["fp16_output"])
454458
numpy.testing.assert_array_equal(fp16_input, fp16_output)
455-
456-
def test_model_repository_not_specified(self):
457-
with self.assertRaises(tritonserver.InvalidArgumentError):
458-
tritonserver.Server(model_repository=None)

python/tritonserver/_api/_server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -343,6 +343,8 @@ def _create_tritonserver_server_options(
343343

344344
options.set_server_id(self.server_id)
345345

346+
if self.model_repository is None:
347+
raise InvalidArgumentError("Model repository must be specified.")
346348
if not isinstance(self.model_repository, list):
347349
self.model_repository = [self.model_repository]
348350
for model_repository_path in self.model_repository:
@@ -526,8 +528,6 @@ def __init__(
526528
if options is None:
527529
options = Options(**kwargs)
528530
self.options: Options = options
529-
if self.options.model_repository is None:
530-
raise InvalidArgumentError("Model repository must be specified.")
531531
self._server = Server._UnstartedServer()
532532

533533
def start(

0 commit comments

Comments
 (0)