diff --git a/odds/common/filters.py b/odds/common/filters.py index 0059eb4..2adfd40 100644 --- a/odds/common/filters.py +++ b/odds/common/filters.py @@ -64,13 +64,13 @@ async def index(self, dataset: Dataset) -> bool: class DatasetFilterNew(DatasetFilter): - async def consider(self) -> bool: + async def consider(self, dataset: Dataset) -> bool: return not await store.hasDataset(self.dataset.storeId()) class DatasetFilterForce(DatasetFilter): - async def consider(self) -> bool: + async def consider(self, dataset: Dataset) -> bool: return True @@ -79,8 +79,8 @@ def __init__(self, datasetId: str): super().__init__() self.datasetId = datasetId - async def consider(self) -> bool: - return self.dataset.id == self.datasetId + async def consider(self, dataset: Dataset) -> bool: + return dataset.id == self.datasetId class CatalogFilter: