diff --git a/serenata_toolbox/chamber_of_deputies/dataset.py b/serenata_toolbox/chamber_of_deputies/dataset.py index 6875cb5..552228c 100644 --- a/serenata_toolbox/chamber_of_deputies/dataset.py +++ b/serenata_toolbox/chamber_of_deputies/dataset.py @@ -14,7 +14,7 @@ class Dataset: def __init__(self, path, years=AVAILABLE_YEARS): self.path = path - self.years = years + self.years = years if isinstance(years, list) else [years] def fetch(self): base_url = "http://www.camara.leg.br/cotas/Ano-{}.csv.zip" diff --git a/serenata_toolbox/federal_senate/dataset.py b/serenata_toolbox/federal_senate/dataset.py index 100148e..a8d3d84 100644 --- a/serenata_toolbox/federal_senate/dataset.py +++ b/serenata_toolbox/federal_senate/dataset.py @@ -12,7 +12,7 @@ class Dataset: def __init__(self, path, years=AVAILABLE_YEARS): self.path = path - self.years = years + self.years = years if isinstance(years, list) else [years] def fetch(self): retrieved_files = []