Fixes get_definitions failing when get_data wasn't called previously

socialfoundations · May 13, 2024 · 086003d · 086003d
1 parent d78adc5
commit 086003d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/folktables/load_acs.py b/folktables/load_acs.py
@@ -149,6 +149,8 @@ def load_definitions(root_dir, year=2018, horizon='1-Year', download=False):
     year_string = year if horizon == '1-Year' else f'{year - 4}-{year}'
     url = f'https://www2.census.gov/programs-surveys/acs/tech_docs/pums/data_dict/PUMS_Data_Dictionary_{year_string}.csv'
 
+    os.makedirs(base_datadir, exist_ok=True)
+
     response = requests.get(url)
     with open(file_path, 'wb') as handle:
         handle.write(response.content)