Repository navigation
Conversation
Deploys the tabular model to a one-GPU Model Serving endpoint that scales to zero, then asks the AdventureWorks products table a color classification, a list price regression, and the color question in PQL. Every Unity Catalog name part is quoted with backticks. Signed-off-by: Adi Wadaskar <awadaskar@nvidia.com>
… too large A tabular request has no entities and no num_context_samples, so the 16 MB refusal now asks for fewer context rows or fewer rows to score. The limits table and the quickstart name the tabular model too, and the notebook test expects the new quickstart. Signed-off-by: Adi Wadaskar <awadaskar@nvidia.com>
The products table carries each product's standard cost, which tracks its list price closely, so the regression error is small. Say so, so nobody reads it as the model's typical error on a hard column. Signed-off-by: Adi Wadaskar <awadaskar@nvidia.com>
The notebook was run end to end on serverless environment 5 against a GPU_MEDIUM endpoint serving Kumo Tabular. The catalog, schema and endpoint names in the outputs are the notebook's defaults. The pip output is left out, because it only repeats the protobuf notice the text already explains. Signed-off-by: Adi Wadaskar <awadaskar@nvidia.com>
This branch has not been deployed
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Adds a Databricks quickstart for Kumo Tabular, next to the relational one. It deploys the model to a one-GPU Model Serving endpoint that scales to zero, loads the AdventureWorks products table, and asks three questions of it: which color a product has, what its list price is, and the color question again as a predictive query. Each answer is checked against the recorded value and compared with a simple rule. The notebook is saved with the outputs of a real run.
The tabular model is the one registered in Unity Catalog as
kumo_tabular. Its 16 MB limit message used to talk about entities, which a one-table request does not have, so that wording is made model-neutral too.Changes
examples/databricks/kumo_tabular_quickstart.ipynb: the notebook, with every Unity Catalog name part quoted so hyphenated catalogs work.databricks_serving.py), with matching text indocs/reference/limits.mdand the quickstart, a changelog line, and tests.%pipcell's output is left out; it only repeats the protobuf notice the text explains.Testing
Run on Databricks (EDSP workspace), 2026-10-22 UTC (2026-10-21 local): one-time job run 107355698597869, task run 923635785761350, serverless environment version 5 (client.5.12), Python 3.12.3, pandas 2.2.3, numpy 2.1.3, databricks-sdk 0.156.0, client, api, pql and connectors built from this branch at f5ae890 (1.0.0). The notebook created the endpoint (GPU_MEDIUM, Small, scale to zero) for Kumo Tabular version 1 and ran from top to bottom in 2.7 minutes, because the endpoint image was already cached from an earlier deployment of the same model. A first deployment took about 18 minutes in an earlier smoke test. The only differences from this notebook were the install cell (wheels from a Volume, as the package is not on PyPI yet) and the constants (that workspace's model, catalog and schema, and a throwaway endpoint name). The cells are otherwise unchanged. After that run only Example 2's text and the saved outputs changed. The endpoint was deleted afterwards.
What the run showed:
probability_<color>, one per color, as the text describes.Also run for this PR:
ruff check,ruff format --check,scripts/check_repo_hygiene.py all, and the five-package pytest suite (2179 passed, 330 skipped), which includes the Databricks notebook tests. The engine suite is not affected.Follow-ups
🤖 Generated with Claude Code