File tree Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -376,6 +376,10 @@ def test_stop(self):
376
376
377
377
server .stop ()
378
378
379
+ def test_model_repository_not_specified (self ):
380
+ with self .assertRaises (tritonserver .InvalidArgumentError ):
381
+ tritonserver .Server (model_repository = None ).start ()
382
+
379
383
380
384
class InferenceTests (unittest .TestCase ):
381
385
def setup_method (self , method ):
@@ -452,7 +456,3 @@ def test_basic_inference(self):
452
456
):
453
457
fp16_output = numpy .from_dlpack (response .outputs ["fp16_output" ])
454
458
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 )
Original file line number Diff line number Diff line change @@ -343,6 +343,8 @@ def _create_tritonserver_server_options(
343
343
344
344
options .set_server_id (self .server_id )
345
345
346
+ if self .model_repository is None :
347
+ raise InvalidArgumentError ("Model repository must be specified." )
346
348
if not isinstance (self .model_repository , list ):
347
349
self .model_repository = [self .model_repository ]
348
350
for model_repository_path in self .model_repository :
@@ -526,8 +528,6 @@ def __init__(
526
528
if options is None :
527
529
options = Options (** kwargs )
528
530
self .options : Options = options
529
- if self .options .model_repository is None :
530
- raise InvalidArgumentError ("Model repository must be specified." )
531
531
self ._server = Server ._UnstartedServer ()
532
532
533
533
def start (
You can’t perform that action at this time.
0 commit comments