Error when accessing 'num_inserted_rows' in Spark SQL (DBR 15.4 LTS)

[Buranapat]

Hello Databricks Community,

I've encountered an issue while trying to capture the number of rows inserted after executing an SQL insert statement in Databricks (DBR 15.4 LTS). My code is attempting to access the number of inserted rows as follows:

rows_insert = spark.sql(insert_statement).first()['num_inserted_rows']

However, I am receiving the following error:

TypeError: 'NoneType' object is not subscriptable

Upon further investigation, I noticed that in the SQL execution result, the fields num_affected_rows and num_inserted_rows are shown, but no rows are actually returned. This can be seen in the screenshots I've attached:

• The first image shows the result of my SQL insert, which displays the columns num_affected_rows and num_inserted_rows but with no actual rows returned.
  
• The second image shows the subsequent Python error when I try to access num_inserted_rows from the result.
  

Has anyone else faced a similar issue? I would greatly appreciate any insights or suggestions on how to work around this problem.

Thanks in advance for your help!

[MuthuLakshmi]

@Buranapat Can you please try this check as a workaround?
result = spark.sql(insert_statement).first()
if result is not None and 'num_inserted_rows' in result:
          num_inserted_rows = result['num_inserted_rows']
else:
          num_inserted_rows = 0

[Buranapat]

Thank you for the suggestion! The workaround seems like a helpful way to quickly check insert success, though I’ve noticed that it sometimes returns None even when records appear to be successfully inserted. Do you know if this is a known issue, or if there are any plans to improve this behavior in future updates?

Thanks again for your help!