changed a name and fixed a ref

ml/src/tools/inputter.py

@@ -8,7 +8,7 @@ from typing import Tuple
 import pandas as pd
 import numpy as np
 from pandas._libs.tslibs import Timestamp
-from utility import data
+from utility.data import Data
 
 def get_ats(df: pd.DataFrame, end_date: Timestamp, settings: dict) -> str:
     """
@@ -78,7 +78,7 @@ def get_gender(sc: pd.DataFrame) -> int:
         return 0
     return 1
 
-def get_citizen_data(data: data.Data, citizen_id: str) -> data.Data:
+def get_citizen_data(data: Data, citizen_id: str) -> Data:
     """
     This method extracts all screening data we have on a citizen
     :param data: a data DTO consisting of screening data of all citizens
@@ -90,8 +90,8 @@ def get_citizen_data(data: data.Data, citizen_id: str) -> data.Data:
     training_done = data.td.loc[data.td['CitizenId'] == str(citizen_id)]
     training_cancelled = data.tc.loc[data.tc['CitizenId'] == str(citizen_id)]
     ats = data.ats.loc[data.ats['CitizenId'] == str(citizen_id)]
-    citizen_data = data.Data(screening_content, status_set,
-                                 training_done, training_cancelled, ats)
+    citizen_data = Data(screening_content, status_set,
+                        training_done, training_cancelled, ats)
     return citizen_data
 
 def get_screening_data(td: pd.DataFrame, tc: pd.DataFrame,