import IPython directly instead of PythonShell (#331)

mengxr · web-flow · commit 1a86501a0d2f · 2020-10-07T11:18:50.000-07:00
diff --git a/databricks_cli/__init__.py b/databricks_cli/__init__.py
@@ -26,9 +26,9 @@
 
 
 def initialize_cli_for_databricks_notebooks():
-    import PythonShell
+    import IPython
     from databricksCli import init_databricks_cli_config_provider
-    init_databricks_cli_config_provider(PythonShell.IPython.get_ipython().user_ns.entry_point)
+    init_databricks_cli_config_provider(IPython.get_ipython().user_ns.entry_point)
 
 
 try:
