diff --git a/tests/example_data/data_loading/pandas/pandas_data_loader.py b/tests/example_data/data_loading/pandas/pandas_data_loader.py index 9b35bbb3721e0..4a0aca7d54761 100644 --- a/tests/example_data/data_loading/pandas/pandas_data_loader.py +++ b/tests/example_data/data_loading/pandas/pandas_data_loader.py @@ -76,7 +76,7 @@ def _take_data_types(self, table: Table) -> dict[str, str] | None: def remove_table(self, table_name: str) -> None: with self._db_engine.connect() as connection: - connection.execute(text(f"DROP TABLE IF EXISTS {table_name}")) + connection.execute(text(f"DROP TABLE IF EXISTS {table_name} CASCADE")) class TableToDfConvertor(ABC):