|
| 1 | +package co.clflushopt.glint.core.datasets; |
| 2 | + |
| 3 | +import java.util.Arrays; |
| 4 | + |
| 5 | +import co.clflushopt.glint.types.ArrowTypes; |
| 6 | +import co.clflushopt.glint.types.Field; |
| 7 | +import co.clflushopt.glint.types.Schema; |
| 8 | + |
| 9 | +public class DatasetUtils { |
| 10 | + |
| 11 | + public static final Schema getNYCYellowTripsCSVSchema() { |
| 12 | + return new Schema(Arrays.asList(new Field("VendorID", ArrowTypes.Int32Type), |
| 13 | + new Field("tpep_pickup_datetime", ArrowTypes.StringType), |
| 14 | + new Field("tpep_dropoff_datetime", ArrowTypes.StringType), |
| 15 | + new Field("passenger_count", ArrowTypes.Int32Type), |
| 16 | + new Field("trip_distance", ArrowTypes.DoubleType), |
| 17 | + new Field("RatecodeID", ArrowTypes.Int32Type), |
| 18 | + new Field("store_and_fwd_flag", ArrowTypes.StringType), |
| 19 | + new Field("PULocationID", ArrowTypes.Int32Type), |
| 20 | + new Field("DOLocationID", ArrowTypes.Int32Type), |
| 21 | + new Field("payment_type", ArrowTypes.Int32Type), |
| 22 | + new Field("fare_amount", ArrowTypes.DoubleType), |
| 23 | + new Field("extra", ArrowTypes.DoubleType), |
| 24 | + new Field("mta_tax", ArrowTypes.DoubleType), |
| 25 | + new Field("tip_amount", ArrowTypes.DoubleType), |
| 26 | + new Field("tolls_amount", ArrowTypes.DoubleType), |
| 27 | + new Field("improvement_surcharge", ArrowTypes.DoubleType), |
| 28 | + new Field("total_amount", ArrowTypes.DoubleType))); |
| 29 | + } |
| 30 | + |
| 31 | +} |
0 commit comments