Fix bug with --skip-large-nb when error message not in resp

databrickslabs · Jul 14, 2023 · f7d92de · f7d92de
1 parent af01d06
commit f7d92de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dbclient/WorkspaceClient.py b/dbclient/WorkspaceClient.py
@@ -351,7 +351,7 @@ def download_notebook_helper(self, notebook_data, checkpoint_notebook_set, error
  logging_utils.log_response_error(error_logger, resp)
  return resp
  if resp.get('error_code', None):
- if self.skip_large_nb and resp.message == 'Size exceeds 10485760 bytes':
+ if self.skip_large_nb and resp.get(resp.message, None) == 'Size exceeds 10485760 bytes':
  logging.info("Notebook {} skipped due to size exceeding limit".format(notebook_path))
  else:
  resp['path'] = notebook_path