Skip to content
Snippets Groups Projects
Commit d3d173dd authored by Rok Roškar's avatar Rok Roškar
Browse files

fix: ignore missing tested column

parent f7f81d64
No related branches found
No related tags found
1 merge request!109fix: ignore missing tested column
Pipeline #23194 passed with stage
in 21 seconds
...@@ -87,14 +87,19 @@ class CaseConverterImpl: ...@@ -87,14 +87,19 @@ class CaseConverterImpl:
except ValueError: except ValueError:
pass pass
df["positive_100k"] = df["positive"] / df["population"] * 100000
df["deceased_100k"] = df["deceased"] / df["population"] * 100000
df["tested_100k"] = df["tested"] / df["population"] * 100000
for column in self.common_columns: for column in self.common_columns:
if column not in df.columns: if column not in df.columns:
df[column] = "" df[column] = ""
df["positive_100k"] = df["positive"] / df["population"] * 100000
df["deceased_100k"] = df["deceased"] / df["population"] * 100000
# lazy solution to tested missing - todo: implement types checks for columns
try:
df["tested_100k"] = df["tested"] / df["population"] * 100000
except TypeError:
pass
return df[self.common_columns] return df[self.common_columns]
@classmethod @classmethod
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment