structured-data-models-client 1.0.0__py3-none-any.whl

sdm_client/__init__.py

@@ -0,0 +1,70 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES.
+# All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+r"""One Python client for NVIDIA's structured-data foundation models.
+
+Structured-data foundation models predict in context: you give the model
+labelled rows and the rows you want scored, and it answers in one pass. There
+is no training loop and no per-dataset artifact.
+
+>>> from sdm_client import Request, SDMClient, KumoTabular
+>>> request = Request(context=labelled, predict=to_score,
+...                   target='churn', task='classification')
+>>> with SDMClient(url='http://localhost:8000') as client:
+...     print(client.predict(request, model=KumoTabular()).to_pandas())
+
+The model is an argument, not a method name. A request is gathered for one
+model and answered by that one: `materialize(model=...)` says which, and
+defaults to what the schema's shape implies.
+"""
+
+from sdm_client._version import __version__
+from sdm_client.base import ModelCapabilities
+from sdm_client.client import FittedModel, SDMClient
+from sdm_client.core.warnings import SDMWarning
+from sdm_client.errors import (
+    MissingExtraError,
+    NimRequestError,
+    StructuredDataError,
+    UnknownModelError,
+    UnsupportedRequestError,
+)
+from sdm_client.models import KumoRelational, KumoTabular, Model
+from sdm_client.query import PredictiveQuery
+from sdm_client.request import Request, Split, Target, Task
+from sdm_client.result import PredictionResult
+from sdm_client.schema import (
+    Column,
+    Relationship,
+    Schema,
+    SourceTable,
+    Table,
+)
+
+__all__ = [
+    'SDMWarning',
+    'Column',
+    'FittedModel',
+    'KumoRelational',
+    'KumoTabular',
+    'MissingExtraError',
+    'Model',
+    'ModelCapabilities',
+    'NimRequestError',
+    'PredictionResult',
+    'PredictiveQuery',
+    'Relationship',
+    'Request',
+    'Schema',
+    'Split',
+    'SDMClient',
+    'SourceTable',
+    'StructuredDataError',
+    'Table',
+    'Target',
+    'Task',
+    'UnknownModelError',
+    'UnsupportedRequestError',
+    '__version__',
+]

sdm_client/_version.py

@@ -0,0 +1,5 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES.
+# All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+__version__ = '1.0.0'

sdm_client/adapters/__init__.py

@@ -0,0 +1,9 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES.
+# All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+"""Per-model adapters. One module per model."""
+
+from sdm_client.adapters.tabular import TabularAdapter
+
+__all__ = ['TabularAdapter']