Add a getter for communication data.

machine_learning/pipeline.py

@@ -50,6 +50,8 @@ class SensorFeatures:
     def get_sensor_data(self, data_type) -> pd.DataFrame:
         if data_type == "proximity":
             return self.df_proximity
+        elif data_type == "communication":
+            return self.df_calls_sms
         else:
             raise KeyError("This data type has not been implemented.")