def _get_vehicle_properties_headers()

in sagemaker/source/dl_utils/dataset.py [0:0]

• 11 lines of code
• 5 McCabe index (conditional complexity)

    def _get_vehicle_properties_headers(self, df):
        vehicle_properties = []
        for col in df.columns:
            col_sensor_header = False
            for sensor_header in self.sensor_headers:
                if sensor_header in col:
                    col_sensor_header = True
            if not col_sensor_header:
                vehicle_properties.append(col)
        vehicle_properties.remove(self.target_column)
        return vehicle_properties