From d3d173ddc3af4d3c8eb73f0160370e96de483b35 Mon Sep 17 00:00:00 2001
From: Rok Roskar <rokroskar@gmail.com>
Date: Thu, 9 Apr 2020 14:52:39 +0200
Subject: [PATCH] fix: ignore missing tested column
---
.../covid_19_utils/converters/__init__.py | 13 +++++++++----
1 file changed, 9 insertions(+), 4 deletions(-)
diff --git a/src/covid-19/covid_19_utils/covid_19_utils/converters/__init__.py b/src/covid-19/covid_19_utils/covid_19_utils/converters/__init__.py
index a698cefa4..5135c5477 100644
--- a/src/covid-19/covid_19_utils/covid_19_utils/converters/__init__.py
+++ b/src/covid-19/covid_19_utils/covid_19_utils/converters/__init__.py
@@ -87,14 +87,19 @@ class CaseConverterImpl:
except ValueError:
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:
if column not in df.columns:
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]
@classmethod
--
GitLab