make optional int

apache · Aug 10, 2023 · dd9c589 · dd9c589
1 parent cd782d1
commit dd9c589
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/superset/sqllab/utils.py b/superset/sqllab/utils.py
@@ -74,7 +74,7 @@ def write_ipc_buffer(table: pa.Table) -> pa.Buffer:
 
     return sink.getvalue()
 
-def bootstrap_sqllab_data(user_id: int) -> dict[str, Any]:
+def bootstrap_sqllab_data(user_id: int | None) -> dict[str, Any]:
     # send list of tab state ids
     tabs_state = (
         db.session.query(TabState.id, TabState.label)

diff --git a/tests/integration_tests/core_tests.py b/tests/integration_tests/core_tests.py
@@ -1137,7 +1137,7 @@ def test_sqllab_backend_persistence_payload(self):
         # we should have only 1 query returned, since the second one is not
         # associated with any tabs
         # TODO: replaces this spec by api/v1/sqllab spec later
-        payload = bootstrap_sqllab_data(int(user_id))
+        payload = bootstrap_sqllab_data(user_id)
         self.assertEqual(len(payload["queries"]), 1)
 
     @mock.patch.dict(