Only drop host_key if present.
@@ -111,6 +111,7 @@ class HostFootprint():
# Load data from host footprint .csv
df = pd.read_csv(self.path)
+ if 'host_key' in df.columns:
df = df.drop(columns=['host_key'])
df = df.fillna(0)
@@ -178,6 +179,7 @@ class HostFootprint():
# Load data from host footprint .csv
df = pd.read_csv(self.path)
+ if 'host_key' in df.columns:
df = df.drop(columns=['host_key'])
# Split dataframe into X (the input features or predictors)
