fix: Fix feature attribution drift visualization for model monitoring…

… SDK PiperOrigin-RevId: 633754094
googleapis · May 15, 2024 · 710f33d · 710f33d
1 parent 6bdcfb3
commit 710f33d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vertexai/resources/preview/ml_monitoring/model_monitors.py b/vertexai/resources/preview/ml_monitoring/model_monitors.py
@@ -121,7 +121,7 @@ def _visualize_feature_attribution(feature_attribution_output: str) -> None:
     import tensorflow as tf
 
     with tf.io.gfile.GFile(feature_attribution_output, "r") as f:
-        return json.loads(f.read())
+        print(json.dumps(json.loads(f.read()), indent=4))
 
 
 def _feature_drift_stats_output_path(output_directory: str, job_id: str) -> (str, str):