Skip to content

Commit 4f88611

Browse files
committed
Add a mock --custom-namespace option to enable run of MR with custom namespace
1 parent c381325 commit 4f88611

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

conftest.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff 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

195202
def pytest_cmdline_main(config: Any) -> None:

0 commit comments

Comments
 (0)