Fixing import

apache · Apr 5, 2022 · fe0ba91 · fe0ba91
1 parent 757ece8
commit fe0ba91
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/integration_tests/utils_tests.py b/tests/integration_tests/utils_tests.py
@@ -24,6 +24,8 @@
 import re
 from typing import Any, Tuple, List, Optional
 from unittest.mock import Mock, patch
+
+from superset.databases.commands.exceptions import DatabaseInvalidError
 from tests.integration_tests.fixtures.birth_names_dashboard import (
     load_birth_names_dashboard_with_slices,
     load_birth_names_data,