fix: also update keyline 3d

TRI-ML · Aug 1, 2024 · 260f5e5 · 260f5e5
1 parent 130f796
commit 260f5e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dgp/utils/structures/key_line_3d.py b/dgp/utils/structures/key_line_3d.py
@@ -95,6 +95,6 @@ def to_proto(self):
             As defined in `proto/annotations.proto`
         """
         return [
-            annotations_pb2.KeyPoint3D(x=int(self.x[j]), y=int(self.y[j]), z=int(self.z[j]))
+            annotations_pb2.KeyPoint3D(x=float(self.x[j]), y=float(self.y[j]), z=float(self.z[j]))
             for j, _ in enumerate(self.x)
         ]