diff --git a/src/HowBadIsMyBatch.ipynb b/src/HowBadIsMyBatch.ipynb index 13bfc0a8733..14afaa20aa3 100644 --- a/src/HowBadIsMyBatch.ipynb +++ b/src/HowBadIsMyBatch.ipynb @@ -80,7 +80,10 @@ "metadata": {}, "outputs": [], "source": [ - "updateVAERSFiles(needsUpdate, getWorkingDirectory())" + "updateVAERSFiles(\n", + " needsUpdate,\n", + " years = [2022],\n", + " workingDirectory = getWorkingDirectory())" ] }, { diff --git a/src/VAERSFileDownloader.py b/src/VAERSFileDownloader.py index 55d9edc4567..3ae21783d97 100644 --- a/src/VAERSFileDownloader.py +++ b/src/VAERSFileDownloader.py @@ -56,7 +56,8 @@ def downloadVAERSFileAndUnzip(file, workingDirectory): zipFile = downloadedFile, dstDir = workingDirectory + '/VAERS/') -def updateVAERSFiles(needsUpdate, workingDirectory): +def updateVAERSFiles(needsUpdate, years, workingDirectory): if needsUpdate: - downloadVAERSFileAndUnzip('2022VAERSData.zip', workingDirectory) + for year in years: + downloadVAERSFileAndUnzip(f'{year}VAERSData.zip', workingDirectory) downloadVAERSFileAndUnzip('NonDomesticVAERSData.zip', workingDirectory) \ No newline at end of file