change default task_tracker input to integer

awslabs · Jan 9, 2024 · 089d977 · 089d977
1 parent 5b2d62e
commit 089d977
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/python/graphstorm/gsf.py b/python/graphstorm/gsf.py
@@ -656,4 +656,4 @@ def check_homo(g):
 
 def create_builtin_task_tracker(config):
     tracker_class = get_task_tracker_class(config.task_tracker)
-    return tracker_class(config)
+    return tracker_class(config.log_report_frequency)
diff --git a/python/graphstorm/tracker/sagemaker_tracker.py b/python/graphstorm/tracker/sagemaker_tracker.py
@@ -25,8 +25,11 @@ class GSSageMakerTaskTracker(GSTaskTrackerAbc):
 
         Parameters
         ----------
-        config: GSConfig
-            Configurations. Users can add their own configures in the yaml config file.
+        log_report_frequency: int
+            The frequency of reporting model performance metrics through task_tracker. 
+            The frequency is defined by using number of iterations, i.e., every N iterations
+            the evaluation metrics will be reported.
+
     """
 
     def _do_report(self, step):