diff --git a/rialto/runner/runner.py b/rialto/runner/runner.py index 3fc13b4..9f9b276 100644 --- a/rialto/runner/runner.py +++ b/rialto/runner/runner.py @@ -121,7 +121,7 @@ def check_dates_have_partition(self, table: Table, dates: List[date]) -> List[bo :param dates: list of dates to check :return: list of bool """ - if utils.table_exists(table.get_table_path()): + if utils.table_exists(self.spark, table.get_table_path()): partitions = utils.get_partitions(self.reader, table) return [(date in partitions) for date in dates] else: