refining VaersDescrReaderTest

2023-01-18 17:26:25 +01:00
parent fc62639a1b
commit bc87b9d34d
2 changed files with 23 additions and 9 deletions
--- a/src/VaersDescrReader.py
+++ b/src/VaersDescrReader.py
@@ -31,7 +31,11 @@ class VaersDescrReader:
        return self._read_csv(
            file = file,
            usecols = ['VAERS_ID', 'VAX_DOSE_SERIES', 'VAX_TYPE', 'VAX_MANU', 'VAX_LOT'],
-            dtype = {"VAX_DOSE_SERIES": "string"})
+            dtype =
+                {
+                    "VAX_DOSE_SERIES": "string",
+                    "VAX_LOT": "string"
+                })

    def _read_csv(self, file, **kwargs):
        return pd.read_csv(