|
3 | 3 | from __future__ import annotations |
4 | 4 |
|
5 | 5 | from typing import Iterable, Optional |
| 6 | +from typing_extensions import Literal |
6 | 7 |
|
7 | 8 | import httpx |
8 | 9 |
|
@@ -51,9 +52,51 @@ def create( |
51 | 52 | *, |
52 | 53 | description: Optional[object], |
53 | 54 | name: str, |
54 | | - subtype: str, |
| 55 | + subtype: Literal[ |
| 56 | + "anomalousColumnCount", |
| 57 | + "characterLength", |
| 58 | + "classImbalanceRatio", |
| 59 | + "expectColumnAToBeInColumnB", |
| 60 | + "columnAverage", |
| 61 | + "columnDrift", |
| 62 | + "columnStatistic", |
| 63 | + "columnValuesMatch", |
| 64 | + "conflictingLabelRowCount", |
| 65 | + "containsPii", |
| 66 | + "containsValidUrl", |
| 67 | + "correlatedFeatureCount", |
| 68 | + "customMetricThreshold", |
| 69 | + "duplicateRowCount", |
| 70 | + "emptyFeature", |
| 71 | + "emptyFeatureCount", |
| 72 | + "driftedFeatureCount", |
| 73 | + "featureMissingValues", |
| 74 | + "featureValueValidation", |
| 75 | + "greatExpectations", |
| 76 | + "groupByColumnStatsCheck", |
| 77 | + "illFormedRowCount", |
| 78 | + "isCode", |
| 79 | + "isJson", |
| 80 | + "llmRubricThresholdV2", |
| 81 | + "labelDrift", |
| 82 | + "metricThreshold", |
| 83 | + "newCategoryCount", |
| 84 | + "newLabelCount", |
| 85 | + "nullRowCount", |
| 86 | + "rowCount", |
| 87 | + "ppScoreValueValidation", |
| 88 | + "quasiConstantFeature", |
| 89 | + "quasiConstantFeatureCount", |
| 90 | + "sqlQuery", |
| 91 | + "dtypeValidation", |
| 92 | + "sentenceLength", |
| 93 | + "sizeRatio", |
| 94 | + "specialCharactersRatio", |
| 95 | + "stringValidation", |
| 96 | + "trainValLeakageRowCount", |
| 97 | + ], |
55 | 98 | thresholds: Iterable[test_create_params.Threshold], |
56 | | - type: str, |
| 99 | + type: Literal["integrity", "consistency", "performance"], |
57 | 100 | archived: bool | NotGiven = NOT_GIVEN, |
58 | 101 | delay_window: Optional[float] | NotGiven = NOT_GIVEN, |
59 | 102 | evaluation_window: Optional[float] | NotGiven = NOT_GIVEN, |
@@ -161,9 +204,51 @@ async def create( |
161 | 204 | *, |
162 | 205 | description: Optional[object], |
163 | 206 | name: str, |
164 | | - subtype: str, |
| 207 | + subtype: Literal[ |
| 208 | + "anomalousColumnCount", |
| 209 | + "characterLength", |
| 210 | + "classImbalanceRatio", |
| 211 | + "expectColumnAToBeInColumnB", |
| 212 | + "columnAverage", |
| 213 | + "columnDrift", |
| 214 | + "columnStatistic", |
| 215 | + "columnValuesMatch", |
| 216 | + "conflictingLabelRowCount", |
| 217 | + "containsPii", |
| 218 | + "containsValidUrl", |
| 219 | + "correlatedFeatureCount", |
| 220 | + "customMetricThreshold", |
| 221 | + "duplicateRowCount", |
| 222 | + "emptyFeature", |
| 223 | + "emptyFeatureCount", |
| 224 | + "driftedFeatureCount", |
| 225 | + "featureMissingValues", |
| 226 | + "featureValueValidation", |
| 227 | + "greatExpectations", |
| 228 | + "groupByColumnStatsCheck", |
| 229 | + "illFormedRowCount", |
| 230 | + "isCode", |
| 231 | + "isJson", |
| 232 | + "llmRubricThresholdV2", |
| 233 | + "labelDrift", |
| 234 | + "metricThreshold", |
| 235 | + "newCategoryCount", |
| 236 | + "newLabelCount", |
| 237 | + "nullRowCount", |
| 238 | + "rowCount", |
| 239 | + "ppScoreValueValidation", |
| 240 | + "quasiConstantFeature", |
| 241 | + "quasiConstantFeatureCount", |
| 242 | + "sqlQuery", |
| 243 | + "dtypeValidation", |
| 244 | + "sentenceLength", |
| 245 | + "sizeRatio", |
| 246 | + "specialCharactersRatio", |
| 247 | + "stringValidation", |
| 248 | + "trainValLeakageRowCount", |
| 249 | + ], |
165 | 250 | thresholds: Iterable[test_create_params.Threshold], |
166 | | - type: str, |
| 251 | + type: Literal["integrity", "consistency", "performance"], |
167 | 252 | archived: bool | NotGiven = NOT_GIVEN, |
168 | 253 | delay_window: Optional[float] | NotGiven = NOT_GIVEN, |
169 | 254 | evaluation_window: Optional[float] | NotGiven = NOT_GIVEN, |
|
0 commit comments