endow Keyframe with label attribute

dmarx · Dec 13, 2023 · 914d974 · 914d974
1 parent f0a78dc
commit 914d974
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/keyframed/curve.py b/src/keyframed/curve.py
@@ -121,8 +121,10 @@ def __init__(
         value,
         interpolation_method:Optional[Union[str,Callable]]=None,
         interpolator_arguments=None,
+        label=None,
     ):
         self.t=t
+        self.label = label
         #self.value=value
         ### <chatgpt>
         #if isinstance(value, np.ndarray):
@@ -168,6 +170,8 @@ def _to_dict(self, *args, **kwargs) -> dict:
         d = {'t':self.t, 'value':self.value, 'interpolation_method':self.interpolation_method}
         if self.interpolator_arguments:
             d['interpolator_arguments'] = self.interpolator_arguments
+        if self.label is not None:
+            d['label'] = self.label
         ### <chatgpt>
         # Ensure the representation of numpy arrays and tensors are handled correctly
         #if isinstance(self.value, np.ndarray):