fix: use open_in_new_tab in the render method. (#926)

googleapis · Dec 23, 2021 · 04618e0 · 04618e0
1 parent 8df74a2
commit 04618e0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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(