diff --git a/python/graphstorm/gsf.py b/python/graphstorm/gsf.py
index 124ed576a6..1dfd3cf780 100644
--- 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
index 06276943bd..3ae5fd5a8a 100644
--- 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):