Skip to content

Commit 63bb245

Browse files
committed
Expose Dolt conformance for in-place public APIs
1 parent 2da412e commit 63bb245

1 file changed

Lines changed: 13 additions & 2 deletions

File tree

‎src/openstatspec/api.py‎

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -221,6 +221,7 @@ def apply_spss_in_place(
221221
*, database_url: Any, dataset_id: str, source_text: str,
222222
actor: str, expected_branch: str | None = None,
223223
expected_head: str | None = None,
224+
dolt_conformance_source: DoltConformanceSource | None = None,
224225
) -> Mapping[str, Any]:
225226
"""Apply supported SPSS-like syntax to the same SQL dataset/table."""
226227
return result(_apply_spss_in_place(
@@ -230,6 +231,7 @@ def apply_spss_in_place(
230231
actor=actor,
231232
expected_branch=expected_branch,
232233
expected_head=expected_head,
234+
dolt_conformance_source=dolt_conformance_source,
233235
))
234236

235237

@@ -238,6 +240,7 @@ def apply_transformation_plan_in_place(
238240
plan: TransformationPlan | Mapping[str, Any],
239241
actor: str, expected_branch: str | None = None,
240242
expected_head: str | None = None,
243+
dolt_conformance_source: DoltConformanceSource | None = None,
241244
) -> Mapping[str, Any]:
242245
"""Apply a canonical plan to the same logical dataset and physical table."""
243246
return result(_apply_transformation_plan_in_place(
@@ -247,12 +250,20 @@ def apply_transformation_plan_in_place(
247250
actor=actor,
248251
expected_branch=expected_branch,
249252
expected_head=expected_head,
253+
dolt_conformance_source=dolt_conformance_source,
250254
))
251255

252256

253-
def install_in_place_transformation_schema(*, database_url: Any) -> None:
257+
def install_in_place_transformation_schema(
258+
*,
259+
database_url: Any,
260+
dolt_conformance_source: DoltConformanceSource | None = None,
261+
) -> None:
254262
"""Install the compact apply-audit relation before the first apply."""
255-
_install_in_place_schema(database_url=str(database_url))
263+
_install_in_place_schema(
264+
database_url=str(database_url),
265+
dolt_conformance_source=dolt_conformance_source,
266+
)
256267

257268

258269
def validate_derived(*, database_url: Any, derived_dataset_id: str) -> Mapping[str, Any]:

0 commit comments

Comments
 (0)