forked from feast-dev/feast
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSavedDataset_pb2.pyi
More file actions
217 lines (189 loc) · 10.4 KB
/
SavedDataset_pb2.pyi
File metadata and controls
217 lines (189 loc) · 10.4 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
"""
@generated by mypy-protobuf. Do not edit manually!
isort:skip_file
Copyright 2021 The Feast Authors
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
https://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
from collections import abc as _abc
from feast.core import DataSource_pb2 as _DataSource_pb2
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import timestamp_pb2 as _timestamp_pb2
from google.protobuf.internal import containers as _containers
import builtins as _builtins
import sys
import typing as _typing
if sys.version_info >= (3, 10):
from typing import TypeAlias as _TypeAlias
else:
from typing_extensions import TypeAlias as _TypeAlias
DESCRIPTOR: _descriptor.FileDescriptor
@_typing.final
class SavedDatasetSpec(_message.Message):
DESCRIPTOR: _descriptor.Descriptor
@_typing.final
class TagsEntry(_message.Message):
DESCRIPTOR: _descriptor.Descriptor
KEY_FIELD_NUMBER: _builtins.int
VALUE_FIELD_NUMBER: _builtins.int
key: _builtins.str
value: _builtins.str
def __init__(
self,
*,
key: _builtins.str = ...,
value: _builtins.str = ...,
) -> None: ...
_ClearFieldArgType: _TypeAlias = _typing.Literal["key", b"key", "value", b"value"] # noqa: Y015
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
NAME_FIELD_NUMBER: _builtins.int
PROJECT_FIELD_NUMBER: _builtins.int
FEATURES_FIELD_NUMBER: _builtins.int
JOIN_KEYS_FIELD_NUMBER: _builtins.int
FULL_FEATURE_NAMES_FIELD_NUMBER: _builtins.int
STORAGE_FIELD_NUMBER: _builtins.int
FEATURE_SERVICE_NAME_FIELD_NUMBER: _builtins.int
TAGS_FIELD_NUMBER: _builtins.int
name: _builtins.str
"""Name of the dataset. Must be unique since it's possible to overwrite dataset by name"""
project: _builtins.str
"""Name of Feast project that this Dataset belongs to."""
full_feature_names: _builtins.bool
"""Whether full feature names are used in stored data"""
feature_service_name: _builtins.str
"""Optional and only populated if generated from a feature service fetch"""
@_builtins.property
def features(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]:
"""list of feature references with format "<view name>:<feature name>" """
@_builtins.property
def join_keys(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]:
"""entity columns + request columns from all feature views used during retrieval"""
@_builtins.property
def storage(self) -> Global___SavedDatasetStorage: ...
@_builtins.property
def tags(self) -> _containers.ScalarMap[_builtins.str, _builtins.str]:
"""User defined metadata"""
def __init__(
self,
*,
name: _builtins.str = ...,
project: _builtins.str = ...,
features: _abc.Iterable[_builtins.str] | None = ...,
join_keys: _abc.Iterable[_builtins.str] | None = ...,
full_feature_names: _builtins.bool = ...,
storage: Global___SavedDatasetStorage | None = ...,
feature_service_name: _builtins.str = ...,
tags: _abc.Mapping[_builtins.str, _builtins.str] | None = ...,
) -> None: ...
_HasFieldArgType: _TypeAlias = _typing.Literal["storage", b"storage"] # noqa: Y015
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
_ClearFieldArgType: _TypeAlias = _typing.Literal["feature_service_name", b"feature_service_name", "features", b"features", "full_feature_names", b"full_feature_names", "join_keys", b"join_keys", "name", b"name", "project", b"project", "storage", b"storage", "tags", b"tags"] # noqa: Y015
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
Global___SavedDatasetSpec: _TypeAlias = SavedDatasetSpec # noqa: Y015
@_typing.final
class SavedDatasetStorage(_message.Message):
DESCRIPTOR: _descriptor.Descriptor
FILE_STORAGE_FIELD_NUMBER: _builtins.int
BIGQUERY_STORAGE_FIELD_NUMBER: _builtins.int
REDSHIFT_STORAGE_FIELD_NUMBER: _builtins.int
SNOWFLAKE_STORAGE_FIELD_NUMBER: _builtins.int
TRINO_STORAGE_FIELD_NUMBER: _builtins.int
SPARK_STORAGE_FIELD_NUMBER: _builtins.int
CUSTOM_STORAGE_FIELD_NUMBER: _builtins.int
ATHENA_STORAGE_FIELD_NUMBER: _builtins.int
@_builtins.property
def file_storage(self) -> _DataSource_pb2.DataSource.FileOptions: ...
@_builtins.property
def bigquery_storage(self) -> _DataSource_pb2.DataSource.BigQueryOptions: ...
@_builtins.property
def redshift_storage(self) -> _DataSource_pb2.DataSource.RedshiftOptions: ...
@_builtins.property
def snowflake_storage(self) -> _DataSource_pb2.DataSource.SnowflakeOptions: ...
@_builtins.property
def trino_storage(self) -> _DataSource_pb2.DataSource.TrinoOptions: ...
@_builtins.property
def spark_storage(self) -> _DataSource_pb2.DataSource.SparkOptions: ...
@_builtins.property
def custom_storage(self) -> _DataSource_pb2.DataSource.CustomSourceOptions: ...
@_builtins.property
def athena_storage(self) -> _DataSource_pb2.DataSource.AthenaOptions: ...
def __init__(
self,
*,
file_storage: _DataSource_pb2.DataSource.FileOptions | None = ...,
bigquery_storage: _DataSource_pb2.DataSource.BigQueryOptions | None = ...,
redshift_storage: _DataSource_pb2.DataSource.RedshiftOptions | None = ...,
snowflake_storage: _DataSource_pb2.DataSource.SnowflakeOptions | None = ...,
trino_storage: _DataSource_pb2.DataSource.TrinoOptions | None = ...,
spark_storage: _DataSource_pb2.DataSource.SparkOptions | None = ...,
custom_storage: _DataSource_pb2.DataSource.CustomSourceOptions | None = ...,
athena_storage: _DataSource_pb2.DataSource.AthenaOptions | None = ...,
) -> None: ...
_HasFieldArgType: _TypeAlias = _typing.Literal["athena_storage", b"athena_storage", "bigquery_storage", b"bigquery_storage", "custom_storage", b"custom_storage", "file_storage", b"file_storage", "kind", b"kind", "redshift_storage", b"redshift_storage", "snowflake_storage", b"snowflake_storage", "spark_storage", b"spark_storage", "trino_storage", b"trino_storage"] # noqa: Y015
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
_ClearFieldArgType: _TypeAlias = _typing.Literal["athena_storage", b"athena_storage", "bigquery_storage", b"bigquery_storage", "custom_storage", b"custom_storage", "file_storage", b"file_storage", "kind", b"kind", "redshift_storage", b"redshift_storage", "snowflake_storage", b"snowflake_storage", "spark_storage", b"spark_storage", "trino_storage", b"trino_storage"] # noqa: Y015
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
_WhichOneofReturnType_kind: _TypeAlias = _typing.Literal["file_storage", "bigquery_storage", "redshift_storage", "snowflake_storage", "trino_storage", "spark_storage", "custom_storage", "athena_storage"] # noqa: Y015
_WhichOneofArgType_kind: _TypeAlias = _typing.Literal["kind", b"kind"] # noqa: Y015
def WhichOneof(self, oneof_group: _WhichOneofArgType_kind) -> _WhichOneofReturnType_kind | None: ...
Global___SavedDatasetStorage: _TypeAlias = SavedDatasetStorage # noqa: Y015
@_typing.final
class SavedDatasetMeta(_message.Message):
DESCRIPTOR: _descriptor.Descriptor
CREATED_TIMESTAMP_FIELD_NUMBER: _builtins.int
LAST_UPDATED_TIMESTAMP_FIELD_NUMBER: _builtins.int
MIN_EVENT_TIMESTAMP_FIELD_NUMBER: _builtins.int
MAX_EVENT_TIMESTAMP_FIELD_NUMBER: _builtins.int
@_builtins.property
def created_timestamp(self) -> _timestamp_pb2.Timestamp:
"""Time when this saved dataset is created"""
@_builtins.property
def last_updated_timestamp(self) -> _timestamp_pb2.Timestamp:
"""Time when this saved dataset is last updated"""
@_builtins.property
def min_event_timestamp(self) -> _timestamp_pb2.Timestamp:
"""Min timestamp in the dataset (needed for retrieval)"""
@_builtins.property
def max_event_timestamp(self) -> _timestamp_pb2.Timestamp:
"""Max timestamp in the dataset (needed for retrieval)"""
def __init__(
self,
*,
created_timestamp: _timestamp_pb2.Timestamp | None = ...,
last_updated_timestamp: _timestamp_pb2.Timestamp | None = ...,
min_event_timestamp: _timestamp_pb2.Timestamp | None = ...,
max_event_timestamp: _timestamp_pb2.Timestamp | None = ...,
) -> None: ...
_HasFieldArgType: _TypeAlias = _typing.Literal["created_timestamp", b"created_timestamp", "last_updated_timestamp", b"last_updated_timestamp", "max_event_timestamp", b"max_event_timestamp", "min_event_timestamp", b"min_event_timestamp"] # noqa: Y015
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
_ClearFieldArgType: _TypeAlias = _typing.Literal["created_timestamp", b"created_timestamp", "last_updated_timestamp", b"last_updated_timestamp", "max_event_timestamp", b"max_event_timestamp", "min_event_timestamp", b"min_event_timestamp"] # noqa: Y015
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
Global___SavedDatasetMeta: _TypeAlias = SavedDatasetMeta # noqa: Y015
@_typing.final
class SavedDataset(_message.Message):
DESCRIPTOR: _descriptor.Descriptor
SPEC_FIELD_NUMBER: _builtins.int
META_FIELD_NUMBER: _builtins.int
@_builtins.property
def spec(self) -> Global___SavedDatasetSpec: ...
@_builtins.property
def meta(self) -> Global___SavedDatasetMeta: ...
def __init__(
self,
*,
spec: Global___SavedDatasetSpec | None = ...,
meta: Global___SavedDatasetMeta | None = ...,
) -> None: ...
_HasFieldArgType: _TypeAlias = _typing.Literal["meta", b"meta", "spec", b"spec"] # noqa: Y015
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
_ClearFieldArgType: _TypeAlias = _typing.Literal["meta", b"meta", "spec", b"spec"] # noqa: Y015
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
Global___SavedDataset: _TypeAlias = SavedDataset # noqa: Y015