Cleanup stub

Author: abey79

rerun_py/rerun_bindings/rerun_bindings.pyi

@@ -180,6 +180,24 @@ class Schema:
 
         """
 
+    def column_for_selector(self, selector: str | ComponentColumnSelector | ComponentColumnDescriptor) -> ComponentColumnDescriptor:
+        """
+        Look up the column descriptor for a specific selector.
+
+        Parameters
+        ----------
+        selector: str | ComponentColumnDescriptor | ComponentColumnSelector
+           The selector to look up.
+
+           String arguments are expected to follow the following format:
+           `"<entity_path>:<component_name>"`
+
+        Returns
+        -------
+        ComponentColumnDescriptor
+            The column descriptor, if it exists. Raise an exception otherwise.
+        """
+
 class RecordingView:
     """
     A view of a recording restricted to a given index, containing a specific set of entities and components.
@@ -1037,7 +1055,7 @@ class Dataset(Entry):
     def create_fts_index(
         self,
         *,
-        column: ComponentColumnSelector,
+        column: str | ComponentColumnSelector | ComponentColumnDescriptor,
         time_index: IndexColumnSelector,
         store_position: bool = False,
         base_tokenizer: str = "simple",
@@ -1047,7 +1065,7 @@ class Dataset(Entry):
     def create_vector_index(
         self,
         *,
-        column: ComponentColumnSelector,
+        column: str | ComponentColumnSelector | ComponentColumnDescriptor,
         time_index: IndexColumnSelector,
         num_partitions: int = 5,
         num_sub_vectors: int = 16,
@@ -1058,14 +1076,14 @@ class Dataset(Entry):
     def search_fts(
         self,
         query: str,
-        column: ComponentColumnSelector,
+        column: str | ComponentColumnSelector | ComponentColumnDescriptor,
     ) -> DataFusionTable:
         """Search the dataset using a full-text search query."""
 
     def search_vector(
         self,
         query: Any,  # VectorLike
-        column: ComponentColumnSelector,
+        column: str | ComponentColumnSelector | ComponentColumnDescriptor,
         top_k: int,
     ) -> DataFusionTable:
         """Search the dataset using a vector search query."""

rerun_py/src/dataframe.rs

@@ -567,16 +567,21 @@ impl PySchema {
     ///
     /// Parameters
     /// ----------
-    /// component_column_selector: str | ComponentColumnDescriptor | ComponentColumnSelector
+    /// selector: str | ComponentColumnDescriptor | ComponentColumnSelector
     ///    The selector to look up.
     ///
     ///    String arguments are expected to follow the following format:
     ///    `"<entity_path>:<component_name>"`
+    ///
+    /// Returns
+    /// -------
+    /// ComponentColumnDescriptor
+    ///     The column descriptor, if it exists. Raise an exception otherwise.
     pub fn column_for_selector(
         &self,
-        component_column_selector: AnyComponentColumn,
+        selector: AnyComponentColumn,
     ) -> PyResult<PyComponentColumnDescriptor> {
-        match component_column_selector {
+        match selector {
             AnyComponentColumn::Name(name) => self.resolve_component_column_selector(
                 &ComponentColumnSelector::from_str(&name).map_err(to_py_err)?,
             ),