From d97533f299cb17fb3e94499c46ba8c8a61640671 Mon Sep 17 00:00:00 2001
From: Luke Irvin <lirvin@sparkfish.com>
Date: Thu, 19 Oct 2023 16:07:02 -0700
Subject: [PATCH] Updated expected error message text for when attempting to
 download excessively large notebooks

---
 dbclient/WorkspaceClient.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dbclient/WorkspaceClient.py b/dbclient/WorkspaceClient.py
index ffd61cc..331b83b 100644
--- 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.get("message", None) == 'Size exceeds 10485760 bytes':
+            if self.skip_large_nb and re.search(r"Notebook size exceeded limit: \d+ > 10485760", resp.get("message", '')):
                 logging.info("Notebook {} skipped due to size exceeding limit".format(notebook_path))
             else:
                 resp['path'] = notebook_path