kumoai 2.14.0.dev202512211732__cp313-cp313-macosx_11_0_arm64.whl → 2.15.0.dev202601181732__cp313-cp313-macosx_11_0_arm64.whl

kumoai/utils/progress_logger.py

@@ -1,6 +1,7 @@
 import re
 import sys
 import time
+from abc import ABC, abstractmethod
 from typing import Any
 
 from rich.console import Console, ConsoleOptions, RenderResult
@@ -20,49 +21,179 @@ from rich.text import Text
 from typing_extensions import Self
 
 
-class ProgressLogger:
+class ProgressLogger(ABC):
+    r"""An abstract base class for logging progress updates."""
     def __init__(self, msg: str, verbose: bool = True) -> None:
-        self.msg = msg
-        self.verbose = verbose
+        self.msg: str = msg
+        self.verbose: bool = verbose
 
         self.logs: list[str] = []
 
         self.start_time: float | None = None
         self.end_time: float | None = None
 
+        # Handle nested loggers gracefully:
+        self._depth: int = 0
+
+        # Internal progress bar cache:
+        self._progress_bar_msg: str | None = None
+        self._total: int = 0
+        self._current: int = 0
+
+    def __repr__(self) -> str:
+        return f'{self.__class__.__name__}()'
+
     @classmethod
     def default(cls, msg: str, verbose: bool = True) -> 'ProgressLogger':
-        from kumoai import in_snowflake_notebook
+        r"""The default progress logger for the current environment."""
+        from kumoai import in_streamlit_notebook, in_vnext_notebook
 
-        if in_snowflake_notebook():
+        if in_streamlit_notebook():
             return StreamlitProgressLogger(msg, verbose)
+        if in_vnext_notebook():
+            return PlainProgressLogger(msg, verbose)
         return RichProgressLogger(msg, verbose)
 
     @property
     def duration(self) -> float:
+        r"""The current/final duration."""
         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 __enter__(self) -> Self:
+        from kumoai import in_notebook
+
+        self._depth += 1
+        if self._depth == 1:
+            self.start_time = time.perf_counter()
+        if self._depth == 1 and not in_notebook():  # Show progress bar in TUI.
+            sys.stdout.write("\x1b]9;4;3\x07")
+            sys.stdout.flush()
+        if self._depth == 1 and self.verbose:
+            self.on_enter()
+        return self
+
+    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
+        from kumoai import in_notebook
+
+        self._depth -= 1
+        if self._depth == 0:
+            self.end_time = time.perf_counter()
+        if self._depth == 0 and self.verbose:
+            self.on_exit(error=exc_val is not None)
+        if self._depth == 0 and not in_notebook():  # Stop progress bar in TUI.
+            sys.stdout.write("\x1b]9;4;0\x07")
+            sys.stdout.flush()
+
     def log(self, msg: str) -> None:
+        r"""Logs a new message."""
         self.logs.append(msg)
+        if self.verbose:
+            self.on_log(msg)
 
-    def init_progress(self, total: int, description: str) -> None:
-        pass
+    def init_progress(self, msg: str, total: int) -> None:
+        r"""Initializes a progress bar."""
+        if self._progress_bar_msg is not None:
+            raise RuntimeError("Current progress not yet finished")
+        self._progress_bar_msg = msg
+        self._current = 0
+        self._total = total
+        if self.verbose:
+            self.on_init_progress(msg, total)
 
     def step(self) -> None:
+        r"""Increments an active progress bar."""
+        assert self._progress_bar_msg is not None
+        self._current += 1
+        if self.verbose:
+            self.on_step(self._progress_bar_msg, self._current, self._total)
+        if self._current >= self._total:
+            self._progress_bar_msg = None
+            self._current = self._total = 0
+
+    @abstractmethod
+    def on_enter(self) -> None:
         pass
 
-    def __enter__(self) -> Self:
-        self.start_time = time.perf_counter()
-        return self
+    @abstractmethod
+    def on_exit(self, error: bool) -> None:
+        pass
 
-    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
-        self.end_time = time.perf_counter()
+    @abstractmethod
+    def on_log(self, msg: str) -> None:
+        pass
 
-    def __repr__(self) -> str:
-        return f'{self.__class__.__name__}({self.msg})'
+    @abstractmethod
+    def on_init_progress(self, msg: str, total: int) -> None:
+        pass
+
+    @abstractmethod
+    def on_step(self, msg: str, current: int, total: int) -> None:
+        pass
+
+
+class PlainProgressLogger(ProgressLogger):
+    RESET: str = '\x1b[0m'
+    BOLD: str = '\x1b[1m'
+    DIM: str = '\x1b[2m'
+    RED: str = '\x1b[31m'
+    GREEN: str = '\x1b[32m'
+    CYAN: str = '\x1b[36m'
+
+    def on_enter(self) -> None:
+        from kumoai import in_vnext_notebook
+
+        msg = self.msg.replace('[bold]', self.BOLD)
+        msg = msg.replace('[/bold]', self.RESET + self.CYAN)
+        msg = self.CYAN + msg + self.RESET
+        print(msg, end='\n' if in_vnext_notebook() else '', flush=True)
+
+    def on_exit(self, error: bool) -> None:
+        from kumoai import in_vnext_notebook
+
+        if error:
+            msg = f"❌ {self.RED}({self.duration:.2f}s){self.RESET}"
+        else:
+            msg = f"✅ {self.GREEN}({self.duration:.2f}s){self.RESET}"
+
+        if in_vnext_notebook():
+            print(f"{self.DIM}↳{self.RESET} {msg}", flush=True)
+        else:
+            print(f" {msg}", flush=True)
+
+    def on_log(self, msg: str) -> None:
+        from kumoai import in_vnext_notebook
+
+        msg = f"{self.DIM}↳ {msg}{self.RESET}"
+
+        if in_vnext_notebook():
+            print(msg, flush=True)
+        else:
+            print(f"\n{msg}", end='', flush=True)
+
+    def on_init_progress(self, msg: str, total: int) -> None:
+        from kumoai import in_vnext_notebook
+
+        msg = f"{self.DIM}↳ {msg}{self.RESET}"
+
+        if in_vnext_notebook():
+            print(msg, flush=True)
+        else:
+            print(f"\n{msg} {self.DIM}[{self.RESET}", end='', flush=True)
+
+    def on_step(self, msg: str, current: int, total: int) -> None:
+        from kumoai import in_vnext_notebook
+
+        if in_vnext_notebook():
+            return
+
+        msg = f"{self.DIM}#{self.RESET}"
+        if current == total:
+            msg += f"{self.DIM}]{self.RESET}"
+
+        print(msg, end='', flush=True)
 
 
 class ColoredMofNCompleteColumn(MofNCompleteColumn):
@@ -100,65 +231,51 @@ class RichProgressLogger(ProgressLogger):
         self._live: Live | None = 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
+    def on_enter(self) -> None:
+        self._live = Live(
+            self,
+            refresh_per_second=self.refresh_per_second,
+            vertical_overflow='visible',
+        )
+        self._live.start()
 
-        super().__enter__()
+    def on_exit(self, error: bool) -> None:
+        self._exception = error
 
-        if not in_notebook():  # Render progress bar in TUI.
-            sys.stdout.write("\x1b]9;4;3\x07")
-            sys.stdout.flush()
+        if self._progress is not None:
+            self._progress.stop()
 
-        if self.verbose:
-            self._live = Live(
-                self,
-                refresh_per_second=self.refresh_per_second,
-                vertical_overflow='visible',
-            )
-            self._live.start()
+        if self._live is not None:
+            self._live.update(self, refresh=True)
+            self._live.stop()
 
-        return self
+        self._progress = None
+        self._task = None
+        self._live = None
 
-    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
-        from kumoai import in_notebook
+    def on_log(self, msg: str) -> None:
+        pass
 
-        super().__exit__(exc_type, exc_val, exc_tb)
+    def on_init_progress(self, msg: str, total: int) -> None:
+        self._progress = Progress(
+            TextColumn(f'   ↳ {msg}', style='dim'),
+            BarColumn(bar_width=None),
+            ColoredMofNCompleteColumn(style='dim'),
+            TextColumn('•', style='dim'),
+            ColoredTimeRemainingColumn(style='dim'),
+        )
+        self._task = self._progress.add_task("Progress", total=total)
 
-        if exc_type is not None:
-            self._exception = True
+    def on_step(self, msg: str, current: int, total: int) -> None:
+        assert self._progress is not None
+        assert self._task is not None
+        self._progress.update(self._task, advance=1)  # type: ignore
 
-        if self._progress is not None:
+        if current == total:
             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,
@@ -189,7 +306,7 @@ class RichProgressLogger(ProgressLogger):
 
         yield table
 
-        if self.verbose and self._progress is not None:
+        if self._progress is not None:
             yield self._progress.get_renderable()
 
 
@@ -202,78 +319,49 @@ class StreamlitProgressLogger(ProgressLogger):
         super().__init__(msg=msg, verbose=verbose)
 
         self._status: Any = None
-
-        self._total = 0
-        self._current = 0
-        self._description: str = ''
         self._progress: Any = None
 
-    def __enter__(self) -> Self:
-        super().__enter__()
+    @staticmethod
+    def _sanitize_text(msg: str) -> str:
+        return re.sub(r'\[/?bold\]', '**', msg)
 
+    def on_enter(self) -> None:
         import streamlit as st
 
         # Adjust layout for prettier output:
         st.markdown(STREAMLIT_CSS, unsafe_allow_html=True)
 
-        if self.verbose:
-            self._status = st.status(
-                f':blue[{self._sanitize_text(self.msg)}]',
+        self._status = st.status(
+            f':blue[{self._sanitize_text(self.msg)}]',
+            expanded=True,
+        )
+
+    def on_exit(self, error: bool) -> None:
+        if self._status is not None:
+            label = f'{self._sanitize_text(self.msg)} ({self.duration:.2f}s)'
+            self._status.update(
+                label=f':red[{label}]' if error else f':green[{label}]',
+                state='error' if error else 'complete',
                 expanded=True,
             )
 
-        return self
+    def on_log(self, msg: str) -> None:
+        if self._status is not None:
+            self._status.write(msg)
 
-    def log(self, msg: str) -> None:
-        super().log(msg)
-        if self.verbose and self._status is not None:
-            self._status.write(self._sanitize_text(msg))
-
-    def init_progress(self, total: int, description: str) -> None:
-        if self.verbose and self._status is not None:
-            self._total = total
-            self._current = 0
-            self._description = self._sanitize_text(description)
-            percent = min(self._current / self._total, 1.0)
+    def on_init_progress(self, msg: str, total: int) -> None:
+        if self._status is not None:
             self._progress = self._status.progress(
-                value=percent,
-                text=f'{self._description} [{self._current}/{self._total}]',
+                value=0.0,
+                text=f'{msg} [{0}/{total}]',
             )
 
-    def step(self) -> None:
-        self._current += 1
-
-        if self.verbose and self._progress is not None:
-            percent = min(self._current / self._total, 1.0)
+    def on_step(self, msg: str, current: int, total: int) -> None:
+        if self._progress is not None:
             self._progress.progress(
-                value=percent,
-                text=f'{self._description} [{self._current}/{self._total}]',
-            )
-
-    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
-        super().__exit__(exc_type, exc_val, exc_tb)
-
-        if not self.verbose or self._status is None:
-            return
-
-        label = f'{self._sanitize_text(self.msg)} ({self.duration:.2f}s)'
-
-        if exc_type is not None:
-            self._status.update(
-                label=f':red[{label}]',
-                state='error',
-                expanded=True,
+                value=min(current / total, 1.0),
+                text=f'{msg} [{current}/{total}]',
             )
-        else:
-            self._status.update(
-                label=f':green[{label}]',
-                state='complete',
-                expanded=True,
-            )
-
-    @staticmethod
-    def _sanitize_text(msg: str) -> str:
-        return re.sub(r'\[/?bold\]', '**', msg)
 
 
 STREAMLIT_CSS = """

kumoai/utils/sql.py

@@ -1,3 +1,3 @@
-def quote_ident(name: str) -> str:
+def quote_ident(ident: str, char: str = '"') -> str:
     r"""Quotes a SQL identifier."""
-    return '"' + name.replace('"', '""') + '"'
+    return char + ident.replace(char, char + char) + char

{kumoai-2.14.0.dev202512211732.dist-info → kumoai-2.15.0.dev202601181732.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kumoai
-Version: 2.14.0.dev202512211732
+Version: 2.15.0.dev202601181732
 Summary: AI on the Modern Data Stack
 Author-email: "Kumo.AI" <hello@kumo.ai>
 License-Expression: MIT
@@ -23,7 +23,7 @@ Requires-Dist: requests>=2.28.2
 Requires-Dist: urllib3
 Requires-Dist: plotly
 Requires-Dist: typing_extensions>=4.5.0
-Requires-Dist: kumo-api==0.49.0
+Requires-Dist: kumo-api<1.0.0,>=0.53.0
 Requires-Dist: tqdm>=4.66.0
 Requires-Dist: aiohttp>=3.10.0
 Requires-Dist: pydantic>=1.10.21

{kumoai-2.14.0.dev202512211732.dist-info → kumoai-2.15.0.dev202601181732.dist-info}/RECORD

@@ -1,8 +1,8 @@
 kumoai/kumolib.cpython-313-darwin.so,sha256=waBv-DiZ3WcasxiCQ-OM9EbSTgTtCfBTZIibXAK-JiQ,232816
 kumoai/_logging.py,sha256=U2_5ROdyk92P4xO4H2WJV8EC7dr6YxmmnM-b7QX9M7I,886
 kumoai/mixin.py,sha256=MP413xzuCqWhxAPUHmloLA3j4ZyF1tEtfi516b_hOXQ,812
-kumoai/_version.py,sha256=r_ZJN4_1Q2z42px3GUoxeFu6s1G0ZVF0s5q_k2WJWS8,39
-kumoai/__init__.py,sha256=Nn9YH_x9kAeEFn8RWbP95slZow0qFnakPZZ1WADe1hY,10843
+kumoai/_version.py,sha256=2ksd2GuX-AZRJGtcuDxCIRV0etIpcKZFRxJlf6Of638,39
+kumoai/__init__.py,sha256=n2Mi2n5S_WKpxpCInQKfGEmsIWVwrX86nGnYn5HwtIE,11171
 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
@@ -11,42 +11,45 @@ kumoai/databricks.py,sha256=e6E4lOFvZHXFwh4CO1kXU1zzDU3AapLQYMxjiHPC-HQ,476
 kumoai/spcs.py,sha256=N31d7rLa-bgYh8e2J4YzX1ScxGLqiVXrqJnCl1y4Mts,4139
 kumoai/_singleton.py,sha256=UTwrbDkoZSGB8ZelorvprPDDv9uZkUi1q_SrmsyngpQ,836
 kumoai/experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kumoai/experimental/rfm/graph.py,sha256=63Sovf7Da4q3FLRKTMcImB4v_VkjC87AdCBQm385w6w,46270
-kumoai/experimental/rfm/__init__.py,sha256=9aelcHodt2Oriw76vdEmtWrmAQ0CXTdFPrKgwVB9eKc,7124
+kumoai/experimental/rfm/relbench.py,sha256=cVsxxV3TIL3PLEoYb-8tAVW3GSef6NQAd3rxdHJL63I,2276
+kumoai/experimental/rfm/graph.py,sha256=4Jo17oYSoZouzvNQT2-Ai9GOX-bIdefPcxj_gcoM3dI,46873
+kumoai/experimental/rfm/__init__.py,sha256=bW2XyYtkbdiu_iICYFF2Fu1Fx5fyGbqne6m_6c1P-fY,7016
 kumoai/experimental/rfm/sagemaker.py,sha256=6fyXO1Jd_scq-DH7kcv6JcV8QPyTbh4ceqwQDPADlZ0,4963
-kumoai/experimental/rfm/rfm.py,sha256=Yssmo-PaCfjT9hll0BKl8fahsuKpG-gViwFdKH1F3os,50247
+kumoai/experimental/rfm/rfm.py,sha256=XsxwiDIvlZ_js7rvvffrOiXFsLX15-C7N0T9M-aptCw,60017
 kumoai/experimental/rfm/authenticate.py,sha256=G2RkRWznMVQUzvhvbKhn0bMCY7VmoNYxluz3THRqSdE,18851
+kumoai/experimental/rfm/task_table.py,sha256=n_gZNQlCqHOiAkbeaa18nnQ-amt1oWKA9riO2rkrZuw,9847
 kumoai/experimental/rfm/backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kumoai/experimental/rfm/backend/sqlite/__init__.py,sha256=jl-DBbhsqQ-dUXyWhyQTM1AU2qNAtXCmi1mokdhtBTg,902
-kumoai/experimental/rfm/backend/sqlite/table.py,sha256=JpHtA6Sdz96VpqC73h_1YnuW8mQ9cMInG6y4KmjZYHE,5983
-kumoai/experimental/rfm/backend/sqlite/sampler.py,sha256=pBCj0bXnh1PMg9cYJw-K983FzJX1_SLOe3QuOxkmvBQ,14242
+kumoai/experimental/rfm/backend/sqlite/table.py,sha256=WqYtd_rwlawItRMXZUfv14qdyU6huQmODuFjDo483dI,6683
+kumoai/experimental/rfm/backend/sqlite/sampler.py,sha256=I-zaSMd5XLg0qaJOoCR8arFBauUfhW_ZMl7gI97ress,18699
 kumoai/experimental/rfm/backend/local/__init__.py,sha256=2s9sSA-E-8pfkkzCH4XPuaSxSznEURMfMgwEIfYYPsg,1014
-kumoai/experimental/rfm/backend/local/table.py,sha256=VP3s-MgWKXPjezgRfc2bxr-9Fp2X5i9V2Do0IiNAmVM,3819
-kumoai/experimental/rfm/backend/local/graph_store.py,sha256=cY9KeLir9Xsp4MJl_K0VZckNa_LMKiiWCZG14uu21JI,11854
-kumoai/experimental/rfm/backend/local/sampler.py,sha256=85HoHCDiFOiuD_vFPZRx9JCyQUlLsqgsuB3NAw50wNw,10836
+kumoai/experimental/rfm/backend/local/table.py,sha256=GKeYGcu52ztCU8EBMqp5UVj85E145Ug41xiCPiTCXq4,3489
+kumoai/experimental/rfm/backend/local/graph_store.py,sha256=RHhkI13KpdPxqb4vXkwEwuFiX5DkrEsfZsOLywNnrvU,11294
+kumoai/experimental/rfm/backend/local/sampler.py,sha256=UKxTjsYs00sYuV_LAlDuZOvQq0BZzPCzZK1Fki2Fd70,10726
 kumoai/experimental/rfm/backend/snow/__init__.py,sha256=BYfsiuJ4Ee30GjG9EuUtitMHXnRfvVKi85zNlIwldV4,993
-kumoai/experimental/rfm/backend/snow/table.py,sha256=0W4jtb3MdYpsGzjU7UBS8xt8yfvYIou3nX4woVd-vZQ,6282
-kumoai/experimental/rfm/backend/snow/sampler.py,sha256=oNiBTo-Dr6LNCFJ9uHzLfKFYtx0rZq-Do2UOluwWWt0,10010
+kumoai/experimental/rfm/backend/snow/table.py,sha256=1RXpPiTxawTTOFprXvu7jDLG0ZGio_vE9lSfB6wqbWM,9078
+kumoai/experimental/rfm/backend/snow/sampler.py,sha256=tDOEiPTFFG6pWDcuuTvaOBAsMJLsxu4PzqryIgH1Kb4,16322
 kumoai/experimental/rfm/pquery/__init__.py,sha256=X0O3EIq5SMfBEE-ii5Cq6iDhR3s3XMXB52Cx5htoePw,152
 kumoai/experimental/rfm/pquery/pandas_executor.py,sha256=MwSvFRwLq-z19LEdF0G0AT7Gj9tCqu-XLEA7mNbqXwc,18454
 kumoai/experimental/rfm/pquery/executor.py,sha256=gs5AVNaA50ci8zXOBD3qt5szdTReSwTs4BGuEyx4BEE,2728
-kumoai/experimental/rfm/infer/multicategorical.py,sha256=0-cLpDnGryhr76QhZNO-klKokJ6MUSfxXcGdQ61oykY,1102
+kumoai/experimental/rfm/infer/multicategorical.py,sha256=lNO_8aJw1whO6QVEMB3PRWMNlEEiX44g3v4tP88TSQY,1119
 kumoai/experimental/rfm/infer/categorical.py,sha256=VwNaKwKbRYkTxEJ1R6gziffC8dGsEThcDEfbi-KqW5c,853
-kumoai/experimental/rfm/infer/time_col.py,sha256=oNenUK6P7ql8uwShodtQ73uG1x3fbFWT78jRcF9DLTI,1789
+kumoai/experimental/rfm/infer/time_col.py,sha256=iw_aUcHD2bHr7uRa3E7uDC30kU37aLIRTVAFdQEpt68,1818
 kumoai/experimental/rfm/infer/pkey.py,sha256=IaJI5GHK8ds_a3AOr3YYVgUlSmYYEgr4Nu92s2RyBV4,4412
 kumoai/experimental/rfm/infer/id.py,sha256=ZIO0DWIoiEoS_8MVc5lkqBfkTWWQ0yGCgjkwLdaYa_Q,908
-kumoai/experimental/rfm/infer/dtype.py,sha256=rzZRafxjr_mhvC7jeAZHVBvlbHJwmHvMlQqDuuePX8k,2635
+kumoai/experimental/rfm/infer/dtype.py,sha256=fbRRyyKSzO4riqX3RlhvBK7DhnjhwTgZVUjQ9inVPYI,2811
 kumoai/experimental/rfm/infer/__init__.py,sha256=8GDxQKd0pxZULdk7mpwl3CsOpL4v2HPuPEsbi2t_vzc,519
 kumoai/experimental/rfm/infer/timestamp.py,sha256=vM9--7eStzaGG13Y-oLYlpNJyhL6f9dp17HDXwtl_DM,1094
 kumoai/experimental/rfm/infer/stype.py,sha256=fu4zsOB-C7jNeMnq6dsK4bOZSewe7PtZe_AkohSRLoM,894
-kumoai/experimental/rfm/base/sql_sampler.py,sha256=-2dyftqvfbzMceIhE6i4wYFt7-p7FDeqlfH4P--qjWw,2598
-kumoai/experimental/rfm/base/__init__.py,sha256=E8Zrh52vczubhepBbS5AIe5wiBFeXbQCzfQ1-jLg_rU,885
-kumoai/experimental/rfm/base/sql_table.py,sha256=JP-fX8SLHGKeltMxkc2gdQ4RwW0DU3QHg5sjixMXRLs,8251
-kumoai/experimental/rfm/base/table.py,sha256=bRn3lP0Vc5uxCgsH8YD3aeMY6MnVTxi7eCMj92MhcfM,20446
-kumoai/experimental/rfm/base/column_expression.py,sha256=OoZXSaQRB7cBLVDEg1gcp65jHpUUO7e-vW-_asiAUPo,1261
-kumoai/experimental/rfm/base/sampler.py,sha256=aCD98t0CUhAvGXEFv24Vq2g4otuclpKkkyL1rMR_mFg,31449
-kumoai/experimental/rfm/base/source.py,sha256=RqlI_kBoRV0ADb8KdEKn15RNHMdFUzEVzb57lIoyBM4,294
-kumoai/experimental/rfm/base/column.py,sha256=wmKlqixNCmrmrAhSHfymJp76izpaykF9bdxNqgKbJzw,2288
+kumoai/experimental/rfm/base/sql_sampler.py,sha256=_go8TnH7AHki-0gg_pB7xd228VYhogQh10OkxT7PEnI,15682
+kumoai/experimental/rfm/base/mapper.py,sha256=WbWXSF8Vkdeud7UeQ2JgSX7z4d27b_b6o7nR4zET1aw,2420
+kumoai/experimental/rfm/base/__init__.py,sha256=rjmMux5lG8srw1bjQGcFQFv6zET9e5riP81nPkw28Jg,724
+kumoai/experimental/rfm/base/utils.py,sha256=Easg1bvjPLR8oZIoxIQCtCyl92pp2dUskdnSv1eayxQ,1133
+kumoai/experimental/rfm/base/table.py,sha256=eJuOUM64VWDkHaslNgeR5A_FZjlPF_4czC8OfFGR62E,26015
+kumoai/experimental/rfm/base/sampler.py,sha256=2G6VmgAGV1mSQWHK4wUgf5Ngr8nnH8Hg6_D3sPZZx1A,31951
+kumoai/experimental/rfm/base/expression.py,sha256=Y7NtLTnKlx6euG_N3fLTcrFKheB6P5KS_jhCfoXV9DE,1252
+kumoai/experimental/rfm/base/source.py,sha256=bwu3GU2TvIXR2fwKAmJ1-5BDoNXMnI1SU3Fgdk8lWnc,301
+kumoai/experimental/rfm/base/column.py,sha256=GXzLC-VpShr6PecUzaj1MJKc_PHzfW5Jn9bOYPA8fFA,4965
 kumoai/encoder/__init__.py,sha256=VPGs4miBC_WfwWeOXeHhFomOUocERFavhKf5fqITcds,182
 kumoai/graph/graph.py,sha256=iyp4klPIMn2ttuEqMJvsrxKb_tmz_DTnvziIhCegduM,38291
 kumoai/graph/__init__.py,sha256=n8X4X8luox4hPBHTRC9R-3JzvYYMoR8n7lF1H4w4Hzc,228
@@ -57,8 +60,9 @@ kumoai/artifact_export/job.py,sha256=GEisSwvcjK_35RgOfsLXGgxMTXIWm765B_BW_Kgs-V0
 kumoai/artifact_export/__init__.py,sha256=BsfDrc3mCHpO9-BqvqKm8qrXDIwfdaoH5UIoG4eQkc4,238
 kumoai/utils/datasets.py,sha256=ptKIUoBONVD55pTVNdRCkQT3NWdN_r9UAUu4xewPa3U,2928
 kumoai/utils/__init__.py,sha256=6S-UtwjeLpnCYRCCIEWhkitPYGaqOGXC1ChE13DzXiU,256
-kumoai/utils/progress_logger.py,sha256=3aYOoVSbQv5i9m2T8IqMydofKf6iNB1jxsl1uGjHZz8,9265
-kumoai/utils/sql.py,sha256=f6lR6rBEW7Dtk0NdM26dOZXUHDizEHb1WPlBCJrwoq0,118
+kumoai/utils/display.py,sha256=QmgeQQT7SzoC1CK2A0ftWbfkEuVN4KQfrKoPCrCDaGc,2626
+kumoai/utils/progress_logger.py,sha256=1PtXxfMteg2nyQAfTGx6qnljiZMZvhwDTndQ9_4_nCE,12161
+kumoai/utils/sql.py,sha256=CNKa-M56QiWoCSe9WLuumahsu3_ugQGr2YoTbveFHq0,147
 kumoai/utils/forecasting.py,sha256=-nDS6ucKNfQhTQOfebjefj0wwWH3-KYNslIomxwwMBM,7415
 kumoai/codegen/generate.py,sha256=SvfWWa71xSAOjH9645yQvgoEM-o4BYjupM_EpUxqB_E,7331
 kumoai/codegen/naming.py,sha256=_XVQGxHfuub4bhvyuBKjltD5Lm_oPpibvP_LZteCGk0,3021
@@ -76,7 +80,7 @@ kumoai/codegen/handlers/__init__.py,sha256=k8TB_Kn-1BycBBi51kqFS2fZHCpCPgR9-3J9g
 kumoai/codegen/handlers/utils.py,sha256=58b2GCgaTBUp2aId7BLMXMV0ENrusbNbfw7mlyXAXPE,1447
 kumoai/codegen/handlers/connector.py,sha256=afGf_GreyQ9y6qF3QTgSiM416qtUcP298SatNqUFhvQ,3828
 kumoai/codegen/handlers/table.py,sha256=POHpA-GFYFGTSuerGmtigYablk-Wq1L3EBvsOI-iFMQ,3956
-kumoai/testing/snow.py,sha256=ubx3yJP0UHxsNiar1-jNdv8ZfszKc8Js3_Gg70uf008,1487
+kumoai/testing/snow.py,sha256=S2ayiJ0WCZQdPKYiAKqT8OkQEw0xjYjOgDtGcjs3o7Q,1526
 kumoai/testing/__init__.py,sha256=goHIIo3JE7uHV7njo4_aTd89mVVR74BEAZ2uyBaOR0w,170
 kumoai/testing/decorators.py,sha256=83tMifuPTpUqX7zHxMttkj1TDdB62EBtAP-Fjj72Zdo,1607
 kumoai/connector/glue_connector.py,sha256=HivT0QYQ8-XeB4QLgWvghiqXuq7jyBK9G2R1py_NnE4,4697
@@ -86,20 +90,20 @@ kumoai/connector/bigquery_connector.py,sha256=IkyRqvF8Cg96kApUuuz86eYnl-BqBmDX1f
 kumoai/connector/source_table.py,sha256=QLT8bEYaxeMwy-b168url0VfnkTrs5K6VKLbxTI4hEY,17539
 kumoai/connector/__init__.py,sha256=9g6oNJ0qHWFlL5enTSoK4_SSH_5hP74xUDZx-9SggC4,842
 kumoai/connector/file_upload_connector.py,sha256=swp03HgChOvmNPJetuujBSAqADe7NRmS_T0F3o9it4w,7008
-kumoai/connector/utils.py,sha256=wlqQxMmPvnFNoCcczGkKYjSu05h8OhWh4fhTzQm_2bQ,64694
+kumoai/connector/utils.py,sha256=sD3_Dmf42FobMfVayzMVkDHIfXzPN-htD3RHd6Kw8hQ,65055
 kumoai/connector/s3_connector.py,sha256=3kbv-h7DwD8O260Q0h1GPm5wwQpLt-Tb3d_CBSaie44,10155
 kumoai/connector/base.py,sha256=cujXSZF3zAfuxNuEw54DSL1T7XCuR4t0shSMDuPUagQ,5291
 kumoai/pquery/__init__.py,sha256=uTXr7t1eXcVfM-ETaM_1ImfEqhrmaj8BjiIvy1YZTL8,533
 kumoai/pquery/predictive_query.py,sha256=UXn1s8ztubYZMNGl4ijaeidMiGlFveb1TGw9qI5-TAo,24901
 kumoai/pquery/prediction_table.py,sha256=QPDH22X1UB0NIufY7qGuV2XW7brG3Pv--FbjNezzM2g,10776
-kumoai/pquery/training_table.py,sha256=elmPDZx11kPiC_dkOhJcBUGtHKgL32GCBvZ9k6U0pMg,15809
+kumoai/pquery/training_table.py,sha256=QsZbqA1o-hFSi8GygtDQgYKFi8-3Ur2PftnpgAMqAec,16566
 kumoai/client/pquery.py,sha256=IQ8As-OOJOkuMoMosphOsA5hxQYLCbzOQJO7RezK8uY,7091
-kumoai/client/client.py,sha256=Jda8V9yiu3LbhxlcgRWPeYi7eF6jzCKcq8-B_vEd1ik,8514
+kumoai/client/client.py,sha256=npTLooBtmZ9xOo7AbEiYQTh9wFktsGSEpSEfdB7vdB4,8715
 kumoai/client/graph.py,sha256=zvLEDExLT_RVbUMHqVl0m6tO6s2gXmYSoWmPF6YMlnA,3831
 kumoai/client/online.py,sha256=pkBBh_DEC3GAnPcNw6bopNRlGe7EUbIFe7_seQqZRaw,2720
 kumoai/client/source_table.py,sha256=VCsCcM7KYcnjGP7HLTb-AOSEGEVsJTWjk8bMg1JdgPU,2101
 kumoai/client/__init__.py,sha256=MkyOuMaHQ2c8GPxjBDQSVFhfRE2d2_6CXQ6rxj4ps4w,64
-kumoai/client/jobs.py,sha256=iu_Wrta6BQMlV6ZtzSnmhjwNPKDMQDXOsqVVIyWodqw,17074
+kumoai/client/jobs.py,sha256=Aq-JO5yfU5BvD5_8ZXJ8NYxsE4yFXj_NdG9-ilymsr4,18164
 kumoai/client/utils.py,sha256=lz1NubwMDHCwzQRowRXm7mjAoYRd5UjRQIwXdtWAl90,3849
 kumoai/client/connector.py,sha256=x3i2aBTJTEMZvYRcWkY-UfWVOANZjqAso4GBbcshFjw,3920
 kumoai/client/table.py,sha256=cQG-RPm-e91idEgse1IPJDvBmzddIDGDkuyrR1rq4wU,3235
@@ -111,9 +115,10 @@ kumoai/trainer/job.py,sha256=Wk69nzFhbvuA3nEvtCstI04z5CxkgvQ6tHnGchE0Lkg,44938
 kumoai/trainer/baseline_trainer.py,sha256=LlfViNOmswNv4c6zJJLsyv0pC2mM2WKMGYx06ogtEVc,4024
 kumoai/trainer/__init__.py,sha256=zUdFl-f-sBWmm2x8R-rdVzPBeU2FaMzUY5mkcgoTa1k,939
 kumoai/trainer/online_serving.py,sha256=9cddb5paeZaCgbUeceQdAOxysCtV5XP-KcsgFz_XR5w,9566
+kumoai/trainer/distilled_trainer.py,sha256=2pPs5clakNxkLfaak7uqPJOrpTWe1RVVM7ztDSqQZvU,6484
 kumoai/trainer/trainer.py,sha256=hBXO7gwpo3t59zKFTeIkK65B8QRmWCwO33sbDuEAPlY,20133
-kumoai-2.14.0.dev202512211732.dist-info/RECORD,,
-kumoai-2.14.0.dev202512211732.dist-info/WHEEL,sha256=oqGJCpG61FZJmvyZ3C_0aCv-2mdfcY9e3fXvyUNmWfM,136
-kumoai-2.14.0.dev202512211732.dist-info/top_level.txt,sha256=YjU6UcmomoDx30vEXLsOU784ED7VztQOsFApk1SFwvs,7
-kumoai-2.14.0.dev202512211732.dist-info/METADATA,sha256=Vz_IsRTT5aua1Y4hwKx5cfyOucfGCb2VsE1ThYwSqI4,2557
-kumoai-2.14.0.dev202512211732.dist-info/licenses/LICENSE,sha256=TbWlyqRmhq9PEzCaTI0H0nWLQCCOywQM8wYH8MbjfLo,1102
+kumoai-2.15.0.dev202601181732.dist-info/RECORD,,
+kumoai-2.15.0.dev202601181732.dist-info/WHEEL,sha256=oqGJCpG61FZJmvyZ3C_0aCv-2mdfcY9e3fXvyUNmWfM,136
+kumoai-2.15.0.dev202601181732.dist-info/top_level.txt,sha256=YjU6UcmomoDx30vEXLsOU784ED7VztQOsFApk1SFwvs,7
+kumoai-2.15.0.dev202601181732.dist-info/METADATA,sha256=ETr8-9Zfq1pM_Smk8KOWZZm14cRdtNl9vcd_hqSQBKg,2564
+kumoai-2.15.0.dev202601181732.dist-info/licenses/LICENSE,sha256=TbWlyqRmhq9PEzCaTI0H0nWLQCCOywQM8wYH8MbjfLo,1102

kumoai/experimental/rfm/base/column_expression.py

@@ -1,50 +0,0 @@
-from dataclasses import dataclass
-from typing import Any, TypeAlias
-
-from kumoapi.typing import Dtype, Stype
-
-from kumoai.experimental.rfm.base import Column
-from kumoai.mixin import CastMixin
-
-
-@dataclass(frozen=True)
-class ColumnExpressionSpec(CastMixin):
-    name: str
-    expr: str
-    dtype: Dtype | None = None
-
-
-ColumnExpressionType: TypeAlias = ColumnExpressionSpec | dict[str, Any]
-
-
-@dataclass(init=False, repr=False, eq=False)
-class ColumnExpression(Column):
-    def __init__(
-        self,
-        name: str,
-        expr: str,
-        stype: Stype,
-        dtype: Dtype,
-    ) -> None:
-        super().__init__(name=name, stype=stype, dtype=dtype)
-        self._expr = expr
-
-    @property
-    def expr(self) -> str:
-        return self._expr
-
-    @property
-    def is_physical(self) -> bool:
-        return False
-
-    def __hash__(self) -> int:
-        return hash((self.name, self.expr, self.stype, self.dtype))
-
-    def __eq__(self, other: Any) -> bool:
-        if not isinstance(other, ColumnExpression):
-            return False
-        return hash(self) == hash(other)
-
-    def __repr__(self) -> str:
-        return (f'{self.__class__.__name__}(name={self.name}, '
-                f'expr={self.expr}, stype={self.stype}, dtype={self.dtype})')