def __init__()

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

• 15 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self,
                 fleet_info_fn='../../data/example_fleet_info.csv',
                 fleet_sensor_logs_fn='../../data/example_fleet_sensor_logs.csv',
                 fleet_statistics_fn='../../data/generation/fleet_statistics.csv',
                 num_vehicles=100,
                 num_sensor_readings=50,
                 period_ms=30000):
        self.statistics_df = pd.read_csv(fleet_statistics_fn)
        self.fleet_info_fn=fleet_info_fn
        self.fleet_sensor_logs_fn=fleet_sensor_logs_fn
        self.num_vehicles = num_vehicles
        self.size_per_type = int(num_vehicles/len(self.statistics_df))  # TODO: allow smaller values of num_vehicles.
        self.start_time = datetime(2020,1,1,0,0,0,0)
        self.time_delta = timedelta(microseconds=period_ms*1000)
        self.num_sensor_readings = num_sensor_readings