Update sql_magic.py

vertica · Oct 27, 2023 · 7861d29 · 7861d29
1 parent 4344310
commit 7861d29
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/verticapy/jupyter/extensions/sql_magic.py b/verticapy/jupyter/extensions/sql_magic.py
@@ -926,8 +926,9 @@ def sql_magic(
                         usecols=columns,
                     )
 
-                try:
-                    if not (query_subtype.upper().startswith(SPECIAL_WORDS)):
+                is_vdf = False
+                if not(query_subtype.upper().startswith(SPECIAL_WORDS)):
+                    try:
                         result = create_new_vdf(
                             query,
                             _is_sql_magic=True,
@@ -938,10 +939,12 @@ def sql_magic(
                             result._vars["max_rows"] = options["-nrows"]
                         if "-ncols" in options:
                             result._vars["max_columns"] = options["-ncols"]
-                    else:
-                        raise
+                        is_vdf = True
+                    except:
+                        pass # we could not create a vDataFrame out of the query.
+
+                if not(is_vdf):
 
-                except:
                     try:
                         final_result = _executeSQL(
                             query, method="fetchfirstelem", print_time_sql=False