sagemaker/source/dataset/dataset_generator.py (1 line): - line 17: self.size_per_type = int(num_vehicles/len(self.statistics_df)) # TODO: allow smaller values of num_vehicles. sagemaker/source/preprocessing/preprocessing.py (1 line): - line 46: # TODO: The order of the columns seems to be the same....