From 04618e0563b8588eec2ccd8342c6085ca08b5adb Mon Sep 17 00:00:00 2001
From: Taisei Klasen <taiseiklasen@gmail.com>
Date: Thu, 23 Dec 2021 08:53:04 -0800
Subject: [PATCH] fix: use open_in_new_tab in the render method. (#926)

---
 google/cloud/aiplatform/explain/lit.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/google/cloud/aiplatform/explain/lit.py b/google/cloud/aiplatform/explain/lit.py
index 432dd9194a..c76840502d 100644
--- a/google/cloud/aiplatform/explain/lit.py
+++ b/google/cloud/aiplatform/explain/lit.py
@@ -206,8 +206,8 @@ def open_lit(
         Raises:
             ImportError if LIT is not installed.
     """
-    widget = notebook.LitWidget(models, datasets, open_in_new_tab=open_in_new_tab)
-    widget.render()
+    widget = notebook.LitWidget(models, datasets)
+    widget.render(open_in_new_tab=open_in_new_tab)
 
 
 def set_up_and_open_lit(