Better organization of types so that they are importable

rerun_py/rerun_bindings/__init__.py

@@ -1,3 +1,5 @@
 from __future__ import annotations
 
+from typing import TYPE_CHECKING, TypeAlias, Union
+
 from .rerun_bindings import *

rerun_py/rerun_bindings/rerun_bindings.pyi

@@ -1,23 +1,8 @@
-from typing import Optional, Sequence, Union
+from typing import Optional, Sequence
 
 import pyarrow as pa
-from rerun._baseclasses import ComponentMixin
 
-AnyColumn = Union[
-    ControlColumnDescriptor,
-    TimeColumnDescriptor,
-    ComponentColumnDescriptor,
-    ControlColumnSelector,
-    TimeColumnSelector,
-    ComponentColumnSelector,
-]
-
-AnyComponentColumn = Union[
-    ComponentColumnDescriptor,
-    ComponentColumnSelector,
-]
-
-ComponentLike = Union[str, type[ComponentMixin]]
+from .types import AnyColumn, AnyComponentColumn, ComponentLike
 
 class ControlColumnDescriptor:
     """A control-level column such as `RowId`."""
@@ -44,7 +29,7 @@ class ComponentColumnDescriptor:
 class ComponentColumnSelector:
     """A selector for a component column."""
 
-    def __init__(self, entity_path: str, component_type: ComponentLike): ...
+    def __new__(cls, entity_path: str, component_type: ComponentLike): ...
     def with_dictionary_encoding(self) -> ComponentColumnSelector: ...
 
 class Schema:
@@ -53,7 +38,7 @@ class Schema:
     def control_columns(self) -> list[ControlColumnDescriptor]: ...
     def time_columns(self) -> list[TimeColumnDescriptor]: ...
     def component_columns(self) -> list[ComponentColumnDescriptor]: ...
-    def column_for(self, entity_path: str, ComponentLike) -> Optional[ComponentColumnDescriptor]: ...
+    def column_for(self, entity_path: str, component: ComponentLike) -> Optional[ComponentColumnDescriptor]: ...
 
 class TimeRange:
     """A time range with a start and end time."""

rerun_py/rerun_bindings/types.py

@@ -0,0 +1,32 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, TypeAlias, Union
+
+if TYPE_CHECKING:
+    from rerun._baseclasses import ComponentMixin
+
+    from .rerun_bindings import (
+        ComponentColumnDescriptor as ComponentColumnDescriptor,
+        ComponentColumnSelector as ComponentColumnSelector,
+        ControlColumnDescriptor as ControlColumnDescriptor,
+        ControlColumnSelector as ControlColumnSelector,
+        TimeColumnDescriptor as TimeColumnDescriptor,
+        TimeColumnSelector as TimeColumnSelector,
+    )
+
+
+ComponentLike: TypeAlias = Union[str, type["ComponentMixin"]]
+
+AnyColumn: TypeAlias = Union[
+    "ControlColumnDescriptor",
+    "TimeColumnDescriptor",
+    "ComponentColumnDescriptor",
+    "ControlColumnSelector",
+    "TimeColumnSelector",
+    "ComponentColumnSelector",
+]
+
+AnyComponentColumn: TypeAlias = Union[
+    "ComponentColumnDescriptor",
+    "ComponentColumnSelector",
+]

rerun_py/rerun_sdk/rerun/dataframe.py

@@ -14,3 +14,8 @@
     load_archive as load_archive,
     load_recording as load_recording,
 )
+from rerun_bindings.types import (
+    AnyColumn as AnyColumn,
+    AnyComponentColumn as AnyComponentColumn,
+    ComponentLike as ComponentLike,
+)