Kill unnecessary check on --safety-shield test param

This commit is contained in:
Ashwin Bharambe 2025-02-20 09:15:23 -08:00
parent 996f27a308
commit 984a8039ad

View file

@ -75,7 +75,6 @@ def pytest_generate_tests(metafunc):
if "safety_shield" in metafunc.fixturenames:
shield_id = metafunc.config.getoption("--safety-shield")
if shield_id:
assert shield_id.startswith("meta-llama/")
params = [pytest.param(shield_id, id="")]
else:
params = SAFETY_SHIELD_PARAMS