fix: Fix feature attribution drift visualization for model monitoring SDK

vertex-sdk-bot · copybara-github · commit 710f33d87e24 · 2024-05-14T17:20:37.000-07:00
PiperOrigin-RevId: 633754094
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):
