From 36f84bb255fe3ef669b7daf99aaf3d8de88a88fd Mon Sep 17 00:00:00 2001 From: PRINCE KUMAR Date: Fri, 6 Oct 2023 16:07:27 +0530 Subject: [PATCH] fix: refactor AimCallback to BaseCallback in huggingface tracker --- pkgs/aimstack/hugging_face_tracker/callbacks/base_callback.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pkgs/aimstack/hugging_face_tracker/callbacks/base_callback.py b/pkgs/aimstack/hugging_face_tracker/callbacks/base_callback.py index c574515a7e..849452713b 100644 --- a/pkgs/aimstack/hugging_face_tracker/callbacks/base_callback.py +++ b/pkgs/aimstack/hugging_face_tracker/callbacks/base_callback.py @@ -110,7 +110,7 @@ def on_log(self, args, state, control, model=None, logs=None, **kwargs): log_name = log_name[len(prefix):] context = {'subset': prefix[:-1]} if '_' in log_name: - sub_dataset = AimCallback.find_most_common_substring( + sub_dataset = BaseCallback.find_most_common_substring( list(logs.keys()) ).split(prefix)[-1] if sub_dataset != prefix.rstrip('_'):