Skip to content

Commit 7566335

Browse files
author
Maciek Pytel
committed
Re-genarate client with swagger including metric previews
1 parent ee71941 commit 7566335

File tree

7 files changed

+73
-0
lines changed

7 files changed

+73
-0
lines changed

src/neptune_retrieval_api/api/default/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,3 +12,4 @@
1212
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1313
# See the License for the specific language governing permissions and
1414
# limitations under the License.
15+

src/neptune_retrieval_api/api/default/get_float_series_values.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ def _get_kwargs(
4242
*,
4343
attribute: str,
4444
experiment_id: Union[Unset, str] = UNSET,
45+
include_preview: Union[Unset, bool] = False,
4546
limit: Union[Unset, int] = UNSET,
4647
lineage: Union[Unset, GetFloatSeriesValuesLineage] = UNSET,
4748
skip_to_step: Union[Unset, float] = UNSET,
@@ -52,6 +53,8 @@ def _get_kwargs(
5253

5354
params["experimentId"] = experiment_id
5455

56+
params["includePreview"] = include_preview
57+
5558
params["limit"] = limit
5659

5760
json_lineage: Union[Unset, str] = UNSET
@@ -126,6 +129,7 @@ def sync_detailed(
126129
client: Union[AuthenticatedClient, Client],
127130
attribute: str,
128131
experiment_id: Union[Unset, str] = UNSET,
132+
include_preview: Union[Unset, bool] = False,
129133
limit: Union[Unset, int] = UNSET,
130134
lineage: Union[Unset, GetFloatSeriesValuesLineage] = UNSET,
131135
skip_to_step: Union[Unset, float] = UNSET,
@@ -135,6 +139,7 @@ def sync_detailed(
135139
Args:
136140
attribute (str):
137141
experiment_id (Union[Unset, str]):
142+
include_preview (Union[Unset, bool]): Default: False.
138143
limit (Union[Unset, int]):
139144
lineage (Union[Unset, GetFloatSeriesValuesLineage]):
140145
skip_to_step (Union[Unset, float]):
@@ -150,6 +155,7 @@ def sync_detailed(
150155
kwargs = _get_kwargs(
151156
attribute=attribute,
152157
experiment_id=experiment_id,
158+
include_preview=include_preview,
153159
limit=limit,
154160
lineage=lineage,
155161
skip_to_step=skip_to_step,
@@ -167,6 +173,7 @@ def sync(
167173
client: Union[AuthenticatedClient, Client],
168174
attribute: str,
169175
experiment_id: Union[Unset, str] = UNSET,
176+
include_preview: Union[Unset, bool] = False,
170177
limit: Union[Unset, int] = UNSET,
171178
lineage: Union[Unset, GetFloatSeriesValuesLineage] = UNSET,
172179
skip_to_step: Union[Unset, float] = UNSET,
@@ -176,6 +183,7 @@ def sync(
176183
Args:
177184
attribute (str):
178185
experiment_id (Union[Unset, str]):
186+
include_preview (Union[Unset, bool]): Default: False.
179187
limit (Union[Unset, int]):
180188
lineage (Union[Unset, GetFloatSeriesValuesLineage]):
181189
skip_to_step (Union[Unset, float]):
@@ -192,6 +200,7 @@ def sync(
192200
client=client,
193201
attribute=attribute,
194202
experiment_id=experiment_id,
203+
include_preview=include_preview,
195204
limit=limit,
196205
lineage=lineage,
197206
skip_to_step=skip_to_step,
@@ -203,6 +212,7 @@ async def asyncio_detailed(
203212
client: Union[AuthenticatedClient, Client],
204213
attribute: str,
205214
experiment_id: Union[Unset, str] = UNSET,
215+
include_preview: Union[Unset, bool] = False,
206216
limit: Union[Unset, int] = UNSET,
207217
lineage: Union[Unset, GetFloatSeriesValuesLineage] = UNSET,
208218
skip_to_step: Union[Unset, float] = UNSET,
@@ -212,6 +222,7 @@ async def asyncio_detailed(
212222
Args:
213223
attribute (str):
214224
experiment_id (Union[Unset, str]):
225+
include_preview (Union[Unset, bool]): Default: False.
215226
limit (Union[Unset, int]):
216227
lineage (Union[Unset, GetFloatSeriesValuesLineage]):
217228
skip_to_step (Union[Unset, float]):
@@ -227,6 +238,7 @@ async def asyncio_detailed(
227238
kwargs = _get_kwargs(
228239
attribute=attribute,
229240
experiment_id=experiment_id,
241+
include_preview=include_preview,
230242
limit=limit,
231243
lineage=lineage,
232244
skip_to_step=skip_to_step,
@@ -242,6 +254,7 @@ async def asyncio(
242254
client: Union[AuthenticatedClient, Client],
243255
attribute: str,
244256
experiment_id: Union[Unset, str] = UNSET,
257+
include_preview: Union[Unset, bool] = False,
245258
limit: Union[Unset, int] = UNSET,
246259
lineage: Union[Unset, GetFloatSeriesValuesLineage] = UNSET,
247260
skip_to_step: Union[Unset, float] = UNSET,
@@ -251,6 +264,7 @@ async def asyncio(
251264
Args:
252265
attribute (str):
253266
experiment_id (Union[Unset, str]):
267+
include_preview (Union[Unset, bool]): Default: False.
254268
limit (Union[Unset, int]):
255269
lineage (Union[Unset, GetFloatSeriesValuesLineage]):
256270
skip_to_step (Union[Unset, float]):
@@ -268,6 +282,7 @@ async def asyncio(
268282
client=client,
269283
attribute=attribute,
270284
experiment_id=experiment_id,
285+
include_preview=include_preview,
271286
limit=limit,
272287
lineage=lineage,
273288
skip_to_step=skip_to_step,

src/neptune_retrieval_api/api/default/get_float_series_values_proto.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ def _get_kwargs(
4343
*,
4444
attribute: str,
4545
experiment_id: Union[Unset, str] = UNSET,
46+
include_preview: Union[Unset, bool] = False,
4647
limit: Union[Unset, int] = 10000,
4748
lineage: Union[Unset, GetFloatSeriesValuesProtoLineage] = UNSET,
4849
skip_to_step: Union[Unset, float] = UNSET,
@@ -53,6 +54,8 @@ def _get_kwargs(
5354

5455
params["experimentId"] = experiment_id
5556

57+
params["includePreview"] = include_preview
58+
5659
params["limit"] = limit
5760

5861
json_lineage: Union[Unset, str] = UNSET
@@ -127,6 +130,7 @@ def sync_detailed(
127130
client: Union[AuthenticatedClient, Client],
128131
attribute: str,
129132
experiment_id: Union[Unset, str] = UNSET,
133+
include_preview: Union[Unset, bool] = False,
130134
limit: Union[Unset, int] = 10000,
131135
lineage: Union[Unset, GetFloatSeriesValuesProtoLineage] = UNSET,
132136
skip_to_step: Union[Unset, float] = UNSET,
@@ -136,6 +140,7 @@ def sync_detailed(
136140
Args:
137141
attribute (str):
138142
experiment_id (Union[Unset, str]):
143+
include_preview (Union[Unset, bool]): Default: False.
139144
limit (Union[Unset, int]): Default: 10000.
140145
lineage (Union[Unset, GetFloatSeriesValuesProtoLineage]):
141146
skip_to_step (Union[Unset, float]):
@@ -151,6 +156,7 @@ def sync_detailed(
151156
kwargs = _get_kwargs(
152157
attribute=attribute,
153158
experiment_id=experiment_id,
159+
include_preview=include_preview,
154160
limit=limit,
155161
lineage=lineage,
156162
skip_to_step=skip_to_step,
@@ -168,6 +174,7 @@ def sync(
168174
client: Union[AuthenticatedClient, Client],
169175
attribute: str,
170176
experiment_id: Union[Unset, str] = UNSET,
177+
include_preview: Union[Unset, bool] = False,
171178
limit: Union[Unset, int] = 10000,
172179
lineage: Union[Unset, GetFloatSeriesValuesProtoLineage] = UNSET,
173180
skip_to_step: Union[Unset, float] = UNSET,
@@ -177,6 +184,7 @@ def sync(
177184
Args:
178185
attribute (str):
179186
experiment_id (Union[Unset, str]):
187+
include_preview (Union[Unset, bool]): Default: False.
180188
limit (Union[Unset, int]): Default: 10000.
181189
lineage (Union[Unset, GetFloatSeriesValuesProtoLineage]):
182190
skip_to_step (Union[Unset, float]):
@@ -193,6 +201,7 @@ def sync(
193201
client=client,
194202
attribute=attribute,
195203
experiment_id=experiment_id,
204+
include_preview=include_preview,
196205
limit=limit,
197206
lineage=lineage,
198207
skip_to_step=skip_to_step,
@@ -204,6 +213,7 @@ async def asyncio_detailed(
204213
client: Union[AuthenticatedClient, Client],
205214
attribute: str,
206215
experiment_id: Union[Unset, str] = UNSET,
216+
include_preview: Union[Unset, bool] = False,
207217
limit: Union[Unset, int] = 10000,
208218
lineage: Union[Unset, GetFloatSeriesValuesProtoLineage] = UNSET,
209219
skip_to_step: Union[Unset, float] = UNSET,
@@ -213,6 +223,7 @@ async def asyncio_detailed(
213223
Args:
214224
attribute (str):
215225
experiment_id (Union[Unset, str]):
226+
include_preview (Union[Unset, bool]): Default: False.
216227
limit (Union[Unset, int]): Default: 10000.
217228
lineage (Union[Unset, GetFloatSeriesValuesProtoLineage]):
218229
skip_to_step (Union[Unset, float]):
@@ -228,6 +239,7 @@ async def asyncio_detailed(
228239
kwargs = _get_kwargs(
229240
attribute=attribute,
230241
experiment_id=experiment_id,
242+
include_preview=include_preview,
231243
limit=limit,
232244
lineage=lineage,
233245
skip_to_step=skip_to_step,
@@ -243,6 +255,7 @@ async def asyncio(
243255
client: Union[AuthenticatedClient, Client],
244256
attribute: str,
245257
experiment_id: Union[Unset, str] = UNSET,
258+
include_preview: Union[Unset, bool] = False,
246259
limit: Union[Unset, int] = 10000,
247260
lineage: Union[Unset, GetFloatSeriesValuesProtoLineage] = UNSET,
248261
skip_to_step: Union[Unset, float] = UNSET,
@@ -252,6 +265,7 @@ async def asyncio(
252265
Args:
253266
attribute (str):
254267
experiment_id (Union[Unset, str]):
268+
include_preview (Union[Unset, bool]): Default: False.
255269
limit (Union[Unset, int]): Default: 10000.
256270
lineage (Union[Unset, GetFloatSeriesValuesProtoLineage]):
257271
skip_to_step (Union[Unset, float]):
@@ -269,6 +283,7 @@ async def asyncio(
269283
client=client,
270284
attribute=attribute,
271285
experiment_id=experiment_id,
286+
include_preview=include_preview,
272287
limit=limit,
273288
lineage=lineage,
274289
skip_to_step=skip_to_step,

src/neptune_retrieval_api/models/custom_metric.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,13 +48,15 @@ class CustomMetric:
4848
custom_id (str):
4949
holder (AttributesHolderIdentifier):
5050
custom_y_formula (Union[Unset, str]):
51+
include_preview (Union[Unset, bool]):
5152
lineage (Union[Unset, CustomMetricLineage]):
5253
"""
5354

5455
attributes: List[str]
5556
custom_id: str
5657
holder: "AttributesHolderIdentifier"
5758
custom_y_formula: Union[Unset, str] = UNSET
59+
include_preview: Union[Unset, bool] = UNSET
5860
lineage: Union[Unset, CustomMetricLineage] = UNSET
5961
additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
6062

@@ -67,6 +69,8 @@ def to_dict(self) -> Dict[str, Any]:
6769

6870
custom_y_formula = self.custom_y_formula
6971

72+
include_preview = self.include_preview
73+
7074
lineage: Union[Unset, str] = UNSET
7175
if not isinstance(self.lineage, Unset):
7276
lineage = self.lineage.value
@@ -82,6 +86,8 @@ def to_dict(self) -> Dict[str, Any]:
8286
)
8387
if custom_y_formula is not UNSET:
8488
field_dict["customYFormula"] = custom_y_formula
89+
if include_preview is not UNSET:
90+
field_dict["includePreview"] = include_preview
8591
if lineage is not UNSET:
8692
field_dict["lineage"] = lineage
8793

@@ -100,6 +106,8 @@ def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
100106

101107
custom_y_formula = d.pop("customYFormula", UNSET)
102108

109+
include_preview = d.pop("includePreview", UNSET)
110+
103111
_lineage = d.pop("lineage", UNSET)
104112
lineage: Union[Unset, CustomMetricLineage]
105113
if isinstance(_lineage, Unset):
@@ -112,6 +120,7 @@ def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
112120
custom_id=custom_id,
113121
holder=holder,
114122
custom_y_formula=custom_y_formula,
123+
include_preview=include_preview,
115124
lineage=lineage,
116125
)
117126

src/neptune_retrieval_api/models/float_point_value_dto.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,17 +31,25 @@
3131
class FloatPointValueDTO:
3232
"""
3333
Attributes:
34+
completion_ratio (float):
35+
is_preview (bool):
3436
step (float):
3537
timestamp_millis (int):
3638
value (float):
3739
"""
3840

41+
completion_ratio: float
42+
is_preview: bool
3943
step: float
4044
timestamp_millis: int
4145
value: float
4246
additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
4347

4448
def to_dict(self) -> Dict[str, Any]:
49+
completion_ratio = self.completion_ratio
50+
51+
is_preview = self.is_preview
52+
4553
step = self.step
4654

4755
timestamp_millis = self.timestamp_millis
@@ -52,6 +60,8 @@ def to_dict(self) -> Dict[str, Any]:
5260
field_dict.update(self.additional_properties)
5361
field_dict.update(
5462
{
63+
"completionRatio": completion_ratio,
64+
"isPreview": is_preview,
5565
"step": step,
5666
"timestampMillis": timestamp_millis,
5767
"value": value,
@@ -63,13 +73,19 @@ def to_dict(self) -> Dict[str, Any]:
6373
@classmethod
6474
def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
6575
d = src_dict.copy()
76+
completion_ratio = d.pop("completionRatio")
77+
78+
is_preview = d.pop("isPreview")
79+
6680
step = d.pop("step")
6781

6882
timestamp_millis = d.pop("timestampMillis")
6983

7084
value = d.pop("value")
7185

7286
float_point_value_dto = cls(
87+
completion_ratio=completion_ratio,
88+
is_preview=is_preview,
7389
step=step,
7490
timestamp_millis=timestamp_millis,
7591
value=value,

src/neptune_retrieval_api/models/float_series_attribute_dto.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ class FloatSeriesAttributeDTO:
4646
attribute_name (str):
4747
attribute_type (AttributeTypeDTO):
4848
config (FloatSeriesAttributeConfigDTO):
49+
has_preview (bool):
4950
average (Union[Unset, float]):
5051
last (Union[Unset, float]):
5152
last_step (Union[Unset, float]):
@@ -57,6 +58,7 @@ class FloatSeriesAttributeDTO:
5758
attribute_name: str
5859
attribute_type: AttributeTypeDTO
5960
config: "FloatSeriesAttributeConfigDTO"
61+
has_preview: bool
6062
average: Union[Unset, float] = UNSET
6163
last: Union[Unset, float] = UNSET
6264
last_step: Union[Unset, float] = UNSET
@@ -72,6 +74,8 @@ def to_dict(self) -> Dict[str, Any]:
7274

7375
config = self.config.to_dict()
7476

77+
has_preview = self.has_preview
78+
7579
average = self.average
7680

7781
last = self.last
@@ -91,6 +95,7 @@ def to_dict(self) -> Dict[str, Any]:
9195
"attributeName": attribute_name,
9296
"attributeType": attribute_type,
9397
"config": config,
98+
"hasPreview": has_preview,
9499
}
95100
)
96101
if average is not UNSET:
@@ -119,6 +124,8 @@ def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
119124

120125
config = FloatSeriesAttributeConfigDTO.from_dict(d.pop("config"))
121126

127+
has_preview = d.pop("hasPreview")
128+
122129
average = d.pop("average", UNSET)
123130

124131
last = d.pop("last", UNSET)
@@ -135,6 +142,7 @@ def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
135142
attribute_name=attribute_name,
136143
attribute_type=attribute_type,
137144
config=config,
145+
has_preview=has_preview,
138146
average=average,
139147
last=last,
140148
last_step=last_step,

0 commit comments

Comments
 (0)