Skip to content

Commit 73a6150

Browse files
committed
Switch order of kwargs to correspond between vision service and camera service
1 parent 0e4fd50 commit 73a6150

File tree

5 files changed

+5
-21
lines changed

5 files changed

+5
-21
lines changed

src/viam/components/camera/service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ async def GetImages(self, stream: Stream[GetImagesRequest, GetImagesResponse]) -
6262
fmt = mime_type.to_proto() # Will be Format.FORMAT_UNSPECIFIED if an unsupported/custom mime type is set
6363

6464
img_bytes = img.data
65-
img_bytes_lst.append(Image(source_name=name, format=fmt, mime_type=img.mime_type, image=img_bytes))
65+
img_bytes_lst.append(Image(source_name=name, mime_type=img.mime_type, format=fmt, image=img_bytes))
6666
response = GetImagesResponse(images=img_bytes_lst, response_metadata=metadata)
6767
await stream.send_message(response)
6868

src/viam/services/vision/service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
from viam.media.video import CameraMimeType, ViamImage
44
from viam.proto.common import DoCommandRequest, DoCommandResponse
5-
from viam.proto.component.camera import Format, Image
5+
from viam.proto.component.camera import Image
66
from viam.proto.service.vision import (
77
CaptureAllFromCameraRequest,
88
CaptureAllFromCameraResponse,

tests/test_camera.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
Format,
1515
CameraServiceStub,
1616
DistortionParameters,
17-
Format,
1817
GetImageRequest,
1918
GetImageResponse,
2019
GetImagesRequest,

tests/test_media.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,6 @@
99
from viam.media.utils.pil import pil_to_viam_image, viam_to_pil_image
1010
from viam.media.video import CameraMimeType, Format, NamedImage, ViamImage
1111

12-
UNSUPPORTED_MIME_TYPE = "unsupported_string_mime_type"
13-
1412

1513
class TestViamImage:
1614
UNSUPPORTED_MIME_TYPE = "unsupported_string_mime_type"
@@ -173,8 +171,5 @@ def test_image_conversion():
173171
v_img2 = pil_to_viam_image(pil_img, CameraMimeType.JPEG)
174172
assert v_img2.data == v_img.data
175173

176-
with pytest.raises(
177-
ValueError,
178-
match=f"Cannot encode to unsupported mimetype: {TestViamImage.UNSUPPORTED_MIME_TYPE}",
179-
):
174+
with pytest.raises(ValueError, match=f"Cannot encode to unsupported mimetype: {TestViamImage.UNSUPPORTED_MIME_TYPE}"):
180175
pil_to_viam_image(i, CameraMimeType.CUSTOM(TestViamImage.UNSUPPORTED_MIME_TYPE))

tests/test_vision_service.py

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -84,12 +84,7 @@
8484
reference_frame="depth1",
8585
geometries=[
8686
Geometry(
87-
center=Pose(
88-
x=282.45238095238096,
89-
y=241.66666666666666,
90-
z=902.8809523809524,
91-
o_z=1.0,
92-
),
87+
center=Pose(x=282.45238095238096, y=241.66666666666666, z=902.8809523809524, o_z=1.0),
9388
box=RectangularPrism(dims_mm=Vector3(x=13, y=7, z=11)),
9489
)
9590
],
@@ -101,12 +96,7 @@
10196
reference_frame="depth1",
10297
geometries=[
10398
Geometry(
104-
center=Pose(
105-
x=-129.84615384615384,
106-
y=165.53846153846155,
107-
z=511.46153846153845,
108-
o_z=1.0,
109-
),
99+
center=Pose(x=-129.84615384615384, y=165.53846153846155, z=511.46153846153845, o_z=1.0),
110100
box=RectangularPrism(dims_mm=Vector3(x=5.0, y=4.0, z=7.0)),
111101
)
112102
],

0 commit comments

Comments
 (0)