fix: Remove proto-plus imports by tokoko · Pull Request #4044 · feast-dev/feast

Expand Up @@ -13,13 +13,20 @@ # limitations under the License. from abc import ABC, abstractmethod from datetime import datetime from typing import Dict, List, Optional, Type from typing import Dict, List, Optional, Type, Union
from google.protobuf.json_format import MessageToJson from proto import Message from google.protobuf.message import Message
from feast.feature_view_projection import FeatureViewProjection from feast.field import Field from feast.protos.feast.core.FeatureView_pb2 import FeatureView as FeatureViewProto from feast.protos.feast.core.OnDemandFeatureView_pb2 import ( OnDemandFeatureView as OnDemandFeatureViewProto, ) from feast.protos.feast.core.StreamFeatureView_pb2 import ( StreamFeatureView as StreamFeatureViewProto, )

class BaseFeatureView(ABC): Expand Down Expand Up @@ -89,7 +96,9 @@ def proto_class(self) -> Type[Message]: pass
@abstractmethod def to_proto(self) -> Message: def to_proto( self, ) -> Union[FeatureViewProto, OnDemandFeatureViewProto, StreamFeatureViewProto]: pass
@classmethod Expand Down