Initialize config provider when running inside DB notebooks (#194)

databricks_cli/__init__.py

@@ -20,3 +20,18 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+# pylint:disable=import-error
+# pylint:disable=bare-except
+
+
+def initialize_cli_for_databricks_notebooks():
+    import PythonShell
+    from databricksCli import init_databricks_cli_config_provider
+    init_databricks_cli_config_provider(PythonShell.IPython.get_ipython().user_ns.entry_point)
+
+try:
+    # Initialize custom config provider which is available in Databricks notebooks.
+    initialize_cli_for_databricks_notebooks()
+except:
+    pass