adding the IPython NoneType handler condition

Author: souvik-databricks

python/tempo/utils.py

@@ -18,9 +18,12 @@
 
 
 def __notdltruntime():
-    user_namespace_keys = list(get_ipython().user_ns.keys())
-    check_bool = (('create_dlt_table_fn' not in user_namespace_keys) and ('dlt_sql_fn' not in user_namespace_keys))
-    return check_bool
+    if type(get_ipython()) != type(None):
+        user_namespace_keys = list(get_ipython().user_ns.keys())
+        check_bool = (('create_dlt_table_fn' not in user_namespace_keys) and ('dlt_sql_fn' not in user_namespace_keys))
+        return check_bool
+    else:
+        return False
 
 
 def __isnotebookenv():
@@ -70,6 +73,7 @@ def display_unavailable(df):
     if __notdltruntime():
         method = get_ipython().user_ns['display']
 
+
         # Under 'display' key in user_ns the original databricks display method is present
         # to know more refer: /databricks/python_shell/scripts/db_ipykernel_launcher.py
         def display_improvised(obj):