diff --git a/HowBadIsMyBatch.ipynb b/HowBadIsMyBatch.ipynb index 9b09fc4307b..b587a747521 100644 --- a/HowBadIsMyBatch.ipynb +++ b/HowBadIsMyBatch.ipynb @@ -37,22 +37,22 @@ " def readVaersDescr(self, year):\n", " folder = self.dataDir + \"/\" + year + \"VAERSData/\"\n", " return {\n", - " 'VAERSDATA': self._readVAERSDATA(folder, year),\n", - " 'VAERSVAX': self._readVAERSVAX(folder, year)\n", - " }\n", + " 'VAERSDATA': self._readVAERSDATA(folder + year + \"VAERSDATA.csv\"),\n", + " 'VAERSVAX': self._readVAERSVAX(folder + year + \"VAERSVAX.csv\")\n", + " }\n", "\n", - " def _readVAERSDATA(self, folder, year):\n", + " def _readVAERSDATA(self, file):\n", " VAERSDATA = self._read_csv(\n", - " file = folder + year + \"VAERSDATA.csv\",\n", + " file = file,\n", " usecols = ['VAERS_ID', 'DIED', 'L_THREAT', 'DISABLE', 'HOSPITAL', 'ER_VISIT'])\n", " VaersDescrReader._convertColumnsOfDataFrameToNumerics(\n", " VAERSDATA,\n", " ['DIED', 'L_THREAT', 'DISABLE', 'HOSPITAL', 'ER_VISIT'])\n", " return VAERSDATA\n", "\n", - " def _readVAERSVAX(self, folder, year):\n", + " def _readVAERSVAX(self, file):\n", " return self._read_csv(\n", - " file = folder + year + \"VAERSVAX.csv\",\n", + " file = file,\n", " usecols = ['VAERS_ID', 'VAX_DOSE_SERIES', 'VAX_TYPE', 'VAX_MANU', 'VAX_LOT'],\n", " dtype = {\"VAX_DOSE_SERIES\": \"string\"})\n", "\n",