kumoai 2.13.0.dev202511131731__cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl

kumoai/utils/forecasting.py

@@ -0,0 +1,209 @@
+from typing import Dict, List
+
+import pandas as pd
+import plotly.graph_objects as go
+from plotly.subplots import make_subplots
+
+
+class ForecastVisualizer:
+    r"""A tool for quickly visualizing forecast results from the holdout
+    dataframe of a forecasting job.
+
+    .. code-block:: python
+
+        import kumoai
+
+        # Retrieve job results from a training training job. Note
+        # that the job ID passed here must be in a completed state:
+        job_result = kumoai.TrainingJob("trainingjob-...").result()
+
+        # Read the holdout table as a Pandas DataFrame:
+        holdout_df = job_result.holdout_df()
+
+        # Pass holdout table to ForecastVisualizer and visualize results
+        holdout_forecast = kumoai.utils.ForecastVisualizer(holdout_df)
+        holdout_forecast.visualize()
+    """
+    def __init__(self, holdout_df: pd.DataFrame) -> None:
+        # Sort the holdout dataframe and extract unique entities:
+        self.forecast = holdout_df.sort_values(['ENTITY', 'TIMESTAMP'])
+        self.entities = holdout_df['ENTITY'].unique().tolist()
+
+        self.fig = self._initialize_subplot()
+        self.buttons: List[Dict] = []
+        self.plot_config = {
+            'target': {
+                'color': 'blue',
+                'name': 'TARGET'
+            },
+            'prediction': {
+                'color': 'red',
+                'name': 'TARGET_PRED'
+            },
+            'residuals': {
+                'color': 'green',
+                'name': 'Residuals'
+            },
+            'residuals_time': {
+                'color': 'orange',
+                'name': 'Residuals Over Time'
+            }
+        }
+
+    @staticmethod
+    def _initialize_subplot() -> go.Figure:
+        r"""Initializes the subplot structure with three rows:
+        Row 1: Line plot of actual forecast vs predicted
+        Row 2: Line plot of residuals overtime
+        Row 3: Histogram distribution of residuals
+        """
+        return make_subplots(
+            rows=3,
+            cols=1,
+            specs=[[{
+                "type": "scatter"
+            }], [{
+                "type": "scatter"
+            }], [{
+                "type": "xy"
+            }]],
+            row_heights=[0.5, 0.25, 0.25],
+            subplot_titles=('Forecast vs Actual', 'Residuals Over Time',
+                            'Residuals Distribution'),
+        )
+
+    def _create_time_series_trace(
+        self,
+        data: pd.Series,
+        entity: str,
+        trace_type: str,
+        visibility: bool,
+    ) -> go.Scatter:
+        r"""Create a time series trace for either target or prediction"""
+        config = self.plot_config[trace_type]
+        return go.Scatter(
+            x=data["TIMESTAMP"],
+            y=data[config['name']],
+            name=f"{entity} - {config['name']}",
+            mode="lines",
+            line=dict(color=config['color']),
+            visible=visibility,
+            opacity=0.75,
+        )
+
+    def _create_residuals_time_trace(
+        self,
+        data: pd.Series,
+        entity: str,
+        visibility: bool,
+    ) -> go.Scatter:
+        r"""Create a time series trace for residuals over time"""
+        residuals = data["TARGET"] - data["TARGET_PRED"]
+        return go.Scatter(
+            x=data["TIMESTAMP"],
+            y=residuals,
+            name=f"{entity} - Residuals Over Time",
+            mode="lines+markers",
+            line=dict(color=self.plot_config['residuals_time']['color']),
+            visible=visibility,
+            opacity=0.75,
+        )
+
+    def _create_residuals_hist_trace(
+        self,
+        data: pd.Series,
+        entity: str,
+        visibility: bool,
+    ) -> go.Histogram:
+        r"""Create a histogram trace for residuals distribution."""
+        residuals = data["TARGET"] - data["TARGET_PRED"]
+        return go.Histogram(
+            x=residuals,
+            name=f"{entity} - Residuals Distribution",
+            marker=dict(color=self.plot_config['residuals']['color']),
+            visible=visibility,
+            opacity=0.75,
+            nbinsx=30,
+        )
+
+    def _create_button(self, index: int, entity: str) -> None:
+        r"""Create visibility toggle button for an entity."""
+        # target, prediction, residuals time, and residuals hist:
+        num_traces_per_entity = 4
+        total_traces = len(self.entities) * num_traces_per_entity
+
+        button = dict(label=entity, method="update", args=[{
+            "visible": [False] * total_traces
+        }])
+
+        # Set visibility for the entity's traces:
+        base_index = index * num_traces_per_entity
+        for i in range(num_traces_per_entity):
+            button["args"][0]["visible"][base_index + i] = True  # type: ignore
+
+        self.buttons.append(button)
+
+    def _create_traces(self) -> None:
+        """Create all traces for the visualization."""
+        for i, entity in enumerate(self.entities):
+            entity_data = self.forecast.loc[self.forecast.ENTITY == entity]
+
+            # First entity's traces are visible by default:
+            visibility = (i == 0)
+
+            # Create traces
+            trace_target = self._create_time_series_trace(
+                entity_data, entity, 'target', visibility)
+            trace_pred = self._create_time_series_trace(
+                entity_data, entity, 'prediction', visibility)
+            trace_residuals_time = self._create_residuals_time_trace(
+                entity_data, entity, visibility)
+            trace_residuals_hist = self._create_residuals_hist_trace(
+                entity_data, entity, visibility)
+
+            # Add traces to appropriate subplots
+            self.fig.add_trace(trace_target, row=1, col=1)
+            self.fig.add_trace(trace_pred, row=1, col=1)
+            self.fig.add_trace(trace_residuals_time, row=2, col=1)
+            self.fig.add_trace(trace_residuals_hist, row=3, col=1)
+
+            self._create_button(i, entity)
+
+    def _update_layout(self) -> None:
+        r"""Update the figure layout with all necessary configurations."""
+        self.fig.update_layout(
+            updatemenus=[
+                dict(active=0, buttons=self.buttons, direction="down", pad={
+                    "r": 10,
+                    "t": 10
+                }, showactive=True, x=1, xanchor="left", y=1.07, yanchor="top")
+            ],
+            title="Forecast Results by Department",
+            height=1000,  # Increased height to accommodate third plot
+            width=1300,
+            showlegend=True,
+            hovermode='x unified')
+
+        # Update axis labels and add zero reference line for residuals
+        self.fig.update_xaxes(title_text="Timestamp", row=1, col=1)
+        self.fig.update_xaxes(title_text="Timestamp", row=2, col=1)
+        self.fig.update_xaxes(title_text="Residual Value", row=3, col=1)
+
+        self.fig.update_yaxes(title_text="Patient Volume", row=1, col=1)
+        self.fig.update_yaxes(title_text="Residual Value", row=2, col=1)
+        self.fig.update_yaxes(title_text="Frequency", row=3, col=1)
+
+        # Add zero reference line for residuals time series
+        self.fig.add_hline(
+            y=0,
+            line_dash="dash",
+            line_color="gray",
+            row=2,
+            col=1,
+        )
+
+    def visualize(self) -> None:
+        r"""Generate and display the complete visualization."""
+        self._create_traces()
+        self._update_layout()
+        self.fig.show()

kumoai/utils/progress_logger.py

@@ -0,0 +1,177 @@
+import sys
+import time
+from typing import Any, List, Optional, Union
+
+from rich.console import Console, ConsoleOptions, RenderResult
+from rich.live import Live
+from rich.padding import Padding
+from rich.progress import (
+    BarColumn,
+    MofNCompleteColumn,
+    Progress,
+    Task,
+    TextColumn,
+    TimeRemainingColumn,
+)
+from rich.spinner import Spinner
+from rich.table import Table
+from rich.text import Text
+from typing_extensions import Self
+
+
+class ProgressLogger:
+    def __init__(self, msg: str) -> None:
+        self.msg = msg
+        self.logs: List[str] = []
+
+        self.start_time: Optional[float] = None
+        self.end_time: Optional[float] = None
+
+    @property
+    def duration(self) -> float:
+        assert self.start_time is not None
+        if self.end_time is not None:
+            return self.end_time - self.start_time
+        return time.perf_counter() - self.start_time
+
+    def log(self, msg: str) -> None:
+        self.logs.append(msg)
+
+    def __enter__(self) -> Self:
+        self.start_time = time.perf_counter()
+        return self
+
+    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
+        self.end_time = time.perf_counter()
+
+    def __repr__(self) -> str:
+        return f'{self.__class__.__name__}({self.msg})'
+
+
+class ColoredMofNCompleteColumn(MofNCompleteColumn):
+    def __init__(self, style: str = 'green') -> None:
+        super().__init__()
+        self.style = style
+
+    def render(self, task: Task) -> Text:
+        return Text(str(super().render(task)), style=self.style)
+
+
+class ColoredTimeRemainingColumn(TimeRemainingColumn):
+    def __init__(self, style: str = 'cyan') -> None:
+        super().__init__()
+        self.style = style
+
+    def render(self, task: Task) -> Text:
+        return Text(str(super().render(task)), style=self.style)
+
+
+class InteractiveProgressLogger(ProgressLogger):
+    def __init__(
+        self,
+        msg: str,
+        verbose: bool = True,
+        refresh_per_second: int = 10,
+    ) -> None:
+        super().__init__(msg=msg)
+
+        self.verbose = verbose
+        self.refresh_per_second = refresh_per_second
+
+        self._progress: Optional[Progress] = None
+        self._task: Optional[int] = None
+
+        self._live: Optional[Live] = None
+        self._exception: bool = False
+
+    def init_progress(self, total: int, description: str) -> None:
+        assert self._progress is None
+        if self.verbose:
+            self._progress = Progress(
+                TextColumn(f'   ↳ {description}', style='dim'),
+                BarColumn(bar_width=None),
+                ColoredMofNCompleteColumn(style='dim'),
+                TextColumn('•', style='dim'),
+                ColoredTimeRemainingColumn(style='dim'),
+            )
+            self._task = self._progress.add_task("Progress", total=total)
+
+    def step(self) -> None:
+        if self.verbose:
+            assert self._progress is not None
+            assert self._task is not None
+            self._progress.update(self._task, advance=1)  # type: ignore
+
+    def __enter__(self) -> Self:
+        from kumoai import in_notebook
+
+        super().__enter__()
+
+        if not in_notebook():  # Render progress bar in TUI.
+            sys.stdout.write("\x1b]9;4;3\x07")
+            sys.stdout.flush()
+
+        if self.verbose:
+            self._live = Live(
+                self,
+                refresh_per_second=self.refresh_per_second,
+                vertical_overflow='visible',
+            )
+            self._live.start()
+
+        return self
+
+    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
+        from kumoai import in_notebook
+
+        super().__exit__(exc_type, exc_val, exc_tb)
+
+        if exc_type is not None:
+            self._exception = True
+
+        if self._progress is not None:
+            self._progress.stop()
+            self._progress = None
+            self._task = None
+
+        if self._live is not None:
+            self._live.update(self, refresh=True)
+            self._live.stop()
+            self._live = None
+
+        if not in_notebook():
+            sys.stdout.write("\x1b]9;4;0\x07")
+            sys.stdout.flush()
+
+    def __rich_console__(
+        self,
+        console: Console,
+        options: ConsoleOptions,
+    ) -> RenderResult:
+
+        table = Table.grid(padding=(0, 1))
+
+        icon: Union[Text, Padding]
+        if self._exception:
+            style = 'red'
+            icon = Text('❌', style=style)
+        elif self.end_time is not None:
+            style = 'green'
+            icon = Text('✅', style=style)
+        else:
+            style = 'cyan'
+            icon = Padding(Spinner('dots', style=style), (0, 1, 0, 0))
+
+        title = Text.from_markup(
+            f'{self.msg} ({self.duration:.2f}s)',
+            style=style,
+        )
+        table.add_row(icon, title)
+
+        for log in self.logs:
+            table.add_row('', Text(f'↳ {log}', style='dim'))
+
+        yield table
+
+        if self.verbose and self._progress is not None:
+            yield self._progress.get_renderable()

kumoai-2.13.0.dev202511131731.dist-info/METADATA

@@ -0,0 +1,60 @@
+Metadata-Version: 2.4
+Name: kumoai
+Version: 2.13.0.dev202511131731
+Summary: AI on the Modern Data Stack
+Author-email: "Kumo.AI" <hello@kumo.ai>
+License-Expression: MIT
+Project-URL: homepage, https://kumo.ai
+Project-URL: documentation, https://kumo.ai/docs
+Keywords: deep-learning,graph-neural-networks,cloud-data-warehouse
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3 :: Only
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pandas
+Requires-Dist: pyarrow<19.0.0,>=8.0.0
+Requires-Dist: requests>=2.28.2
+Requires-Dist: urllib3
+Requires-Dist: plotly
+Requires-Dist: typing_extensions>=4.5.0
+Requires-Dist: kumo-api==0.45.0
+Requires-Dist: tqdm>=4.66.0
+Requires-Dist: aiohttp>=3.10.0
+Requires-Dist: pydantic>=1.10.21
+Requires-Dist: rich>=9.0.0
+Provides-Extra: doc
+Requires-Dist: sphinx; extra == "doc"
+Requires-Dist: sphinx-book-theme; extra == "doc"
+Requires-Dist: sphinx-copybutton; extra == "doc"
+Requires-Dist: sphinx-autodoc-typehints; extra == "doc"
+Requires-Dist: graphviz; extra == "doc"
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: pytest-mock; extra == "test"
+Requires-Dist: requests-mock; extra == "test"
+Dynamic: license-file
+Dynamic: requires-dist
+
+<p align="center">
+  <img height="180" src="https://s3.us-west-1.amazonaws.com/data.kumo.ai/img/kumo_pink_md.svg" />
+</p>
+
+______________________________________________________________________
+
+The Kumo SDK implements a pythonic interface for users to programmatically
+interact with the Kumo machine learning platform
+([documentation](https://kumo-ai.github.io/kumo-sdk/docs/#)).
+
+## Installation
+
+The Kumo SDK is available for Python 3.10 to Python 3.13. To install, simply run
+
+```
+pip install kumoai
+```

kumoai-2.13.0.dev202511131731.dist-info/RECORD

@@ -0,0 +1,98 @@
+kumoai/__init__.py,sha256=LU1zmKYc0KV5hy2VGKUuXgSvbJwj2rSRQ_R_bpHyl1o,10708
+kumoai/_logging.py,sha256=U2_5ROdyk92P4xO4H2WJV8EC7dr6YxmmnM-b7QX9M7I,886
+kumoai/_singleton.py,sha256=UTwrbDkoZSGB8ZelorvprPDDv9uZkUi1q_SrmsyngpQ,836
+kumoai/_version.py,sha256=wvQ56Gsck9T4OuBZmYnas0mpqXe3G7yxp7naBy-NTH4,39
+kumoai/databricks.py,sha256=e6E4lOFvZHXFwh4CO1kXU1zzDU3AapLQYMxjiHPC-HQ,476
+kumoai/exceptions.py,sha256=b-_sdbAKOg50uaJZ65GmBLdTo4HANdjl8_R0sJpwaN0,833
+kumoai/formatting.py,sha256=jA_rLDCGKZI8WWCha-vtuLenVKTZvli99Tqpurz1H84,953
+kumoai/futures.py,sha256=oJFIfdCM_3nWIqQteBKYMY4fPhoYlYWE_JA2o6tx-ng,3737
+kumoai/jobs.py,sha256=NrdLEFNo7oeCYSy-kj2nAvCFrz9BZ_xrhkqHFHk5ksY,2496
+kumoai/kumolib.cpython-313-x86_64-linux-gnu.so,sha256=SDReSaBnZSOzE8FHJuAzJr3Zj6zqXhWr68v5RUflqyk,229072
+kumoai/mixin.py,sha256=MP413xzuCqWhxAPUHmloLA3j4ZyF1tEtfi516b_hOXQ,812
+kumoai/spcs.py,sha256=N4ddeoHAc4I3bKrDitsb91lUx5VKvCyPyMT3zWiuCcY,4275
+kumoai/artifact_export/__init__.py,sha256=BsfDrc3mCHpO9-BqvqKm8qrXDIwfdaoH5UIoG4eQkc4,238
+kumoai/artifact_export/config.py,sha256=jOPDduduxv0uuB-7xVlDiZglfpmFF5lzQhhH1SMkGvw,8024
+kumoai/artifact_export/job.py,sha256=GEisSwvcjK_35RgOfsLXGgxMTXIWm765B_BW_Kgs-V0,3275
+kumoai/client/__init__.py,sha256=MkyOuMaHQ2c8GPxjBDQSVFhfRE2d2_6CXQ6rxj4ps4w,64
+kumoai/client/client.py,sha256=S1OfGDwTzoyf40fhg111xGQGNfEP-OnoXqFV6X9iMEc,8580
+kumoai/client/connector.py,sha256=x3i2aBTJTEMZvYRcWkY-UfWVOANZjqAso4GBbcshFjw,3920
+kumoai/client/endpoints.py,sha256=iF2ZD25AJCIVbmBJ8tTZ8y1Ch0m6nTp18ydN7h4WiTk,5382
+kumoai/client/graph.py,sha256=zvLEDExLT_RVbUMHqVl0m6tO6s2gXmYSoWmPF6YMlnA,3831
+kumoai/client/jobs.py,sha256=iu_Wrta6BQMlV6ZtzSnmhjwNPKDMQDXOsqVVIyWodqw,17074
+kumoai/client/online.py,sha256=pkBBh_DEC3GAnPcNw6bopNRlGe7EUbIFe7_seQqZRaw,2720
+kumoai/client/pquery.py,sha256=R2hc-M8vPoyIDH0ywLwFVxCznVAqpZz3w2HszjdNW-o,6891
+kumoai/client/rfm.py,sha256=NxKk8mH2A-B58rSXhDWaph4KeiSyJYDq-RO-vAHh7es,3726
+kumoai/client/source_table.py,sha256=VCsCcM7KYcnjGP7HLTb-AOSEGEVsJTWjk8bMg1JdgPU,2101
+kumoai/client/table.py,sha256=cQG-RPm-e91idEgse1IPJDvBmzddIDGDkuyrR1rq4wU,3235
+kumoai/client/utils.py,sha256=lz1NubwMDHCwzQRowRXm7mjAoYRd5UjRQIwXdtWAl90,3849
+kumoai/codegen/__init__.py,sha256=-a1QPslwq1g0j9Hsg4Ky-m1FzQesRRQ3u22-pYUlYDs,399
+kumoai/codegen/cli.py,sha256=WTH81l3yplX01bvA6KjUyTKPoHcs51FD20A3dhRxuEQ,2848
+kumoai/codegen/context.py,sha256=mW1-jdy_cZqZsnYNTtMb-90KthlkJaJ9iU60ON9TXBU,519
+kumoai/codegen/edits.py,sha256=E3xT9k3wROeCZ6FZqZelch-S6ylRdJegPS-Ke7pCSgE,16436
+kumoai/codegen/exceptions.py,sha256=_wEalz96R8R60-a7klo_u-11q8mFFiOZM15lIuJU554,149
+kumoai/codegen/generate.py,sha256=SvfWWa71xSAOjH9645yQvgoEM-o4BYjupM_EpUxqB_E,7331
+kumoai/codegen/identity.py,sha256=P_QVdE1h22L3SqWIPcycZ3XYvIRRKQ50OQ8BOH7RzSs,3985
+kumoai/codegen/loader.py,sha256=X9zYfrFHM0kz9xAq4MSFNxsDw7nF-X2_kd24fg757xE,3564
+kumoai/codegen/naming.py,sha256=_XVQGxHfuub4bhvyuBKjltD5Lm_oPpibvP_LZteCGk0,3021
+kumoai/codegen/registry.py,sha256=JtvQT3UpWgFO3fhVQyZW-iisVegFzie01cJKDVn1HjA,4170
+kumoai/codegen/handlers/__init__.py,sha256=k8TB_Kn-1BycBBi51kqFS2fZHCpCPgR9-3J9g9tvLQU,117
+kumoai/codegen/handlers/connector.py,sha256=afGf_GreyQ9y6qF3QTgSiM416qtUcP298SatNqUFhvQ,3828
+kumoai/codegen/handlers/graph.py,sha256=dGbPP99Lt3YrZdc_61IacrxG4seNdKqt9pbXc1z-qHE,2520
+kumoai/codegen/handlers/pquery.py,sha256=m8rOLp8YcVB-hTpMH1OZui5tQ5QJkIrwt2OpSsCkO3g,1977
+kumoai/codegen/handlers/table.py,sha256=POHpA-GFYFGTSuerGmtigYablk-Wq1L3EBvsOI-iFMQ,3956
+kumoai/codegen/handlers/utils.py,sha256=58b2GCgaTBUp2aId7BLMXMV0ENrusbNbfw7mlyXAXPE,1447
+kumoai/connector/__init__.py,sha256=9g6oNJ0qHWFlL5enTSoK4_SSH_5hP74xUDZx-9SggC4,842
+kumoai/connector/base.py,sha256=cujXSZF3zAfuxNuEw54DSL1T7XCuR4t0shSMDuPUagQ,5291
+kumoai/connector/bigquery_connector.py,sha256=IkyRqvF8Cg96kApUuuz86eYnl-BqBmDX1f_jIKL0pxc,7082
+kumoai/connector/databricks_connector.py,sha256=YQy203XHZGzNJ8bPUjUOnrVt2KlpgMdVuTHpc6sVCcs,7574
+kumoai/connector/file_upload_connector.py,sha256=swp03HgChOvmNPJetuujBSAqADe7NRmS_T0F3o9it4w,7008
+kumoai/connector/glue_connector.py,sha256=HivT0QYQ8-XeB4QLgWvghiqXuq7jyBK9G2R1py_NnE4,4697
+kumoai/connector/s3_connector.py,sha256=3kbv-h7DwD8O260Q0h1GPm5wwQpLt-Tb3d_CBSaie44,10155
+kumoai/connector/snowflake_connector.py,sha256=K0s-H9tW3rve8g2x1PbyxvzSpkROfGQZz-Qa4PoT4UE,9022
+kumoai/connector/source_table.py,sha256=QLT8bEYaxeMwy-b168url0VfnkTrs5K6VKLbxTI4hEY,17539
+kumoai/connector/utils.py,sha256=PUjunLpfqMZsrPDo2EmnyJRBl_mt-E6ugv2kNkf5Rn8,64011
+kumoai/encoder/__init__.py,sha256=VPGs4miBC_WfwWeOXeHhFomOUocERFavhKf5fqITcds,182
+kumoai/experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kumoai/experimental/rfm/__init__.py,sha256=ornmi2x947jkQLptMn7ZLvTf2Sw-RMcVW73AnjVsWAo,1709
+kumoai/experimental/rfm/authenticate.py,sha256=FiuHMvP7V3zBZUlHMDMbNLhc-UgDZgz4hjVSTuQ7DRw,18888
+kumoai/experimental/rfm/local_graph.py,sha256=2iJDlsGVzqCe1bD_puXWlhwGkn7YnQyJ4p4C-fwCZNE,30076
+kumoai/experimental/rfm/local_graph_sampler.py,sha256=o60_sdMa_fr60DrdmCIaE6lKQAD2msp1t-GGubFNt-o,6738
+kumoai/experimental/rfm/local_graph_store.py,sha256=8BqonuaMftAAsjgZpB369i5AeNd1PkisMbbEqc0cKBo,13847
+kumoai/experimental/rfm/local_pquery_driver.py,sha256=aO7Jfwx9gxGKYvpqxZx1LLWdI1MhuZQOPtAITxoOQO0,26162
+kumoai/experimental/rfm/local_table.py,sha256=r8xZ33Mjs6JD8ud6h23tZ99Dag2DvZ4h6tWjmGrKQg4,19605
+kumoai/experimental/rfm/rfm.py,sha256=OHE6xpVREfm5GtZ4LXPY71FKzjNc94LWiBTPrmE6Xzc,48097
+kumoai/experimental/rfm/utils.py,sha256=3IiBvT_aLBkkcJh3H11_50yt_XlEzHR0cm9Kprrtl8k,11123
+kumoai/experimental/rfm/infer/__init__.py,sha256=xQ8_SuejIzXyn2J7bIKX3pXumFtRuEfBtE5oEDUDJjI,293
+kumoai/experimental/rfm/infer/categorical.py,sha256=VwNaKwKbRYkTxEJ1R6gziffC8dGsEThcDEfbi-KqW5c,853
+kumoai/experimental/rfm/infer/id.py,sha256=ZIO0DWIoiEoS_8MVc5lkqBfkTWWQ0yGCgjkwLdaYa_Q,908
+kumoai/experimental/rfm/infer/multicategorical.py,sha256=0-cLpDnGryhr76QhZNO-klKokJ6MUSfxXcGdQ61oykY,1102
+kumoai/experimental/rfm/infer/timestamp.py,sha256=vM9--7eStzaGG13Y-oLYlpNJyhL6f9dp17HDXwtl_DM,1094
+kumoai/experimental/rfm/pquery/__init__.py,sha256=X0O3EIq5SMfBEE-ii5Cq6iDhR3s3XMXB52Cx5htoePw,152
+kumoai/experimental/rfm/pquery/executor.py,sha256=f7-pJhL0BgFU9E4o4gQpQyArOvyrZtwxFmks34-QOAE,2741
+kumoai/experimental/rfm/pquery/pandas_executor.py,sha256=kiBJq7uVGbasG7TiqsubEl6ey3UYzZiM4bwxILqp_54,18487
+kumoai/graph/__init__.py,sha256=n8X4X8luox4hPBHTRC9R-3JzvYYMoR8n7lF1H4w4Hzc,228
+kumoai/graph/column.py,sha256=t7wBmcx0VYKXjIoESU9Nq-AisiJOdlqd80t8zby1R8Y,4189
+kumoai/graph/graph.py,sha256=iyp4klPIMn2ttuEqMJvsrxKb_tmz_DTnvziIhCegduM,38291
+kumoai/graph/table.py,sha256=nZqYX8xlyAz6kVtlE2vf9BAIOCoWeFNIfbGbReDCb7k,33888
+kumoai/pquery/__init__.py,sha256=uTXr7t1eXcVfM-ETaM_1ImfEqhrmaj8BjiIvy1YZTL8,533
+kumoai/pquery/prediction_table.py,sha256=QPDH22X1UB0NIufY7qGuV2XW7brG3Pv--FbjNezzM2g,10776
+kumoai/pquery/predictive_query.py,sha256=oUqwdOWLLkPM-G4PhpUk_6mwSJGBtaD3t37Wp5Oow8M,24971
+kumoai/pquery/training_table.py,sha256=elmPDZx11kPiC_dkOhJcBUGtHKgL32GCBvZ9k6U0pMg,15809
+kumoai/testing/__init__.py,sha256=goHIIo3JE7uHV7njo4_aTd89mVVR74BEAZ2uyBaOR0w,170
+kumoai/testing/decorators.py,sha256=RiFrJcP-ym-mB1BYSGC26bBiryxoR9-GwL1G4EHc2sc,1591
+kumoai/trainer/__init__.py,sha256=zUdFl-f-sBWmm2x8R-rdVzPBeU2FaMzUY5mkcgoTa1k,939
+kumoai/trainer/baseline_trainer.py,sha256=LlfViNOmswNv4c6zJJLsyv0pC2mM2WKMGYx06ogtEVc,4024
+kumoai/trainer/config.py,sha256=-2RfK10AsVVThSyfWtlyfH4Fc4EwTdu0V3yrDRtIOjk,98
+kumoai/trainer/job.py,sha256=Wk69nzFhbvuA3nEvtCstI04z5CxkgvQ6tHnGchE0Lkg,44938
+kumoai/trainer/online_serving.py,sha256=9cddb5paeZaCgbUeceQdAOxysCtV5XP-KcsgFz_XR5w,9566
+kumoai/trainer/trainer.py,sha256=hBXO7gwpo3t59zKFTeIkK65B8QRmWCwO33sbDuEAPlY,20133
+kumoai/trainer/util.py,sha256=bDPGkMF9KOy4HgtA-OwhXP17z9cbrfMnZGtyGuUq_Eo,4062
+kumoai/utils/__init__.py,sha256=wGDC_31XJ-7ipm6eawjLAJaP4EfmtNOH8BHzaetQ9Ko,268
+kumoai/utils/datasets.py,sha256=ptKIUoBONVD55pTVNdRCkQT3NWdN_r9UAUu4xewPa3U,2928
+kumoai/utils/forecasting.py,sha256=-nDS6ucKNfQhTQOfebjefj0wwWH3-KYNslIomxwwMBM,7415
+kumoai/utils/progress_logger.py,sha256=pngEGzMHkiOUKOa6fbzxCEc2xlA4SJKV4TDTVVoqObM,5062
+kumoai-2.13.0.dev202511131731.dist-info/METADATA,sha256=jnAtMb9qlOGKD4KIpUq9Tt65dnfVaeCNZUtJ3d6-Ihw,2052
+kumoai-2.13.0.dev202511131731.dist-info/WHEEL,sha256=FGhoiLBD6z3eO88tnYRS0TgLaWveAr5uyXIqBlx6mno,152
+kumoai-2.13.0.dev202511131731.dist-info/top_level.txt,sha256=YjU6UcmomoDx30vEXLsOU784ED7VztQOsFApk1SFwvs,7
+kumoai-2.13.0.dev202511131731.dist-info/RECORD,,
+kumoai-2.13.0.dev202511131731.dist-info/licenses/LICENSE,sha256=TbWlyqRmhq9PEzCaTI0H0nWLQCCOywQM8wYH8MbjfLo,1102

kumoai-2.13.0.dev202511131731.dist-info/WHEEL

@@ -0,0 +1,6 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: false
+Tag: cp313-cp313-manylinux_2_27_x86_64
+Tag: cp313-cp313-manylinux_2_28_x86_64
+

kumoai-2.13.0.dev202511131731.dist-info/licenses/LICENSE

@@ -0,0 +1,9 @@
+MIT License
+
+Copyright (c) 2025 Kumo.ai, Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the " Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice (including the next paragraph) shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

kumoai-2.13.0.dev202511131731.dist-info/top_level.txt

@@ -0,0 +1 @@
+kumoai