(feature_view: BaseFeatureView)
| 146 | |
| 147 | |
| 148 | def _build_any_feature_view_proto(feature_view: BaseFeatureView): |
| 149 | from feast.labeling.label_view import LabelView |
| 150 | |
| 151 | if isinstance(feature_view, StreamFeatureView): |
| 152 | arg_name = "stream_feature_view" |
| 153 | feature_view_proto = feature_view.to_proto() |
| 154 | elif isinstance(feature_view, FeatureView): |
| 155 | arg_name = "feature_view" |
| 156 | feature_view_proto = feature_view.to_proto() |
| 157 | elif isinstance(feature_view, OnDemandFeatureView): |
| 158 | arg_name = "on_demand_feature_view" |
| 159 | feature_view_proto = feature_view.to_proto() |
| 160 | elif isinstance(feature_view, LabelView): |
| 161 | arg_name = "label_view" |
| 162 | feature_view_proto = feature_view.to_proto() |
| 163 | else: |
| 164 | raise ValueError(f"Unexpected feature view type: {type(feature_view)}") |
| 165 | |
| 166 | return RegistryServer_pb2.AnyFeatureView( |
| 167 | feature_view=feature_view_proto if arg_name == "feature_view" else None, |
| 168 | stream_feature_view=feature_view_proto |
| 169 | if arg_name == "stream_feature_view" |
| 170 | else None, |
| 171 | on_demand_feature_view=feature_view_proto |
| 172 | if arg_name == "on_demand_feature_view" |
| 173 | else None, |
| 174 | label_view=feature_view_proto if arg_name == "label_view" else None, |
| 175 | ) |
| 176 | |
| 177 | |
| 178 | class RegistryServer(RegistryServer_pb2_grpc.RegistryServerServicer): |
no test coverage detected