fix: fix conversion in key point 3d

TRI-ML · Aug 1, 2024 · 130f796 · 130f796
1 parent d5a9a9d
commit 130f796
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dgp/utils/structures/key_point_3d.py b/dgp/utils/structures/key_point_3d.py
@@ -91,4 +91,4 @@ def to_proto(self):
         KeyPoint3D.pb2
             As defined in `proto/annotations.proto`
         """
-        return annotations_pb2.KeyPoint3D(x=int(self.x), y=int(self.y), z=int(self.z))
+        return annotations_pb2.KeyPoint3D(x=float(self.x), y=float(self.y), z=float(self.z))