Skip to content

Commit

Permalink
look at the right var for NIM name checking (NVIDIA#871)
Browse files Browse the repository at this point in the history
should be `self.name` not `name`
  • Loading branch information
leondz authored Sep 3, 2024
2 parents 8b2cd96 + cbfc9f7 commit 175263c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion garak/generators/nim.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ def _call_model(

def __init__(self, name="", config_root=_config):
super().__init__(name, config_root=config_root)
if "/" not in name:
if "/" not in self.name:
msg = "❓ Is this a valid NIM name? expected a slash-formatted name, e.g. 'org/model'"
logging.info(msg)
print(msg)
Expand Down

0 comments on commit 175263c

Please sign in to comment.