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

taiseiak · web-flow · commit 04618e0563b8 · 2021-12-23T11:53:04.000-05:00
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(
