Update src/utils.py

src/utils.py

@@ -32,24 +32,42 @@ def parse_string(string, tags):
 
     return results
 
+# def parse_annotations(annotations_list):
+#     """
+#     Converts a list of annotations into a dictionary of key-value pairs.
+
+#     Args:
+#         annotations_list (list): A list of annotations in the format 'key: value'.
+
+#     Returns:
+#         dict: A dictionary with annotation keys and values.
+#     """
+#     annotations_dict = {}
+#     for annotation in annotations_list:
+#         key, value = annotation.split(': ')
+#         annotations_dict[key] = int(value) 
+#     return annotations_dict
+
 def parse_annotations(annotations_list):
     """
     Converts a list of annotations into a dictionary of key-value pairs.
-
     Args:
         annotations_list (list): A list of annotations in the format 'key: value'.
-
     Returns:
         dict: A dictionary with annotation keys and values.
     """
     annotations_dict = {}
     for annotation in annotations_list:
         key, value = annotation.split(': ')
-        annotations_dict[key] = int(value) 
+        if value == 'None':
+            annotations_dict[key] = None
+        else:
+            annotations_dict[key] = int(value)
     return annotations_dict
 
 
 
+
 def encode_video(video_path):
     """
     Encodes a video file into a list of frames.