Update doc & fix lints

Author: abey79

docs/content/howto/logging/send-columns.md

@@ -26,7 +26,18 @@ snippet: archetypes/image_send_columns
 
 
 ### Using `send_columns` for logging points
-Each row the in the component column can be a batch of data, e.g. a batch of positions.
+Each row in the component column can be a batch of data, e.g. a batch of positions.
 This lets you log the evolution of a point cloud over time efficiently.
 
 snippet: archetypes/points3d_send_columns.py
+
+### Using `send_columns` for logging custom components
+
+An entire batch of a custom component can be logged at once using [`rr.AnyBatchValue`](https://ref.rerun.io/docs/python/0.20.0/common/custom_data/#rerun.AnyBatchValue?speculative-link) along with `send_column`:
+
+snippet: howto/any_batch_value_send_columns
+
+The [`rr.AnyValues`](https://ref.rerun.io/docs/python/0.20.0/common/custom_data/#rerun.AnyValues) class can also be used to log multiple components at a time.
+It does not support partitioning, so each component batch and the timeline must hold the same number of elements.
+
+snippet: howto/any_values_send_columns

docs/snippets/all/howto/any_batch_value_send_columns.py

@@ -0,0 +1,24 @@
+"""Use `AnyBatchValue` and `send_column` to send an entire column of custom data to Rerun."""
+
+from __future__ import annotations
+
+import numpy as np
+import rerun as rr
+
+rr.init("rerun_example_any_batch_value_send_columns", spawn=True)
+
+N = 64
+timestamps = np.arange(0, N)
+one_per_timestamp = np.sin(timestamps / 10.0)
+ten_per_timestamp = np.cos(np.arange(0, N * 10) / 100.0)
+
+rr.send_columns(
+    "/",
+    times=[rr.TimeSequenceColumn("step", timestamps)],
+    components=[
+        # log one value per timestamp
+        rr.AnyBatchValue("custom_component_single", one_per_timestamp),
+        # log ten values per timestamp
+        rr.AnyBatchValue("custom_component_multi", ten_per_timestamp).partition([10] * N),
+    ],
+)

docs/snippets/all/howto/any_values_send_columns.py

@@ -0,0 +1,22 @@
+"""Use `AnyValues` and `send_column` to send entire columns of custom data to Rerun."""
+
+from __future__ import annotations
+
+import numpy as np
+import rerun as rr
+
+rr.init("rerun_example_any_values_send_columns", spawn=True)
+
+timestamps = np.arange(0, 64)
+
+# Log two components, named "sin" and "cos", with the corresponding values
+values = rr.AnyValues(
+    sin=np.sin(timestamps / 10.0),
+    cos=np.cos(timestamps / 10.0),
+)
+
+rr.send_columns(
+    "/",
+    times=[rr.TimeSequenceColumn("step", timestamps)],
+    components=values.as_component_batches(),
+)

docs/snippets/snippets.toml

@@ -62,6 +62,14 @@ views = [
 "archetypes/points3d_send_columns" = [
   "rust", # Doesn't support partitioned component batches yet.
 ]
+"howto/any_batch_value_send_columns" = [
+  "cpp",  # Not implemented
+  "rust", # Not implemented
+]
+"howto/any_values_send_columns" = [
+  "cpp",  # Not implemented
+  "rust", # Not implemented
+]
 "migration/log_line" = [ # Not a complete example -- just a single log line
   "cpp",
   "rust",

rerun_py/docs/gen_common_index.py

@@ -145,7 +145,10 @@ class Section:
     ),
     Section(
         title="Custom Data",
-        class_list=["AnyValues"],
+        class_list=[
+            "AnyValues",
+            "AnyBatchValues",
+        ],
     ),
     ################################################################################
     # These are tables but don't need their own pages since they refer to types that

rerun_py/rerun_sdk/rerun/__init__.py

@@ -57,6 +57,7 @@
     send_columns as send_columns,
 )
 from .any_value import (
+    AnyBatchValue as AnyBatchValue,
     AnyValues as AnyValues,
 )
 from .archetypes import (

rerun_py/rerun_sdk/rerun/_send_columns.py

@@ -232,7 +232,7 @@ def send_columns(
             array = c.as_arrow_array()
             if array is None:
                 raise ValueError(f"Expected a non-null value for component: {component_name}")
-            component_column = c.partition([1] * len(c.as_arrow_array()))
+            component_column = c.partition([1] * len(c.as_arrow_array()))  # type: ignore[arg-type]
         else:
             raise TypeError(
                 f"Expected either a type that implements the `ComponentMixin` or a `ComponentColumn`, got: {type(c)}"

rerun_py/rerun_sdk/rerun/any_value.py

@@ -186,5 +186,5 @@ def __init__(self, drop_untyped_nones: bool = True, **kwargs: Any) -> None:
                 if batch.is_valid():
                     self.component_batches.append(batch)
 
-    def as_component_batches(self) -> Iterable[AnyBatchValue]:
+    def as_component_batches(self) -> Iterable[ComponentBatchLike]:
         return self.component_batches