rocket-welder-sdk 1.1.46__tar.gz → 1.2.0__tar.gz

{rocket_welder_sdk-1.1.46 → rocket_welder_sdk-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rocket-welder-sdk
-Version: 1.1.46
+Version: 1.2.0
 Summary: High-performance video streaming SDK for RocketWelder services using ZeroBuffer IPC
 Home-page: https://github.com/modelingevolution/rocket-welder-sdk
 Author: ModelingEvolution

rocket_welder_sdk-1.2.0/VERSION

@@ -0,0 +1 @@
+1.2.0

{rocket_welder_sdk-1.1.46 → rocket_welder_sdk-1.2.0}/rocket_welder_sdk/high_level/data_context.py

@@ -85,6 +85,7 @@ class ISegmentationDataContext(ABC):
         self,
         segment_class: SegmentClass,
         instance_id: int,
+        confidence: float,
         points: Union[Sequence[Point], npt.NDArray[np.int32]],
     ) -> None:
         """
@@ -93,6 +94,7 @@ class ISegmentationDataContext(ABC):
         Args:
             segment_class: SegmentClass from schema definition
             instance_id: Instance ID (for multiple instances of same class, 0-255)
+            confidence: Detection confidence score (0.0-1.0)
             points: Contour points defining the instance boundary
         """
         pass
@@ -150,6 +152,7 @@ class SegmentationDataContext(ISegmentationDataContext):
         self,
         segment_class: SegmentClass,
         instance_id: int,
+        confidence: float,
         points: Union[Sequence[Point], npt.NDArray[np.int32]],
     ) -> None:
         """Add a segmentation instance for this frame."""
@@ -162,7 +165,7 @@ class SegmentationDataContext(ISegmentationDataContext):
         else:
             points_array = np.array(points, dtype=np.int32)
 
-        self._writer.append(segment_class.class_id, instance_id, points_array)
+        self._writer.append(segment_class.class_id, instance_id, confidence, points_array)
 
     def commit(self) -> None:
         """Commit the context (called automatically when delegate returns)."""

{rocket_welder_sdk-1.1.46 → rocket_welder_sdk-1.2.0}/rocket_welder_sdk/rocket_welder_client.py

@@ -85,6 +85,42 @@ class RocketWelderClient:
         with self._lock:
             return self._controller is not None and self._controller.is_running
 
+    @property
+    def ml_model_path(self) -> Optional[str]:
+        """
+        Path to the ML model file mounted in the container.
+
+        Reads from ML_MODEL_PATH environment variable set by RocketWelder
+        when a model is mapped to this container's pipeline element.
+
+        Returns:
+            The path to the model file (e.g., /var/models/model.hef),
+            or None if no model is configured.
+        """
+        import os
+
+        return os.environ.get("ML_MODEL_PATH")
+
+    @property
+    def ml_model_version(self) -> Optional[int]:
+        """
+        Version of the ML model.
+
+        Reads from ML_MODEL_VERSION environment variable set by RocketWelder.
+
+        Returns:
+            The model version as integer, or None if not set or not a valid number.
+        """
+        import os
+
+        value = os.environ.get("ML_MODEL_VERSION")
+        if value is None:
+            return None
+        try:
+            return int(value)
+        except ValueError:
+            return None
+
     def get_metadata(self) -> Optional[GstMetadata]:
         """
         Get the current GStreamer metadata.
@@ -577,6 +613,47 @@ class RocketWelderClient:
         """Context manager exit."""
         self.stop()
 
+    @staticmethod
+    def get_ml_model_path() -> Optional[str]:
+        """
+        Get the ML model path from environment (static method).
+
+        Convenience method to get the model path without creating a client instance.
+        Useful during initialization before the client is created.
+
+        Returns:
+            The path to the model file (e.g., /var/models/model.hef),
+            or None if no model is configured.
+
+        Example:
+            model_path = RocketWelderClient.get_ml_model_path()
+            if model_path:
+                model = load_model(model_path)
+            else:
+                raise RuntimeError("No ML model configured")
+        """
+        import os
+
+        return os.environ.get("ML_MODEL_PATH")
+
+    @staticmethod
+    def get_ml_model_version() -> Optional[int]:
+        """
+        Get the ML model version from environment (static method).
+
+        Returns:
+            The model version as integer, or None if not set or not a valid number.
+        """
+        import os
+
+        value = os.environ.get("ML_MODEL_VERSION")
+        if value is None:
+            return None
+        try:
+            return int(value)
+        except ValueError:
+            return None
+
     @classmethod
     def from_connection_string(cls, connection_string: str) -> RocketWelderClient:
         """
@@ -750,7 +827,13 @@ class _NoOpKeyPointsWriter(IKeyPointsWriter):
 class _NoOpSegmentationWriter(ISegmentationResultWriter):
     """No-op segmentation writer that discards all data."""
 
-    def append(self, class_id: int, instance_id: int, points: Any) -> None:
+    def append(
+        self,
+        class_id: int,
+        instance_id: int,
+        confidence: float,
+        points: Any,
+    ) -> None:
         """Discard segmentation data."""
         pass
 

{rocket_welder_sdk-1.1.46 → rocket_welder_sdk-1.2.0}/rocket_welder_sdk/segmentation_result.py

@@ -6,8 +6,8 @@ Compatible with C# implementation for cross-platform interoperability.
 
 Protocol (per frame):
     [FrameId: 8B little-endian][Width: varint][Height: varint]
-    [classId: 1B][instanceId: 1B][pointCount: varint][points: delta+varint...]
-    [classId: 1B][instanceId: 1B][pointCount: varint][points: delta+varint...]
+    [classId: 1B][instanceId: 1B][confidence: 2B LE][pointCount: varint][points: delta+varint...]
+    [classId: 1B][instanceId: 1B][confidence: 2B LE][pointCount: varint][points: delta+varint...]
     ...
 
 Features:
@@ -104,6 +104,7 @@ class SegmentationInstance:
 
     class_id: int
     instance_id: int
+    confidence: float  # Detection confidence score (0.0-1.0)
     points: PointArray  # NumPy array of shape (N, 2) with dtype int32
 
     def to_normalized(self, width: int, height: int) -> npt.NDArray[np.float32]:
@@ -237,6 +238,7 @@ class SegmentationResultWriter:
         self,
         class_id: int,
         instance_id: int,
+        confidence: float,
         points: Union[List[Point], PointArray],
     ) -> None:
         """
@@ -245,6 +247,7 @@ class SegmentationResultWriter:
         Args:
             class_id: Object class ID (0-255)
             instance_id: Instance ID within class (0-255)
+            confidence: Detection confidence score (0.0-1.0)
             points: List of (x, y) tuples or NumPy array of shape (N, 2)
         """
         if class_id < 0 or class_id > 255:
@@ -266,6 +269,11 @@ class SegmentationResultWriter:
         # Write class_id and instance_id
         self._buffer.write(bytes([class_id, instance_id]))
 
+        # Write confidence as 2 bytes little-endian (ushort 0-65535)
+        confidence_clamped = max(0.0, min(1.0, confidence))
+        confidence_raw = int(confidence_clamped * 65535)
+        self._buffer.write(struct.pack("<H", confidence_raw))
+
         # Write point count
         point_count = len(points_array)
         _write_varint(self._buffer, point_count)
@@ -402,6 +410,13 @@ class SegmentationResultReader:
         class_id = header[0]
         instance_id = header[1]
 
+        # Read confidence (2 bytes little-endian)
+        confidence_bytes = self._stream.read(2)
+        if len(confidence_bytes) != 2:
+            raise EOFError("Unexpected end of stream reading confidence")
+        confidence_raw = struct.unpack("<H", confidence_bytes)[0]
+        confidence = confidence_raw / 65535.0
+
         # Read point count with validation
         point_count = _read_varint(self._stream)
         if point_count > self._max_points_per_instance:
@@ -412,7 +427,7 @@ class SegmentationResultReader:
         if point_count == 0:
             # Empty points array
             points = np.empty((0, 2), dtype=np.int32)
-            return SegmentationInstance(class_id, instance_id, points)
+            return SegmentationInstance(class_id, instance_id, confidence, points)
 
         # Allocate NumPy array for points
         points = np.empty((point_count, 2), dtype=np.int32)
@@ -430,7 +445,7 @@ class SegmentationResultReader:
             y += delta_y
             points[i] = [x, y]
 
-        return SegmentationInstance(class_id, instance_id, points)
+        return SegmentationInstance(class_id, instance_id, confidence, points)
 
     def read_all(self) -> List[SegmentationInstance]:
         """
@@ -472,6 +487,7 @@ class ISegmentationResultWriter(ABC):
         self,
         class_id: int,
         instance_id: int,
+        confidence: float,
         points: Union[List[Point], PointArray],
     ) -> None:
         """
@@ -480,6 +496,7 @@ class ISegmentationResultWriter(ABC):
         Args:
             class_id: Object class ID (0-255)
             instance_id: Instance ID within class (0-255)
+            confidence: Detection confidence score (0.0-1.0)
             points: List of (x, y) tuples or NumPy array of shape (N, 2)
         """
         pass
@@ -741,6 +758,7 @@ class SegmentationProtocol:
     Instance Format:
         [ClassId: 1 byte]
         [InstanceId: 1 byte]
+        [Confidence: 2 bytes, little-endian uint16]
         [PointCount: varint]
         [Point0: X zigzag-varint, Y zigzag-varint]  (absolute)
         [Point1+: deltaX zigzag-varint, deltaY zigzag-varint]
@@ -777,6 +795,12 @@ class SegmentationProtocol:
     def _write_instance_core(stream: BinaryIO, instance: SegmentationInstance) -> None:
         """Write a single instance to the stream."""
         stream.write(bytes([instance.class_id, instance.instance_id]))
+
+        # Write confidence as 2 bytes little-endian (ushort 0-65535)
+        confidence_clamped = max(0.0, min(1.0, instance.confidence))
+        confidence_raw = int(confidence_clamped * 65535)
+        stream.write(struct.pack("<H", confidence_raw))
+
         point_count = len(instance.points)
         _write_varint(stream, point_count)
 
@@ -823,6 +847,7 @@ class SegmentationProtocol:
         buffer: bytearray,
         class_id: int,
         instance_id: int,
+        confidence: float,
         points: Union[List[Point], PointArray],
     ) -> int:
         """
@@ -834,6 +859,7 @@ class SegmentationProtocol:
             buffer: Pre-allocated buffer to write to.
             class_id: Class identifier (0-255).
             instance_id: Instance identifier (0-255).
+            confidence: Detection confidence score (0.0-1.0).
             points: Polygon points.
 
         Returns:
@@ -845,7 +871,7 @@ class SegmentationProtocol:
         else:
             points_array = points.astype(np.int32)
 
-        instance = SegmentationInstance(class_id, instance_id, points_array)
+        instance = SegmentationInstance(class_id, instance_id, confidence, points_array)
 
         stream = io.BytesIO()
         SegmentationProtocol._write_instance_core(stream, instance)
@@ -864,8 +890,8 @@ class SegmentationProtocol:
         Returns:
             Maximum bytes needed.
         """
-        # classId(1) + instanceId(1) + pointCount(varint, max 5) + points(max 10 bytes each)
-        return 1 + 1 + 5 + (point_count * 10)
+        # classId(1) + instanceId(1) + confidence(2) + pointCount(varint, max 5) + points(max 10 bytes each)
+        return 1 + 1 + 2 + 5 + (point_count * 10)
 
     @staticmethod
     def read(data: bytes) -> SegmentationFrame:
@@ -899,6 +925,14 @@ class SegmentationProtocol:
 
             class_id = header[0]
             instance_id = header[1]
+
+            # Read confidence (2 bytes little-endian)
+            confidence_bytes = stream.read(2)
+            if len(confidence_bytes) != 2:
+                raise EOFError("Unexpected end of stream reading confidence")
+            confidence_raw = struct.unpack("<H", confidence_bytes)[0]
+            confidence = confidence_raw / 65535.0
+
             point_count = _read_varint(stream)
 
             if point_count == 0:
@@ -916,7 +950,7 @@ class SegmentationProtocol:
                     points[i] = [x, y]
                     prev_x, prev_y = x, y
 
-            instances.append(SegmentationInstance(class_id, instance_id, points))
+            instances.append(SegmentationInstance(class_id, instance_id, confidence, points))
 
         return SegmentationFrame(frame_id, width, height, instances)
 

{rocket_welder_sdk-1.1.46 → rocket_welder_sdk-1.2.0}/rocket_welder_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rocket-welder-sdk
-Version: 1.1.46
+Version: 1.2.0
 Summary: High-performance video streaming SDK for RocketWelder services using ZeroBuffer IPC
 Home-page: https://github.com/modelingevolution/rocket-welder-sdk
 Author: ModelingEvolution

{rocket_welder_sdk-1.1.46 → rocket_welder_sdk-1.2.0}/tests/test_segmentation_cross_platform.py

@@ -96,7 +96,7 @@ class TestCrossPlatform:
             frame_id, width, height, f
         ) as writer:
             for class_id, instance_id, points in instances:
-                writer.append(class_id, instance_id, points)
+                writer.append(class_id, instance_id, 0.95, points)
 
         # Verify file exists and has data
         assert test_file.exists()
@@ -129,7 +129,7 @@ class TestCrossPlatform:
             frame_id, width, height, f
         ) as writer:
             for class_id, instance_id, points in instances:
-                writer.append(class_id, instance_id, points)
+                writer.append(class_id, instance_id, 0.95, points)
 
         # Act - Read (via transport layer for framing)
         with open(test_file, "rb") as f:

{rocket_welder_sdk-1.1.46 → rocket_welder_sdk-1.2.0}/tests/test_segmentation_result.py

@@ -45,7 +45,7 @@ class TestRoundTrip:
 
         # Act - Write
         with SegmentationResultWriter(frame_id, width, height, stream) as writer:
-            writer.append(class_id, instance_id, points)
+            writer.append(class_id, instance_id, 0.95, points)
 
         # Act - Read via transport layer
         with _read_frame_via_transport(stream) as reader:
@@ -82,7 +82,7 @@ class TestRoundTrip:
         # Act - Write
         with SegmentationResultWriter(frame_id, width, height, stream) as writer:
             for class_id, instance_id, points in instances:
-                writer.append(class_id, instance_id, points)
+                writer.append(class_id, instance_id, 0.95, points)
 
         # Act - Read via transport layer
         # Via transport layer
@@ -104,7 +104,7 @@ class TestRoundTrip:
         stream = io.BytesIO()
 
         with SegmentationResultWriter(1, 100, 100, stream) as writer:
-            writer.append(1, 1, np.empty((0, 2), dtype=np.int32))
+            writer.append(1, 1, 0.95, np.empty((0, 2), dtype=np.int32))
 
         # Via transport layer
         with _read_frame_via_transport(stream) as reader:
@@ -128,7 +128,7 @@ class TestRoundTrip:
         stream = io.BytesIO()
 
         with SegmentationResultWriter(999, 3840, 2160, stream) as writer:
-            writer.append(10, 5, points)
+            writer.append(10, 5, 0.95, points)
 
         # Via transport layer
         with _read_frame_via_transport(stream) as reader:
@@ -153,7 +153,7 @@ class TestRoundTrip:
         stream = io.BytesIO()
 
         with SegmentationResultWriter(1, 200, 200, stream) as writer:
-            writer.append(1, 1, points)
+            writer.append(1, 1, 0.95, points)
 
         # Via transport layer
         with _read_frame_via_transport(stream) as reader:
@@ -174,7 +174,7 @@ class TestRoundTrip:
             1, 640, 480, frame_sink=StreamFrameSink(stream, leave_open=True)
         ) as writer:
             for class_id, instance_id, points in frame1_points:
-                writer.append(class_id, instance_id, points)
+                writer.append(class_id, instance_id, 0.95, points)
 
         # Frame 2
         frame2_points = [
@@ -186,7 +186,7 @@ class TestRoundTrip:
             2, 1920, 1080, frame_sink=StreamFrameSink(stream, leave_open=True)
         ) as writer:
             for class_id, instance_id, points in frame2_points:
-                writer.append(class_id, instance_id, points)
+                writer.append(class_id, instance_id, 0.95, points)
 
         # Read both frames via transport layer
         stream.seek(0)
@@ -235,7 +235,7 @@ class TestNormalization:
     def test_to_normalized_converts_to_float_range(self) -> None:
         """Test normalization to [0-1] range."""
         points = np.array([[0, 0], [1920, 1080], [960, 540]], dtype=np.int32)
-        instance = SegmentationInstance(1, 1, points)
+        instance = SegmentationInstance(1, 1, 0.95, points)
 
         normalized = instance.to_normalized(1920, 1080)
 
@@ -247,7 +247,7 @@ class TestNormalization:
     def test_to_normalized_raises_on_zero_dimensions(self) -> None:
         """Test that normalization raises on zero width/height."""
         points = np.array([[10, 20]], dtype=np.int32)
-        instance = SegmentationInstance(1, 1, points)
+        instance = SegmentationInstance(1, 1, 0.95, points)
 
         with pytest.raises(ValueError, match="must be positive"):
             instance.to_normalized(0, 1080)
@@ -271,7 +271,7 @@ class TestIterator:
 
         with SegmentationResultWriter(1, 100, 100, stream) as writer:
             for class_id, instance_id, points in instances_data:
-                writer.append(class_id, instance_id, points)
+                writer.append(class_id, instance_id, 0.95, points)
 
         # Via transport layer
         with _read_frame_via_transport(stream) as reader:
@@ -287,8 +287,8 @@ class TestIterator:
         stream = io.BytesIO()
 
         with SegmentationResultWriter(1, 100, 100, stream) as writer:
-            writer.append(1, 1, np.array([[10, 20]], dtype=np.int32))
-            writer.append(2, 1, np.array([[30, 40]], dtype=np.int32))
+            writer.append(1, 1, 0.95, np.array([[10, 20]], dtype=np.int32))
+            writer.append(2, 1, 0.95, np.array([[30, 40]], dtype=np.int32))
 
         # Via transport layer
         with _read_frame_via_transport(stream) as reader:
@@ -306,14 +306,14 @@ class TestFlush:
         stream = io.BytesIO()
         writer = SegmentationResultWriter(1, 100, 100, stream)
 
-        writer.append(1, 1, np.array([[10, 20]], dtype=np.int32))
+        writer.append(1, 1, 0.95, np.array([[10, 20]], dtype=np.int32))
         writer.flush()
 
         # Should have data
         assert stream.tell() > 0
 
         # Can still write more
-        writer.append(2, 1, np.array([[30, 40]], dtype=np.int32))
+        writer.append(2, 1, 0.95, np.array([[30, 40]], dtype=np.int32))
         writer.close()
 
 
@@ -328,9 +328,9 @@ class TestValidation:
         points = np.array([[10, 20]], dtype=np.int32)
 
         # 255 is now valid (no end-marker)
-        writer.append(255, 1, points)
-        writer.append(1, 255, points)
-        writer.append(255, 255, points)
+        writer.append(255, 1, 0.95, points)
+        writer.append(1, 255, 0.95, points)
+        writer.append(255, 255, 0.95, points)
         writer.close()
 
         # Read back and verify via transport layer
@@ -361,6 +361,7 @@ class TestValidation:
 
         # Write instance with huge point count
         stream.write(bytes([1, 1]))  # class_id, instance_id
+        stream.write(struct.pack("<H", 32767))  # confidence = 0.5 (2 bytes LE)
         # Write varint for > 10M points (will fail validation)
         # 20M = 0x1312D00
         stream.write(b"\x80\xba\xc8\x89\x01")  # varint encoding of 20000000
@@ -379,7 +380,7 @@ class TestListConversion:
     def test_to_list_converts_numpy_to_tuples(self) -> None:
         """Test conversion from NumPy array to list of tuples."""
         points = np.array([[10, 20], [30, 40]], dtype=np.int32)
-        instance = SegmentationInstance(1, 1, points)
+        instance = SegmentationInstance(1, 1, 0.95, points)
 
         points_list = instance.to_list()
 
@@ -396,7 +397,7 @@ class TestListInput:
         points_list: List[Tuple[int, int]] = [(10, 20), (30, 40), (50, 60)]
 
         with SegmentationResultWriter(1, 100, 100, stream) as writer:
-            writer.append(1, 1, points_list)
+            writer.append(1, 1, 0.95, points_list)
 
         # Via transport layer
         with _read_frame_via_transport(stream) as reader:
@@ -439,8 +440,8 @@ class TestSegmentationFrame:
     def test_frame_properties(self) -> None:
         """Test basic frame properties."""
         instances = [
-            SegmentationInstance(1, 1, np.array([[10, 20]], dtype=np.int32)),
-            SegmentationInstance(2, 1, np.array([[30, 40], [50, 60]], dtype=np.int32)),
+            SegmentationInstance(1, 1, 0.95, np.array([[10, 20]], dtype=np.int32)),
+            SegmentationInstance(2, 1, 0.95, np.array([[30, 40], [50, 60]], dtype=np.int32)),
         ]
         frame = SegmentationFrame(frame_id=42, width=1920, height=1080, instances=instances)
 
@@ -452,9 +453,9 @@ class TestSegmentationFrame:
     def test_find_by_class(self) -> None:
         """Test finding instances by class ID."""
         instances = [
-            SegmentationInstance(1, 1, np.array([[10, 20]], dtype=np.int32)),
-            SegmentationInstance(2, 1, np.array([[30, 40]], dtype=np.int32)),
-            SegmentationInstance(1, 2, np.array([[50, 60]], dtype=np.int32)),
+            SegmentationInstance(1, 1, 0.95, np.array([[10, 20]], dtype=np.int32)),
+            SegmentationInstance(2, 1, 0.95, np.array([[30, 40]], dtype=np.int32)),
+            SegmentationInstance(1, 2, 0.95, np.array([[50, 60]], dtype=np.int32)),
         ]
         frame = SegmentationFrame(frame_id=1, width=100, height=100, instances=instances)
 
@@ -472,9 +473,9 @@ class TestSegmentationFrame:
     def test_find_by_instance(self) -> None:
         """Test finding instances by instance ID."""
         instances = [
-            SegmentationInstance(1, 1, np.array([[10, 20]], dtype=np.int32)),
-            SegmentationInstance(2, 1, np.array([[30, 40]], dtype=np.int32)),
-            SegmentationInstance(1, 2, np.array([[50, 60]], dtype=np.int32)),
+            SegmentationInstance(1, 1, 0.95, np.array([[10, 20]], dtype=np.int32)),
+            SegmentationInstance(2, 1, 0.95, np.array([[30, 40]], dtype=np.int32)),
+            SegmentationInstance(1, 2, 0.95, np.array([[50, 60]], dtype=np.int32)),
         ]
         frame = SegmentationFrame(frame_id=1, width=100, height=100, instances=instances)
 
@@ -491,8 +492,8 @@ class TestSegmentationProtocol:
     def test_write_read_roundtrip(self) -> None:
         """Test write/read roundtrip through protocol."""
         instances = [
-            SegmentationInstance(1, 1, np.array([[10, 20], [30, 40]], dtype=np.int32)),
-            SegmentationInstance(2, 1, np.array([[100, 200], [101, 201]], dtype=np.int32)),
+            SegmentationInstance(1, 1, 0.95, np.array([[10, 20], [30, 40]], dtype=np.int32)),
+            SegmentationInstance(2, 1, 0.95, np.array([[100, 200], [101, 201]], dtype=np.int32)),
         ]
         original = SegmentationFrame(frame_id=42, width=1920, height=1080, instances=instances)
 
@@ -528,20 +529,20 @@ class TestSegmentationProtocol:
         """Test writing a single instance."""
         points: List[Tuple[int, int]] = [(10, 20), (30, 40)]
         buffer = bytearray(100)
-        written = SegmentationProtocol.write_instance(buffer, 5, 3, points)
+        written = SegmentationProtocol.write_instance(buffer, 5, 3, 0.95, points)
 
-        # Should have classId(1) + instanceId(1) + pointCount varint + points
-        assert written > 4
+        # Should have classId(1) + instanceId(1) + confidence(2) + pointCount varint + points
+        assert written > 6
 
     def test_calculate_instance_size(self) -> None:
         """Test size calculation."""
         size = SegmentationProtocol.calculate_instance_size(10)
-        # classId(1) + instanceId(1) + pointCount(max 5) + points(max 10 each)
-        assert size == 1 + 1 + 5 + (10 * 10)
+        # classId(1) + instanceId(1) + confidence(2) + pointCount(max 5) + points(max 10 each)
+        assert size == 1 + 1 + 2 + 5 + (10 * 10)
 
     def test_try_read_success(self) -> None:
         """Test try_read with valid data."""
-        instances = [SegmentationInstance(1, 1, np.array([[10, 20]], dtype=np.int32))]
+        instances = [SegmentationInstance(1, 1, 0.95, np.array([[10, 20]], dtype=np.int32))]
         original = SegmentationFrame(frame_id=1, width=100, height=100, instances=instances)
 
         buffer = bytearray(100)
@@ -571,7 +572,7 @@ class TestSegmentationProtocol:
 
     def test_empty_points(self) -> None:
         """Test roundtrip with instance having empty points."""
-        instances = [SegmentationInstance(1, 1, np.empty((0, 2), dtype=np.int32))]
+        instances = [SegmentationInstance(1, 1, 0.95, np.empty((0, 2), dtype=np.int32))]
         original = SegmentationFrame(frame_id=1, width=100, height=100, instances=instances)
 
         buffer = bytearray(100)
@@ -592,8 +593,8 @@ class TestSegmentationResultSource:
         with SegmentationResultWriter(
             42, 1920, 1080, frame_sink=StreamFrameSink(stream, leave_open=True)
         ) as writer:
-            writer.append(1, 1, np.array([[10, 20], [30, 40]], dtype=np.int32))
-            writer.append(2, 1, np.array([[100, 200]], dtype=np.int32))
+            writer.append(1, 1, 0.95, np.array([[10, 20], [30, 40]], dtype=np.int32))
+            writer.append(2, 1, 0.95, np.array([[100, 200]], dtype=np.int32))
 
         # Read via source
         stream.seek(0)
@@ -618,12 +619,12 @@ class TestSegmentationResultSource:
 
         # Write frame 1
         with SegmentationResultWriter(1, 640, 480, frame_sink=sink) as writer:
-            writer.append(1, 1, np.array([[10, 20]], dtype=np.int32))
+            writer.append(1, 1, 0.95, np.array([[10, 20]], dtype=np.int32))
 
         # Write frame 2
         with SegmentationResultWriter(2, 1920, 1080, frame_sink=sink) as writer:
-            writer.append(2, 1, np.array([[100, 200]], dtype=np.int32))
-            writer.append(3, 1, np.array([[300, 400]], dtype=np.int32))
+            writer.append(2, 1, 0.95, np.array([[100, 200]], dtype=np.int32))
+            writer.append(3, 1, 0.95, np.array([[300, 400]], dtype=np.int32))
 
         # Read via source
         stream.seek(0)
@@ -645,7 +646,7 @@ class TestSegmentationResultSource:
         with SegmentationResultWriter(
             1, 100, 100, frame_sink=StreamFrameSink(stream, leave_open=True)
         ) as writer:
-            writer.append(1, 1, np.array([[10, 20]], dtype=np.int32))
+            writer.append(1, 1, 0.95, np.array([[10, 20]], dtype=np.int32))
 
         stream.seek(0)
         frame_source = StreamFrameSource(stream)

{rocket_welder_sdk-1.1.46 → rocket_welder_sdk-1.2.0}/tests/test_transport_cross_platform.py

@@ -167,7 +167,7 @@ class TestUnixSocketTransportRoundTrip:
                 frame_id=42, width=1920, height=1080, stream=buffer
             ) as writer:
                 points = np.array([[100, 200], [101, 201], [102, 199]], dtype=np.int32)
-                writer.append(class_id=1, instance_id=1, points=points)
+                writer.append(class_id=1, instance_id=1, confidence=0.95, points=points)
 
             # Get frame data (with varint prefix)
             buffer.seek(0)

rocket_welder_sdk-1.1.46/VERSION

@@ -1 +0,0 @@
-1.1.46