.

Volue-Public · Oct 24, 2024 · c706ce5 · c706ce5
1 parent 544ed46
commit c706ce5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/volue/mesh/_common.py b/src/volue/mesh/_common.py
@@ -380,7 +380,7 @@ def _to_proto_curve_type(curve: Timeseries.Curve) -> type.resources_pb2.Curve:
     """
     proto_curve = type.resources_pb2.Curve()
 
-    proto_curve.type = CURVE_TYPES.get(curve, type.resources_pb2.Curve.UNKNOWN)
+    proto_curve.type = CURVE_TYPES[curve]
 
     return proto_curve
 
@@ -392,7 +392,7 @@ def _from_proto_curve_type(proto_curve: type.resources_pb2.Curve) -> Timeseries.
     Args:
         proto_curve: The protobuf curve to convert.
     """
-    return CURVE_TYPES.inverse.get(proto_curve.type, Timeseries.Curve.UNKNOWN)
+    return CURVE_TYPES.inverse[proto_curve.type]
 
 
 RESOLUTIONS = bidict(