import tensorflow_data_validation as tfdv

# Use tensorflow_data_validation to generate initial schema for validation
train_stats = tfdv.generate_statistics_from_csv(data_location="/data/train.csv")
schema = tfdv.infer_schema(statistics=train_stats)

# Create a new FeatureSet or retrieve an existing FeatureSet in Feast
feature_set = FeatureSet(
            name="myfeatureset",
            entities=[Entity(name="id", dtype=ValueType.INT64),],
            features=[
                Feature(name="feature1", dtype=ValueType.STRING),
                Feature(name="feature2", dtype=ValueType.INT), ...
            ],
        )
# Update the entities and features with constraints defined in the schema
feature_set.update_schema(schema)