diff --git a/src/axolotl/utils/callbacks/vessl_.py b/src/axolotl/utils/callbacks/vessl_.py index f5d051d8f7..4d436d6161 100644 --- a/src/axolotl/utils/callbacks/vessl_.py +++ b/src/axolotl/utils/callbacks/vessl_.py @@ -10,7 +10,7 @@ class VesslLogMetricsCallback(TrainerCallback): - def __init__(self, credential_path: str, metrics: List[str]) -> None: + def __init__(self, credential_path: str) -> None: vessl.configure(credentials_file=credential_path) def on_log(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, logs: Dict[str, float] = None, **kwargs):