fix missing marker size serialization

rerun-io · Feb 6, 2024 · c0a0026 · c0a0026
1 parent f4faefe
commit c0a0026
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/rerun_py/rerun_sdk/rerun/components/marker_size.py b/rerun_py/rerun_sdk/rerun/components/marker_size.py
diff --git a/rerun_py/rerun_sdk/rerun/components/marker_size_ext.py b/rerun_py/rerun_sdk/rerun/components/marker_size_ext.py
@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import numpy as np
+import pyarrow as pa
+
+if TYPE_CHECKING:
+    from . import MarkerSizeArrayLike
+
+
+class MarkerSizeExt:
+    """Extension for [MarkerSize][rerun.components.MarkerSize]."""
+
+    @staticmethod
+    def native_to_pa_array_override(data: MarkerSizeArrayLike, data_type: pa.DataType) -> pa.Array:
+        array = np.asarray(data, dtype=np.float32).flatten()
+        return pa.array(array, type=data_type)