File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed
Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -53,7 +53,7 @@ def pytest_addoption(parser: Parser) -> None:
5353 serving_arguments_group = parser .getgroup (name = "Serving arguments" )
5454 model_validation_automation_group = parser .getgroup (name = "Model Validation Automation" )
5555 hf_group = parser .getgroup (name = "Hugging Face" )
56-
56+ model_registry_group = parser . getgroup ( name = "Model Registry options" )
5757 # AWS config and credentials options
5858 aws_group .addoption (
5959 "--aws-secret-access-key" ,
@@ -190,6 +190,13 @@ def pytest_addoption(parser: Parser) -> None:
190190
191191 # HuggingFace options
192192 hf_group .addoption ("--hf-access-token" , default = os .environ .get ("HF_ACCESS_TOKEN" ), help = "HF access token" )
193+ # Model Registry options
194+ model_registry_group .addoption (
195+ "--custom-namespace" ,
196+ default = False ,
197+ action = "store_true" ,
198+ help = "Indicates if the model registry tests are to be run against custom namespace" ,
199+ )
193200
194201
195202def pytest_cmdline_main (config : Any ) -> None :
You can’t perform that action at this time.
0 commit comments