Exemple #1
0
    def from_proto(cls, feature_set_proto: FeatureSetSpecProto):
        """
        Creates a feature set from a protobuf representation of a feature set

        Args:
            from_proto: A protobuf representation of a feature set

        Returns:
            Returns a FeatureSet object based on the feature set protobuf
        """

        feature_set = cls(
            name=feature_set_proto.name,
            features=[
                Feature.from_proto(feature) for feature in feature_set_proto.features
            ],
            entities=[
                Entity.from_proto(entity) for entity in feature_set_proto.entities
            ],
            max_age=feature_set_proto.max_age,
            source=(
                None
                if feature_set_proto.source.type == 0
                else Source.from_proto(feature_set_proto.source)
            ),
        )
        feature_set._version = feature_set_proto.version
        feature_set._is_dirty = False
        return feature_set
Exemple #2
0
    def from_proto(cls, feature_set_proto: FeatureSetProto):
        """
        Creates a feature set from a protobuf representation of a feature set

        Args:
            feature_set_proto: A protobuf representation of a feature set

        Returns:
            Returns a FeatureSet object based on the feature set protobuf
        """

        feature_set = cls(
            name=feature_set_proto.spec.name,
            features=[
                Feature.from_proto(feature)
                for feature in feature_set_proto.spec.features
            ],
            entities=[
                Entity.from_proto(entity)
                for entity in feature_set_proto.spec.entities
            ],
            max_age=feature_set_proto.spec.max_age,
            source=(None if feature_set_proto.spec.source.type == 0 else
                    Source.from_proto(feature_set_proto.spec.source)),
            project=feature_set_proto.spec.project
            if len(feature_set_proto.spec.project) == 0 else
            feature_set_proto.spec.project,
        )
        feature_set._version = feature_set_proto.spec.version
        feature_set._status = feature_set_proto.meta.status
        feature_set._created_timestamp = feature_set_proto.meta.created_timestamp
        return feature_set
Exemple #3
0
 def from_proto(cls, feature_set_proto: FeatureSetSpecProto):
     feature_set = cls(
         name=feature_set_proto.name,
         features=[
             Feature.from_proto(feature) for feature in feature_set_proto.features
         ],
         entities=[
             Entity.from_proto(entity) for entity in feature_set_proto.entities
         ],
         max_age=feature_set_proto.max_age,
         source=(
             None
             if feature_set_proto.source.type == 0
             else Source.from_proto(feature_set_proto.source)
         ),
     )
     feature_set._version = feature_set_proto.version
     feature_set._is_dirty = False
     return feature_set
Exemple #4
0
 def source(self) -> Source:
     """
     Getter for the IngestJob's data source.
     """
     return Source.from_proto(self.proto.source)