Skip to content

Commit 450da89

Browse files
authored
Compatibility fix for Databricks SDK 0.51 (#158)
## Changes [Databricks SDK 0.51](https://github.com/databricks/databricks-sdk-py/releases/tag/v0.51.0) included some breaking changes, including removal of an enumeration where a set of well-known string constants must now be used instead: > Breaking] Removed large, medium and small enum values for databricks.sdk.service.serving.ServedModelInputWorkloadSize. The change is not forward-compatible, so in addition to updating the fixture we need to bump the minimum-required SDK version because older versions expect the enum. ### Credits This was earlier discovered and reported by @cornzyblack as #156. ### Linked issues Resolves #156. Supersedes #157. This is also causing CI test failures on downstream projects including: - [databrickslabs/lsql](https://github.com/databrickslabs/lsql) - [databrickslabs/blueprint](https://github.com/databrickslabs/blueprint) ### Tests - existing unit tests
1 parent 241e9bf commit 450da89

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ classifiers = [
4848
]
4949

5050
dependencies = [
51-
"databricks-sdk>=0.40,<1.0",
51+
"databricks-sdk>=0.51.0,<1.0",
5252
"databricks-labs-lsql>=0.10",
5353
"pytest>=8.3",
5454
]

src/databricks/labs/pytester/fixtures/ml.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
EndpointPendingConfig,
1111
EndpointTag,
1212
ServedModelInput,
13-
ServedModelInputWorkloadSize,
1413
ServedModelOutput,
1514
ServingEndpointDetailed,
1615
)
@@ -159,7 +158,7 @@ def create(
159158
model_name=model_name,
160159
model_version=model_version,
161160
scale_to_zero_enabled=True,
162-
workload_size=ServedModelInputWorkloadSize.SMALL,
161+
workload_size="Small",
163162
)
164163
endpoint = ws.serving_endpoints.create(
165164
endpoint_name,
@@ -171,7 +170,7 @@ def create(
171170
model_name=model_name,
172171
model_version=model_version,
173172
scale_to_zero_enabled=True,
174-
workload_size=ServedModelInputWorkloadSize.SMALL.value,
173+
workload_size="Small",
175174
)
176175
endpoint = ServingEndpointDetailed(
177176
name=endpoint_name,

0 commit comments

Comments
 (0)