Update schema for ground truth admin workflow

Reviewed-on: #228

bin/assert-no-changes.sh

@@ -9,6 +9,7 @@ for proto in $(find ./rbproto -iname '*.proto'); do
 		protoc -I=./rbproto --python_out=./rbproto/python --pyi_out=./rbproto/python ./rbproto/shot.proto
     yarn pbjs ./rbproto/shot.proto --ts ./rbproto/ts/shot.ts
 done
+black rbproto/python
 prettier ./rbproto --write
 
 git ls-files | xargs md5sum > after.txt

flake.nix

@@ -21,7 +21,7 @@
     in
       with pkgs; {
         devShell = mkShell {
-          buildInputs = [nodejs yarn watchman alejandra nodePackages.prettier just protobuf];
+          buildInputs = [nodejs yarn watchman alejandra nodePackages.prettier just protobuf black];
         };
       });
 }

pyproject.toml

@@ -8,7 +8,7 @@ packages = [{include = "rbproto"}]
 
 [tool.poetry.dependencies]
 python = ">=3.10,<=3.13"
-protobuf = "^4.25.3"
+protobuf = ">=4.25.3"
 
 
 [build-system]

rbproto/python/shot_pb2.py

@@ -6,44 +6,45 @@ from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
+
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
 
 
-
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-
+    b'\n\nshot.proto\x12\x0fserialized_shot"?\n\x03\x42ox\x12\x0c\n\x04left\x18\x01 \x01(\x02\x12\x0b\n\x03top\x18\x02 \x01(\x02\x12\r\n\x05width\x18\x03 \x01(\x02\x12\x0e\n\x06height\x18\x04 \x01(\x02"\x1d\n\x05Point\x12\t\n\x01x\x18\x01 \x01(\x02\x12\t\n\x01y\x18\x02 \x01(\x02"\x7f\n\rBallDetection\x12.\n\x0eplane_position\x18\x01 \x01(\x0b\x32\x16.serialized_shot.Point\x12(\n\nannotation\x18\x02 \x01(\x0b\x32\x14.serialized_shot.Box\x12\x14\n\x0cinterpolated\x18\x03 \x01(\x08"T\n\x10RLEBallDetection\x12\x31\n\tdetection\x18\x01 \x01(\x0b\x32\x1e.serialized_shot.BallDetection\x12\r\n\x05\x63ount\x18\x02 \x01(\r"L\n\x13RLEDetectionHistory\x12\x35\n\ndetections\x18\x01 \x03(\x0b\x32!.serialized_shot.RLEBallDetection"F\n\x10\x44\x65tectionHistory\x12\x32\n\ndetections\x18\x01 \x03(\x0b\x32\x1e.serialized_shot.BallDetection"\xfc\x01\n\rCollisionInfo\x12\x0e\n\x06source\x18\x01 \x01(\r\x12M\n\x10\x62\x61ll_identifiers\x18\x02 \x03(\x0b\x32\x33.serialized_shot.CollisionInfo.BallIdentifiersEntry\x12\x17\n\x0fwall_identifier\x18\x03 \x01(\r\x12\x13\n\x0b\x66rame_index\x18\x04 \x01(\r\x12\x0e\n\x06static\x18\x05 \x01(\x08\x1aN\n\x14\x42\x61llIdentifiersEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.serialized_shot.Point:\x02\x38\x01"\xcc\x02\n\x04Path\x12\x13\n\x0bstart_frame\x18\x01 \x01(\r\x12\x11\n\tend_frame\x18\x02 \x01(\r\x12\x37\n\ndetections\x18\x03 \x01(\x0b\x32!.serialized_shot.DetectionHistoryH\x00\x12>\n\x0erle_detections\x18\x04 \x01(\x0b\x32$.serialized_shot.RLEDetectionHistoryH\x00\x12\x15\n\x0bnot_present\x18\x05 \x01(\x08H\x00\x12\x11\n\tis_static\x18\x06 \x01(\x08\x12\x32\n\nstart_info\x18\x07 \x01(\x0b\x32\x1e.serialized_shot.CollisionInfo\x12\x30\n\x08\x65nd_info\x18\x08 \x01(\x0b\x32\x1e.serialized_shot.CollisionInfoB\x13\n\x11\x64\x65tection_history"R\n\x11IdentifierHistory\x12\x17\n\x0f\x62\x61ll_identifier\x18\x01 \x01(\r\x12$\n\x05paths\x18\x02 \x03(\x0b\x32\x15.serialized_shot.Path"\xf4\x01\n\x12KeyBallIdentifiers\x12\x10\n\x08\x63ue_ball\x18\x01 \x01(\r\x12\x13\n\x0bobject_ball\x18\x02 \x01(\r\x12\x13\n\x0btarget_ball\x18\x03 \x01(\r\x12\x18\n\x10\x63ontact_sequence\x18\x04 \x03(\r\x12\x18\n\x0b\x63ue_ball_id\x18\x05 \x01(\rH\x00\x88\x01\x01\x12\x1b\n\x0eobject_ball_id\x18\x06 \x01(\rH\x01\x88\x01\x01\x12\x1b\n\x0etarget_ball_id\x18\x07 \x01(\rH\x02\x88\x01\x01\x42\x0e\n\x0c_cue_ball_idB\x11\n\x0f_object_ball_idB\x11\n\x0f_target_ball_id"\xa8\x01\n\x04Shot\x12@\n\x14identifier_histories\x18\x03 \x03(\x0b\x32".serialized_shot.IdentifierHistory\x12\x36\n\tkey_balls\x18\x04 \x01(\x0b\x32#.serialized_shot.KeyBallIdentifiers\x12\x13\n\x0bstart_index\x18\x05 \x01(\r\x12\x11\n\tend_index\x18\x06 \x01(\rb\x06proto3'
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\nshot.proto\x12\x0fserialized_shot\"?\n\x03\x42ox\x12\x0c\n\x04left\x18\x01 \x01(\x02\x12\x0b\n\x03top\x18\x02 \x01(\x02\x12\r\n\x05width\x18\x03 \x01(\x02\x12\x0e\n\x06height\x18\x04 \x01(\x02\"\x1d\n\x05Point\x12\t\n\x01x\x18\x01 \x01(\x02\x12\t\n\x01y\x18\x02 \x01(\x02\"\x7f\n\rBallDetection\x12.\n\x0eplane_position\x18\x01 \x01(\x0b\x32\x16.serialized_shot.Point\x12(\n\nannotation\x18\x02 \x01(\x0b\x32\x14.serialized_shot.Box\x12\x14\n\x0cinterpolated\x18\x03 \x01(\x08\"T\n\x10RLEBallDetection\x12\x31\n\tdetection\x18\x01 \x01(\x0b\x32\x1e.serialized_shot.BallDetection\x12\r\n\x05\x63ount\x18\x02 \x01(\r\"L\n\x13RLEDetectionHistory\x12\x35\n\ndetections\x18\x01 \x03(\x0b\x32!.serialized_shot.RLEBallDetection\"F\n\x10\x44\x65tectionHistory\x12\x32\n\ndetections\x18\x01 \x03(\x0b\x32\x1e.serialized_shot.BallDetection\"\xfc\x01\n\rCollisionInfo\x12\x0e\n\x06source\x18\x01 \x01(\r\x12M\n\x10\x62\x61ll_identifiers\x18\x02 \x03(\x0b\x32\x33.serialized_shot.CollisionInfo.BallIdentifiersEntry\x12\x17\n\x0fwall_identifier\x18\x03 \x01(\r\x12\x13\n\x0b\x66rame_index\x18\x04 \x01(\r\x12\x0e\n\x06static\x18\x05 \x01(\x08\x1aN\n\x14\x42\x61llIdentifiersEntry\x12\x0b\n\x03key\x18\x01 \x01(\r\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.serialized_shot.Point:\x02\x38\x01\"\xcc\x02\n\x04Path\x12\x13\n\x0bstart_frame\x18\x01 \x01(\r\x12\x11\n\tend_frame\x18\x02 \x01(\r\x12\x37\n\ndetections\x18\x03 \x01(\x0b\x32!.serialized_shot.DetectionHistoryH\x00\x12>\n\x0erle_detections\x18\x04 \x01(\x0b\x32$.serialized_shot.RLEDetectionHistoryH\x00\x12\x15\n\x0bnot_present\x18\x05 \x01(\x08H\x00\x12\x11\n\tis_static\x18\x06 \x01(\x08\x12\x32\n\nstart_info\x18\x07 \x01(\x0b\x32\x1e.serialized_shot.CollisionInfo\x12\x30\n\x08\x65nd_info\x18\x08 \x01(\x0b\x32\x1e.serialized_shot.CollisionInfoB\x13\n\x11\x64\x65tection_history\"R\n\x11IdentifierHistory\x12\x17\n\x0f\x62\x61ll_identifier\x18\x01 \x01(\r\x12$\n\x05paths\x18\x02 \x03(\x0b\x32\x15.serialized_shot.Path\"\xf4\x01\n\x12KeyBallIdentifiers\x12\x10\n\x08\x63ue_ball\x18\x01 \x01(\r\x12\x13\n\x0bobject_ball\x18\x02 \x01(\r\x12\x13\n\x0btarget_ball\x18\x03 \x01(\r\x12\x18\n\x10\x63ontact_sequence\x18\x04 \x03(\r\x12\x18\n\x0b\x63ue_ball_id\x18\x05 \x01(\rH\x00\x88\x01\x01\x12\x1b\n\x0eobject_ball_id\x18\x06 \x01(\rH\x01\x88\x01\x01\x12\x1b\n\x0etarget_ball_id\x18\x07 \x01(\rH\x02\x88\x01\x01\x42\x0e\n\x0c_cue_ball_idB\x11\n\x0f_object_ball_idB\x11\n\x0f_target_ball_id\"\xa8\x01\n\x04Shot\x12@\n\x14identifier_histories\x18\x03 \x03(\x0b\x32\".serialized_shot.IdentifierHistory\x12\x36\n\tkey_balls\x18\x04 \x01(\x0b\x32#.serialized_shot.KeyBallIdentifiers\x12\x13\n\x0bstart_index\x18\x05 \x01(\r\x12\x11\n\tend_index\x18\x06 \x01(\rb\x06proto3')
+)
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'shot_pb2', _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "shot_pb2", _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
+    DESCRIPTOR._options = None
-  _globals['_COLLISIONINFO_BALLIDENTIFIERSENTRY']._options = None
+    _globals["_COLLISIONINFO_BALLIDENTIFIERSENTRY"]._options = None
-  _globals['_COLLISIONINFO_BALLIDENTIFIERSENTRY']._serialized_options = b'8\001'
+    _globals["_COLLISIONINFO_BALLIDENTIFIERSENTRY"]._serialized_options = b"8\001"
-  _globals['_BOX']._serialized_start=31
+    _globals["_BOX"]._serialized_start = 31
-  _globals['_BOX']._serialized_end=94
+    _globals["_BOX"]._serialized_end = 94
-  _globals['_POINT']._serialized_start=96
+    _globals["_POINT"]._serialized_start = 96
-  _globals['_POINT']._serialized_end=125
+    _globals["_POINT"]._serialized_end = 125
-  _globals['_BALLDETECTION']._serialized_start=127
+    _globals["_BALLDETECTION"]._serialized_start = 127
-  _globals['_BALLDETECTION']._serialized_end=254
+    _globals["_BALLDETECTION"]._serialized_end = 254
-  _globals['_RLEBALLDETECTION']._serialized_start=256
+    _globals["_RLEBALLDETECTION"]._serialized_start = 256
-  _globals['_RLEBALLDETECTION']._serialized_end=340
+    _globals["_RLEBALLDETECTION"]._serialized_end = 340
-  _globals['_RLEDETECTIONHISTORY']._serialized_start=342
+    _globals["_RLEDETECTIONHISTORY"]._serialized_start = 342
-  _globals['_RLEDETECTIONHISTORY']._serialized_end=418
+    _globals["_RLEDETECTIONHISTORY"]._serialized_end = 418
-  _globals['_DETECTIONHISTORY']._serialized_start=420
+    _globals["_DETECTIONHISTORY"]._serialized_start = 420
-  _globals['_DETECTIONHISTORY']._serialized_end=490
+    _globals["_DETECTIONHISTORY"]._serialized_end = 490
-  _globals['_COLLISIONINFO']._serialized_start=493
+    _globals["_COLLISIONINFO"]._serialized_start = 493
-  _globals['_COLLISIONINFO']._serialized_end=745
+    _globals["_COLLISIONINFO"]._serialized_end = 745
-  _globals['_COLLISIONINFO_BALLIDENTIFIERSENTRY']._serialized_start=667
+    _globals["_COLLISIONINFO_BALLIDENTIFIERSENTRY"]._serialized_start = 667
-  _globals['_COLLISIONINFO_BALLIDENTIFIERSENTRY']._serialized_end=745
+    _globals["_COLLISIONINFO_BALLIDENTIFIERSENTRY"]._serialized_end = 745
-  _globals['_PATH']._serialized_start=748
+    _globals["_PATH"]._serialized_start = 748
-  _globals['_PATH']._serialized_end=1080
+    _globals["_PATH"]._serialized_end = 1080
-  _globals['_IDENTIFIERHISTORY']._serialized_start=1082
+    _globals["_IDENTIFIERHISTORY"]._serialized_start = 1082
-  _globals['_IDENTIFIERHISTORY']._serialized_end=1164
+    _globals["_IDENTIFIERHISTORY"]._serialized_end = 1164
-  _globals['_KEYBALLIDENTIFIERS']._serialized_start=1167
+    _globals["_KEYBALLIDENTIFIERS"]._serialized_start = 1167
-  _globals['_KEYBALLIDENTIFIERS']._serialized_end=1411
+    _globals["_KEYBALLIDENTIFIERS"]._serialized_end = 1411
-  _globals['_SHOT']._serialized_start=1414
+    _globals["_SHOT"]._serialized_start = 1414
-  _globals['_SHOT']._serialized_end=1582
+    _globals["_SHOT"]._serialized_end = 1582
 # @@protoc_insertion_point(module_scope)

rbproto/python/shot_pb2.pyi

@@ -1,7 +1,13 @@
 from google.protobuf.internal import containers as _containers
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
-from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+from typing import (
+    ClassVar as _ClassVar,
+    Iterable as _Iterable,
+    Mapping as _Mapping,
+    Optional as _Optional,
+    Union as _Union,
+)
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
@@ -15,7 +21,13 @@ class Box(_message.Message):
     top: float
     width: float
     height: float
-    def __init__(self, left: _Optional[float] = ..., top: _Optional[float] = ..., width: _Optional[float] = ..., height: _Optional[float] = ...) -> None: ...
+    def __init__(
+        self,
+        left: _Optional[float] = ...,
+        top: _Optional[float] = ...,
+        width: _Optional[float] = ...,
+        height: _Optional[float] = ...,
+    ) -> None: ...
 
 class Point(_message.Message):
     __slots__ = ["x", "y"]
@@ -23,7 +35,9 @@ class Point(_message.Message):
     Y_FIELD_NUMBER: _ClassVar[int]
     x: float
     y: float
-    def __init__(self, x: _Optional[float] = ..., y: _Optional[float] = ...) -> None: ...
+    def __init__(
+        self, x: _Optional[float] = ..., y: _Optional[float] = ...
+    ) -> None: ...
 
 class BallDetection(_message.Message):
     __slots__ = ["plane_position", "annotation", "interpolated"]
@@ -33,7 +47,12 @@ class BallDetection(_message.Message):
     plane_position: Point
     annotation: Box
     interpolated: bool
-    def __init__(self, plane_position: _Optional[_Union[Point, _Mapping]] = ..., annotation: _Optional[_Union[Box, _Mapping]] = ..., interpolated: bool = ...) -> None: ...
+    def __init__(
+        self,
+        plane_position: _Optional[_Union[Point, _Mapping]] = ...,
+        annotation: _Optional[_Union[Box, _Mapping]] = ...,
+        interpolated: bool = ...,
+    ) -> None: ...
 
 class RLEBallDetection(_message.Message):
     __slots__ = ["detection", "count"]
@@ -41,29 +60,48 @@ class RLEBallDetection(_message.Message):
     COUNT_FIELD_NUMBER: _ClassVar[int]
     detection: BallDetection
     count: int
-    def __init__(self, detection: _Optional[_Union[BallDetection, _Mapping]] = ..., count: _Optional[int] = ...) -> None: ...
+    def __init__(
+        self,
+        detection: _Optional[_Union[BallDetection, _Mapping]] = ...,
+        count: _Optional[int] = ...,
+    ) -> None: ...
 
 class RLEDetectionHistory(_message.Message):
     __slots__ = ["detections"]
     DETECTIONS_FIELD_NUMBER: _ClassVar[int]
     detections: _containers.RepeatedCompositeFieldContainer[RLEBallDetection]
-    def __init__(self, detections: _Optional[_Iterable[_Union[RLEBallDetection, _Mapping]]] = ...) -> None: ...
+    def __init__(
+        self, detections: _Optional[_Iterable[_Union[RLEBallDetection, _Mapping]]] = ...
+    ) -> None: ...
 
 class DetectionHistory(_message.Message):
     __slots__ = ["detections"]
     DETECTIONS_FIELD_NUMBER: _ClassVar[int]
     detections: _containers.RepeatedCompositeFieldContainer[BallDetection]
-    def __init__(self, detections: _Optional[_Iterable[_Union[BallDetection, _Mapping]]] = ...) -> None: ...
+    def __init__(
+        self, detections: _Optional[_Iterable[_Union[BallDetection, _Mapping]]] = ...
+    ) -> None: ...
 
 class CollisionInfo(_message.Message):
-    __slots__ = ["source", "ball_identifiers", "wall_identifier", "frame_index", "static"]
+    __slots__ = [
+        "source",
+        "ball_identifiers",
+        "wall_identifier",
+        "frame_index",
+        "static",
+    ]
+
     class BallIdentifiersEntry(_message.Message):
         __slots__ = ["key", "value"]
         KEY_FIELD_NUMBER: _ClassVar[int]
         VALUE_FIELD_NUMBER: _ClassVar[int]
         key: int
         value: Point
-        def __init__(self, key: _Optional[int] = ..., value: _Optional[_Union[Point, _Mapping]] = ...) -> None: ...
+        def __init__(
+            self,
+            key: _Optional[int] = ...,
+            value: _Optional[_Union[Point, _Mapping]] = ...,
+        ) -> None: ...
     SOURCE_FIELD_NUMBER: _ClassVar[int]
     BALL_IDENTIFIERS_FIELD_NUMBER: _ClassVar[int]
     WALL_IDENTIFIER_FIELD_NUMBER: _ClassVar[int]
@@ -74,10 +112,26 @@ class CollisionInfo(_message.Message):
     wall_identifier: int
     frame_index: int
     static: bool
-    def __init__(self, source: _Optional[int] = ..., ball_identifiers: _Optional[_Mapping[int, Point]] = ..., wall_identifier: _Optional[int] = ..., frame_index: _Optional[int] = ..., static: bool = ...) -> None: ...
+    def __init__(
+        self,
+        source: _Optional[int] = ...,
+        ball_identifiers: _Optional[_Mapping[int, Point]] = ...,
+        wall_identifier: _Optional[int] = ...,
+        frame_index: _Optional[int] = ...,
+        static: bool = ...,
+    ) -> None: ...
 
 class Path(_message.Message):
-    __slots__ = ["start_frame", "end_frame", "detections", "rle_detections", "not_present", "is_static", "start_info", "end_info"]
+    __slots__ = [
+        "start_frame",
+        "end_frame",
+        "detections",
+        "rle_detections",
+        "not_present",
+        "is_static",
+        "start_info",
+        "end_info",
+    ]
     START_FRAME_FIELD_NUMBER: _ClassVar[int]
     END_FRAME_FIELD_NUMBER: _ClassVar[int]
     DETECTIONS_FIELD_NUMBER: _ClassVar[int]
@@ -94,7 +148,17 @@ class Path(_message.Message):
     is_static: bool
     start_info: CollisionInfo
     end_info: CollisionInfo
-    def __init__(self, start_frame: _Optional[int] = ..., end_frame: _Optional[int] = ..., detections: _Optional[_Union[DetectionHistory, _Mapping]] = ..., rle_detections: _Optional[_Union[RLEDetectionHistory, _Mapping]] = ..., not_present: bool = ..., is_static: bool = ..., start_info: _Optional[_Union[CollisionInfo, _Mapping]] = ..., end_info: _Optional[_Union[CollisionInfo, _Mapping]] = ...) -> None: ...
+    def __init__(
+        self,
+        start_frame: _Optional[int] = ...,
+        end_frame: _Optional[int] = ...,
+        detections: _Optional[_Union[DetectionHistory, _Mapping]] = ...,
+        rle_detections: _Optional[_Union[RLEDetectionHistory, _Mapping]] = ...,
+        not_present: bool = ...,
+        is_static: bool = ...,
+        start_info: _Optional[_Union[CollisionInfo, _Mapping]] = ...,
+        end_info: _Optional[_Union[CollisionInfo, _Mapping]] = ...,
+    ) -> None: ...
 
 class IdentifierHistory(_message.Message):
     __slots__ = ["ball_identifier", "paths"]
@@ -102,10 +166,22 @@ class IdentifierHistory(_message.Message):
     PATHS_FIELD_NUMBER: _ClassVar[int]
     ball_identifier: int
     paths: _containers.RepeatedCompositeFieldContainer[Path]
-    def __init__(self, ball_identifier: _Optional[int] = ..., paths: _Optional[_Iterable[_Union[Path, _Mapping]]] = ...) -> None: ...
+    def __init__(
+        self,
+        ball_identifier: _Optional[int] = ...,
+        paths: _Optional[_Iterable[_Union[Path, _Mapping]]] = ...,
+    ) -> None: ...
 
 class KeyBallIdentifiers(_message.Message):
-    __slots__ = ["cue_ball", "object_ball", "target_ball", "contact_sequence", "cue_ball_id", "object_ball_id", "target_ball_id"]
+    __slots__ = [
+        "cue_ball",
+        "object_ball",
+        "target_ball",
+        "contact_sequence",
+        "cue_ball_id",
+        "object_ball_id",
+        "target_ball_id",
+    ]
     CUE_BALL_FIELD_NUMBER: _ClassVar[int]
     OBJECT_BALL_FIELD_NUMBER: _ClassVar[int]
     TARGET_BALL_FIELD_NUMBER: _ClassVar[int]
@@ -120,7 +196,16 @@ class KeyBallIdentifiers(_message.Message):
     cue_ball_id: int
     object_ball_id: int
     target_ball_id: int
-    def __init__(self, cue_ball: _Optional[int] = ..., object_ball: _Optional[int] = ..., target_ball: _Optional[int] = ..., contact_sequence: _Optional[_Iterable[int]] = ..., cue_ball_id: _Optional[int] = ..., object_ball_id: _Optional[int] = ..., target_ball_id: _Optional[int] = ...) -> None: ...
+    def __init__(
+        self,
+        cue_ball: _Optional[int] = ...,
+        object_ball: _Optional[int] = ...,
+        target_ball: _Optional[int] = ...,
+        contact_sequence: _Optional[_Iterable[int]] = ...,
+        cue_ball_id: _Optional[int] = ...,
+        object_ball_id: _Optional[int] = ...,
+        target_ball_id: _Optional[int] = ...,
+    ) -> None: ...
 
 class Shot(_message.Message):
     __slots__ = ["identifier_histories", "key_balls", "start_index", "end_index"]
@@ -132,4 +217,12 @@ class Shot(_message.Message):
     key_balls: KeyBallIdentifiers
     start_index: int
     end_index: int
-    def __init__(self, identifier_histories: _Optional[_Iterable[_Union[IdentifierHistory, _Mapping]]] = ..., key_balls: _Optional[_Union[KeyBallIdentifiers, _Mapping]] = ..., start_index: _Optional[int] = ..., end_index: _Optional[int] = ...) -> None: ...
+    def __init__(
+        self,
+        identifier_histories: _Optional[
+            _Iterable[_Union[IdentifierHistory, _Mapping]]
+        ] = ...,
+        key_balls: _Optional[_Union[KeyBallIdentifiers, _Mapping]] = ...,
+        start_index: _Optional[int] = ...,
+        end_index: _Optional[int] = ...,
+    ) -> None: ...

src/index.tsx

@@ -224,6 +224,14 @@ export type CreateBucketSetInput = {
   keyName: Scalars["String"]["input"];
 };
 
+export type CreateGroundTruthShotInputGql = {
+  endFrame: Scalars["Int"]["input"];
+  make?: InputMaybe<Scalars["Boolean"]["input"]>;
+  notes?: InputMaybe<Scalars["String"]["input"]>;
+  processingId: Scalars["Int"]["input"];
+  startFrame: Scalars["Int"]["input"];
+};
+
 export type CreateSubscriptionResultGql = {
   __typename?: "CreateSubscriptionResultGQL";
   checkoutUrl: Scalars["String"]["output"];
@@ -269,10 +277,17 @@ export type DatetimeRangeAggregationInput = {
   startDatetime?: InputMaybe<Scalars["DateTime"]["input"]>;
 };
 
+export type DeleteGroundTruthShotReturn = {
+  __typename?: "DeleteGroundTruthShotReturn";
+  deletedShotId: Scalars["Int"]["output"];
+};
+
 export type DeployedConfigGql = {
   __typename?: "DeployedConfigGQL";
   allowNewUsers: Scalars["Boolean"]["output"];
   bannerMessages: Array<BannerGql>;
+  bucketUrl: Scalars["String"]["output"];
+  defaultAndroidRecordingFormat: StreamSegmentTypeEnum;
   devMode: Scalars["Boolean"]["output"];
   environment: Scalars["String"]["output"];
   firebase: Scalars["Boolean"]["output"];
@@ -306,9 +321,12 @@ export type EditUserInputGql = {
 export type EditableShotFieldInputGql = {
   backcut?: InputMaybe<Scalars["Boolean"]["input"]>;
   excludeFromStats?: InputMaybe<Scalars["Boolean"]["input"]>;
+  intendedBallId?: InputMaybe<Scalars["Int"]["input"]>;
+  intendedPocketId?: InputMaybe<PocketIdentifier>;
   intendedPocketType?: InputMaybe<PocketEnum>;
   make?: InputMaybe<Scalars["Boolean"]["input"]>;
   notes?: InputMaybe<Scalars["String"]["input"]>;
+  pathMetadataIndex?: InputMaybe<Scalars["Int"]["input"]>;
   shotDirection?: InputMaybe<ShotDirectionEnum>;
   spinType?: InputMaybe<SpinTypeEnum>;
   targetPocketAngleDirection?: InputMaybe<ShotDirectionEnum>;
@@ -349,6 +367,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -393,6 +412,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -437,6 +457,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -481,6 +502,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -525,6 +547,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -569,6 +592,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -613,6 +637,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -657,6 +682,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -701,6 +727,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -745,6 +772,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -789,6 +817,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -833,6 +862,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -877,6 +907,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -921,6 +952,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -965,6 +997,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1009,6 +1042,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1053,6 +1087,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1097,6 +1132,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1141,6 +1177,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1185,6 +1222,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1229,6 +1267,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1273,6 +1312,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1317,6 +1357,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1361,6 +1402,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1405,6 +1447,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1449,6 +1492,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1493,6 +1537,7 @@ export type FilterInput =
       missAngleInDegrees: FloatRangeFilter;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1537,6 +1582,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter: FilterInput;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1581,6 +1627,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters: Array<FilterInput>;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1625,6 +1672,52 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId: Array<Scalars["Int"]["input"]>;
+      runId?: never;
+      runLength?: never;
+      shotDirection?: never;
+      spinType?: never;
+      tableSize?: never;
+      tags?: never;
+      targetPocketAngle?: never;
+      targetPocketAngleDirection?: never;
+      targetPocketDistance?: never;
+      totalDistance?: never;
+      userId?: never;
+      username?: never;
+      videoId?: never;
+    }
+  | {
+      andFilters?: never;
+      annotations?: never;
+      backcut?: never;
+      bankAngle?: never;
+      bankDistance?: never;
+      createdAt?: never;
+      cueAngleAfterObject?: never;
+      cueBallSpeed?: never;
+      cueObjectAngle?: never;
+      cueObjectDistance?: never;
+      cueSpeedAfterObject?: never;
+      difficulty?: never;
+      falsePositiveScore?: never;
+      fargoRating?: never;
+      intendedPocketType?: never;
+      isBreakHeuristic?: never;
+      isDirect?: never;
+      isLeft?: never;
+      isLeftMiss?: never;
+      isRight?: never;
+      isRightMiss?: never;
+      isStraight?: never;
+      kickAngle?: never;
+      kickDistance?: never;
+      make?: never;
+      marginOfErrorInDegrees?: never;
+      missAngleInDegrees?: never;
+      notFilter?: never;
+      orFilters?: never;
+      processingId?: never;
       runId: Array<Scalars["Int"]["input"]>;
       runLength?: never;
       shotDirection?: never;
@@ -1669,6 +1762,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength: FloatRangeFilter;
       shotDirection?: never;
@@ -1713,6 +1807,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection: Array<ShotDirectionEnum>;
@@ -1757,6 +1852,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1801,6 +1897,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1845,6 +1942,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1889,6 +1987,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1933,6 +2032,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1977,6 +2077,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -2021,6 +2122,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -2065,6 +2167,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -2109,6 +2212,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -2153,6 +2257,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -2344,10 +2449,12 @@ export type Mutation = {
   commentOnVideo: Scalars["Boolean"]["output"];
   createBucketSet: BucketSetGql;
   createChallenge: Challenge;
+  createGroundTruthShot: ShotGql;
   createRuleSet: RuleSet;
   createSubscription: CreateSubscriptionResultGql;
   createUploadStream: CreateUploadStreamReturn;
   deleteComment: Scalars["Boolean"]["output"];
+  deleteGroundTruthShot: DeleteGroundTruthShotReturn;
   deleteNotification: Scalars["Boolean"]["output"];
   deleteTags: Scalars["Boolean"]["output"];
   deleteUser: Scalars["Boolean"]["output"];
@@ -2362,12 +2469,14 @@ export type Mutation = {
   findPrerecordTableLayout?: Maybe<HomographyInfoGql>;
   followUser: UserGql;
   getHlsInitUploadLink: GetUploadLinkReturn;
+  getOrCreateGroundTruthProcessing: VideoProcessingGql;
   getProfileImageUploadLink: GetProfileUploadLinkReturn;
   getUploadLink: GetUploadLinkReturn;
   inviteUsersToChallenge: Array<ChallengeInvitation>;
   markAllNotificationsAsRead: Scalars["Boolean"]["output"];
   markNotificationAsRead: Scalars["Boolean"]["output"];
   markNotificationsAsRead: Scalars["Boolean"]["output"];
+  promoteGroundTruthProcessing: VideoProcessingGql;
   reactToVideo: Scalars["Boolean"]["output"];
   recalculateChallengeEntry: ChallengeEntry;
   reportContent: Scalars["Boolean"]["output"];
@@ -2375,11 +2484,13 @@ export type Mutation = {
   retireTags: Scalars["Boolean"]["output"];
   setLoggerLevel: Scalars["Boolean"]["output"];
   setSegmentDuration: Scalars["Boolean"]["output"];
+  splitGroundTruthShot: SplitGroundTruthShotReturn;
   startChallenge: ChallengeEntry;
   submitCancellationFeedback: Scalars["Boolean"]["output"];
   submitChallengeEntry: ChallengeEntry;
   undismissChallenge: Scalars["Boolean"]["output"];
   unfollowUser: UserGql;
+  updateGroundTruthShotFrameRange: ShotGql;
   updateShotAnnotations: UpdateShotAnnotationReturn;
 };
 
@@ -2420,6 +2531,10 @@ export type MutationCreateChallengeArgs = {
   startDate: Scalars["DateTime"]["input"];
 };
 
+export type MutationCreateGroundTruthShotArgs = {
+  input: CreateGroundTruthShotInputGql;
+};
+
 export type MutationCreateRuleSetArgs = {
   description?: InputMaybe<Scalars["String"]["input"]>;
   name: Scalars["String"]["input"];
@@ -2430,6 +2545,7 @@ export type MutationCreateSubscriptionArgs = {
 };
 
 export type MutationCreateUploadStreamArgs = {
+  expectedDurationSeconds?: InputMaybe<Scalars["Float"]["input"]>;
   videoMetadata: VideoMetadataInput;
 };
 
@@ -2438,6 +2554,10 @@ export type MutationDeleteCommentArgs = {
   videoId: Scalars["Int"]["input"];
 };
 
+export type MutationDeleteGroundTruthShotArgs = {
+  shotId: Scalars["Int"]["input"];
+};
+
 export type MutationDeleteNotificationArgs = {
   notificationId: Scalars["Int"]["input"];
 };
@@ -2492,6 +2612,11 @@ export type MutationGetHlsInitUploadLinkArgs = {
   videoId: Scalars["Int"]["input"];
 };
 
+export type MutationGetOrCreateGroundTruthProcessingArgs = {
+  sourceProcessingId?: InputMaybe<Scalars["Int"]["input"]>;
+  videoId: Scalars["Int"]["input"];
+};
+
 export type MutationGetProfileImageUploadLinkArgs = {
   fileExt?: InputMaybe<Scalars["String"]["input"]>;
 };
@@ -2514,6 +2639,10 @@ export type MutationMarkNotificationsAsReadArgs = {
   notificationIds: Array<Scalars["Int"]["input"]>;
 };
 
+export type MutationPromoteGroundTruthProcessingArgs = {
+  processingId: Scalars["Int"]["input"];
+};
+
 export type MutationReactToVideoArgs = {
   reaction?: InputMaybe<ReactionEnum>;
   videoId: Scalars["Int"]["input"];
@@ -2549,6 +2678,12 @@ export type MutationSetSegmentDurationArgs = {
   videoId: Scalars["Int"]["input"];
 };
 
+export type MutationSplitGroundTruthShotArgs = {
+  notes?: InputMaybe<Scalars["String"]["input"]>;
+  shotId: Scalars["Int"]["input"];
+  splitFrame: Scalars["Int"]["input"];
+};
+
 export type MutationStartChallengeArgs = {
   challengeId: Scalars["ID"]["input"];
 };
@@ -2572,6 +2707,13 @@ export type MutationUnfollowUserArgs = {
   followedUserId: Scalars["Int"]["input"];
 };
 
+export type MutationUpdateGroundTruthShotFrameRangeArgs = {
+  endFrame?: InputMaybe<Scalars["Int"]["input"]>;
+  notes?: InputMaybe<Scalars["String"]["input"]>;
+  shotId: Scalars["Int"]["input"];
+  startFrame?: InputMaybe<Scalars["Int"]["input"]>;
+};
+
 export type MutationUpdateShotAnnotationsArgs = {
   annotations: Array<UpdateAnnotationInputGql>;
   shotId: Scalars["Int"]["input"];
@@ -2699,6 +2841,7 @@ export enum ProcessingStatusEnum {
 
 export type Query = {
   __typename?: "Query";
+  assessVideoProcessing: VideoProcessingAssessmentGql;
   challenge?: Maybe<Challenge>;
   challengeLeaderboard: Array<ChallengeEntry>;
   challenges: Array<Challenge>;
@@ -2714,6 +2857,7 @@ export type Query = {
   getMedals: RequestedMedalsGql;
   getOrderedShots: GetShotsResult;
   getPlayTime: UserPlayTimeGql;
+  getQuotaStatus: QuotaStatusGql;
   getRuns: GetRunsResult;
   getShotAnnotationTypes: Array<ShotAnnotationTypeGql>;
   getShots: Array<ShotGql>;
@@ -2729,6 +2873,7 @@ export type Query = {
   getUsersMatching: Array<UserGql>;
   getVideo: VideoGql;
   getVideoMakePercentageIntervals: Array<MakePercentageIntervalGql>;
+  getVideoProcessings: Array<VideoProcessingGql>;
   getVideos: Array<VideoGql>;
   isChallengeDismissed: Scalars["Boolean"]["output"];
   myChallengeEntries: Array<ChallengeEntry>;
@@ -2740,6 +2885,11 @@ export type Query = {
   waitFor: Scalars["Float"]["output"];
 };
 
+export type QueryAssessVideoProcessingArgs = {
+  groundTruthProcessingId?: InputMaybe<Scalars["Int"]["input"]>;
+  processingId: Scalars["Int"]["input"];
+};
+
 export type QueryChallengeArgs = {
   id: Scalars["ID"]["input"];
 };
@@ -2795,7 +2945,9 @@ export type QueryGetOrderedShotsArgs = {
   countRespectsLimit?: Scalars["Boolean"]["input"];
   filterInput: FilterInput;
   ids?: InputMaybe<Array<Scalars["Int"]["input"]>>;
+  includeExcluded?: Scalars["Boolean"]["input"];
   limit?: Scalars["Int"]["input"];
+  processingId?: InputMaybe<Scalars["Int"]["input"]>;
   shotsOrdering?: InputMaybe<GetShotsOrdering>;
 };
 
@@ -2819,7 +2971,9 @@ export type QueryGetShotAnnotationTypesArgs = {
 export type QueryGetShotsArgs = {
   countRespectsLimit?: Scalars["Boolean"]["input"];
   filterInput: FilterInput;
+  includeExcluded?: Scalars["Boolean"]["input"];
   limit?: Scalars["Int"]["input"];
+  processingId?: InputMaybe<Scalars["Int"]["input"]>;
   shotsPagination?: InputMaybe<GetShotsPagination>;
 };
 
@@ -2831,7 +2985,9 @@ export type QueryGetShotsWithMetadataArgs = {
   countRespectsLimit?: Scalars["Boolean"]["input"];
   filterInput: FilterInput;
   ids?: InputMaybe<Array<Scalars["Int"]["input"]>>;
+  includeExcluded?: Scalars["Boolean"]["input"];
   limit?: Scalars["Int"]["input"];
+  processingId?: InputMaybe<Scalars["Int"]["input"]>;
   shotsPagination?: InputMaybe<GetShotsPagination>;
 };
 
@@ -2885,6 +3041,10 @@ export type QueryGetVideoMakePercentageIntervalsArgs = {
   videoId: Scalars["ID"]["input"];
 };
 
+export type QueryGetVideoProcessingsArgs = {
+  videoId: Scalars["Int"]["input"];
+};
+
 export type QueryGetVideosArgs = {
   videoIds: Array<Scalars["Int"]["input"]>;
 };
@@ -2903,6 +3063,18 @@ export type QueryWaitForArgs = {
   duration: Scalars["Float"]["input"];
 };
 
+export type QuotaStatusGql = {
+  __typename?: "QuotaStatusGQL";
+  canUpload: Scalars["Boolean"]["output"];
+  durationLimitSeconds?: Maybe<Scalars["Int"]["output"]>;
+  durationRemainingSeconds?: Maybe<Scalars["Float"]["output"]>;
+  durationUsedSeconds: Scalars["Float"]["output"];
+  maxVideoDurationSeconds?: Maybe<Scalars["Int"]["output"]>;
+  periodEnd: Scalars["DateTime"]["output"];
+  periodStart: Scalars["DateTime"]["output"];
+  tierName: Scalars["String"]["output"];
+};
+
 export enum ReactionEnum {
   Bullseye = "BULLSEYE",
   Heart = "HEART",
@@ -3169,6 +3341,12 @@ export enum SpinTypeEnum {
   Unknown = "UNKNOWN",
 }
 
+export type SplitGroundTruthShotReturn = {
+  __typename?: "SplitGroundTruthShotReturn";
+  createdShot: ShotGql;
+  originalShot: ShotGql;
+};
+
 export type StreamErrorGql = {
   __typename?: "StreamErrorGQL";
   message: Scalars["String"]["output"];
@@ -3477,6 +3655,39 @@ export type VideoMetadataInput = {
   videoName?: InputMaybe<Scalars["String"]["input"]>;
 };
 
+export type VideoProcessingAssessmentGql = {
+  __typename?: "VideoProcessingAssessmentGQL";
+  correctlyFlaggedFalsePositivePairs: Array<VideoProcessingAssessmentShotPairGql>;
+  correctlyFlaggedFalsePositives: Scalars["Int"]["output"];
+  extraFlaggedFalsePositiveShotIds: Array<Scalars["Int"]["output"]>;
+  extraFlaggedFalsePositives: Scalars["Int"]["output"];
+  groundTruthFalsePositiveCount: Scalars["Int"]["output"];
+  groundTruthPositiveShotCount: Scalars["Int"]["output"];
+  groundTruthProcessingId: Scalars["Int"]["output"];
+  groundTruthShotCount: Scalars["Int"]["output"];
+  groundTruthShotsFlaggedFalsePositive: Scalars["Int"]["output"];
+  groundTruthShotsFlaggedFalsePositivePairs: Array<VideoProcessingAssessmentShotPairGql>;
+  makeDisagreementPairs: Array<VideoProcessingAssessmentShotPairGql>;
+  makeDisagreements: Scalars["Int"]["output"];
+  matchedGroundTruthPairs: Array<VideoProcessingAssessmentShotPairGql>;
+  matchedGroundTruthShots: Scalars["Int"]["output"];
+  missedGroundTruthShotIds: Array<Scalars["Int"]["output"]>;
+  missedGroundTruthShots: Scalars["Int"]["output"];
+  processingId: Scalars["Int"]["output"];
+  sourceFalsePositiveCount: Scalars["Int"]["output"];
+  sourceShotCount: Scalars["Int"]["output"];
+  unexpectedShotIds: Array<Scalars["Int"]["output"]>;
+  unexpectedShots: Scalars["Int"]["output"];
+  unflaggedFalsePositivePairs: Array<VideoProcessingAssessmentShotPairGql>;
+  unflaggedFalsePositives: Scalars["Int"]["output"];
+};
+
+export type VideoProcessingAssessmentShotPairGql = {
+  __typename?: "VideoProcessingAssessmentShotPairGQL";
+  groundTruthShotId: Scalars["Int"]["output"];
+  processingShotId: Scalars["Int"]["output"];
+};
+
 export type VideoProcessingErrorGql = {
   __typename?: "VideoProcessingErrorGQL";
   endSegmentIndex?: Maybe<Scalars["Int"]["output"]>;
@@ -3490,6 +3701,8 @@ export type VideoProcessingGql = {
   errors: Array<VideoProcessingErrorGql>;
   framesProcessed?: Maybe<Scalars["Int"]["output"]>;
   id: Scalars["Int"]["output"];
+  labels: Array<Scalars["String"]["output"]>;
+  parentProcessingId?: Maybe<Scalars["Int"]["output"]>;
   progressPercentage?: Maybe<Scalars["Float"]["output"]>;
   status: ProcessingStatusEnum;
   statuses: Array<VideoProcessingStatusGql>;
@@ -3975,6 +4188,8 @@ export type GetDeployedConfigQuery = {
     firebase: boolean;
     minimumAllowedAppVersion: string;
     subscriptionGatingEnabled: boolean;
+    defaultAndroidRecordingFormat: StreamSegmentTypeEnum;
+    bucketUrl: string;
     bannerMessages: Array<{
       __typename?: "BannerGQL";
       color: string;
@@ -4209,6 +4424,21 @@ export type VideoCardFieldsFragment = {
   }>;
 };
 
+export type GetVideoFeedSessionCountQueryVariables = Exact<{
+  limit?: Scalars["Int"]["input"];
+  filters?: InputMaybe<VideoFilterInput>;
+  includePrivate?: InputMaybe<IncludePrivateEnum>;
+  feedInput?: InputMaybe<VideoFeedInputGql>;
+}>;
+
+export type GetVideoFeedSessionCountQuery = {
+  __typename?: "Query";
+  getFeedVideos: {
+    __typename?: "VideoHistoryGQL";
+    videos: Array<{ __typename?: "VideoGQL"; id: number; totalShots: number }>;
+  };
+};
+
 export type GetVideoFeedQueryVariables = Exact<{
   limit?: Scalars["Int"]["input"];
   after?: InputMaybe<Scalars["String"]["input"]>;
@@ -8216,6 +8446,8 @@ export const GetDeployedConfigDocument = gql`
         message
         priority
       }
+      defaultAndroidRecordingFormat
+      bucketUrl
     }
   }
 `;
@@ -8516,6 +8748,95 @@ export type GetFeedQueryResult = Apollo.QueryResult<
   GetFeedQuery,
   GetFeedQueryVariables
 >;
+export const GetVideoFeedSessionCountDocument = gql`
+  query GetVideoFeedSessionCount(
+    $limit: Int! = 100
+    $filters: VideoFilterInput = null
+    $includePrivate: IncludePrivateEnum = MINE
+    $feedInput: VideoFeedInputGQL = null
+  ) {
+    getFeedVideos(
+      limit: $limit
+      filters: $filters
+      includePrivate: $includePrivate
+      feedInput: $feedInput
+    ) {
+      videos {
+        id
+        totalShots
+      }
+    }
+  }
+`;
+
+/**
+ * __useGetVideoFeedSessionCountQuery__
+ *
+ * To run a query within a React component, call `useGetVideoFeedSessionCountQuery` and pass it any options that fit your needs.
+ * When your component renders, `useGetVideoFeedSessionCountQuery` returns an object from Apollo Client that contains loading, error, and data properties
+ * you can use to render your UI.
+ *
+ * @param baseOptions options that will be passed into the query, supported options are listed on: https://www.apollographql.com/docs/react/api/react-hooks/#options;
+ *
+ * @example
+ * const { data, loading, error } = useGetVideoFeedSessionCountQuery({
+ *   variables: {
+ *      limit: // value for 'limit'
+ *      filters: // value for 'filters'
+ *      includePrivate: // value for 'includePrivate'
+ *      feedInput: // value for 'feedInput'
+ *   },
+ * });
+ */
+export function useGetVideoFeedSessionCountQuery(
+  baseOptions?: Apollo.QueryHookOptions<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useQuery<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >(GetVideoFeedSessionCountDocument, options);
+}
+export function useGetVideoFeedSessionCountLazyQuery(
+  baseOptions?: Apollo.LazyQueryHookOptions<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useLazyQuery<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >(GetVideoFeedSessionCountDocument, options);
+}
+export function useGetVideoFeedSessionCountSuspenseQuery(
+  baseOptions?: Apollo.SuspenseQueryHookOptions<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useSuspenseQuery<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >(GetVideoFeedSessionCountDocument, options);
+}
+export type GetVideoFeedSessionCountQueryHookResult = ReturnType<
+  typeof useGetVideoFeedSessionCountQuery
+>;
+export type GetVideoFeedSessionCountLazyQueryHookResult = ReturnType<
+  typeof useGetVideoFeedSessionCountLazyQuery
+>;
+export type GetVideoFeedSessionCountSuspenseQueryHookResult = ReturnType<
+  typeof useGetVideoFeedSessionCountSuspenseQuery
+>;
+export type GetVideoFeedSessionCountQueryResult = Apollo.QueryResult<
+  GetVideoFeedSessionCountQuery,
+  GetVideoFeedSessionCountQueryVariables
+>;
 export const GetVideoFeedDocument = gql`
   query GetVideoFeed(
     $limit: Int! = 5

src/operations/config.gql

@@ -14,5 +14,7 @@ query getDeployedConfig {
       message
       priority
     }
+    defaultAndroidRecordingFormat
+    bucketUrl
   }
 }

src/operations/feed.gql

@@ -90,6 +90,26 @@ fragment VideoCardFields on VideoGQL {
   }
 }
 
+# Lightweight version of GetVideoFeed for counting sessions only
+query GetVideoFeedSessionCount(
+  $limit: Int! = 100
+  $filters: VideoFilterInput = null
+  $includePrivate: IncludePrivateEnum = MINE
+  $feedInput: VideoFeedInputGQL = null
+) {
+  getFeedVideos(
+    limit: $limit
+    filters: $filters
+    includePrivate: $includePrivate
+    feedInput: $feedInput
+  ) {
+    videos {
+      id
+      totalShots
+    }
+  }
+}
+
 query GetVideoFeed(
   $limit: Int! = 5
   $after: String = null

src/schema.gql

@@ -61,6 +61,8 @@ type Query {
     shotsOrdering: GetShotsOrdering = null
     limit: Int! = 500
     countRespectsLimit: Boolean! = false
+    processingId: Int = null
+    includeExcluded: Boolean! = false
   ): GetShotsResult!
   getShotsWithMetadata(
     filterInput: FilterInput!
@@ -68,12 +70,16 @@ type Query {
     shotsPagination: GetShotsPagination = null
     limit: Int! = 500
     countRespectsLimit: Boolean! = false
+    processingId: Int = null
+    includeExcluded: Boolean! = false
   ): GetShotsResult!
   getShots(
     filterInput: FilterInput!
     shotsPagination: GetShotsPagination = null
     limit: Int! = 500
     countRespectsLimit: Boolean! = false
+    processingId: Int = null
+    includeExcluded: Boolean! = false
   ): [ShotGQL!]!
   getShotsByIds(ids: [Int!]!): [ShotGQL!]!
   getUser(userId: Int!): UserGQL
@@ -97,6 +103,7 @@ type Query {
   ): UserRelationshipsResult!
   getAvailableSubscriptionOptions: StripeSubscriptionOptionsGQL!
   getUserSubscriptionStatus: UserSubscriptionStatusGQL!
+  getQuotaStatus: QuotaStatusGQL!
   getPlayTime(userId: Int!, filters: VideoFilterInput = null): UserPlayTimeGQL!
   getUserVideos(
     userId: Int = null
@@ -107,6 +114,11 @@ type Query {
   getUserTags(includeRetiredTags: Boolean = false): [TagGQL!]!
   getVideo(videoId: Int!, debuggingJson: JSON = null): VideoGQL!
   getVideos(videoIds: [Int!]!): [VideoGQL!]!
+  getVideoProcessings(videoId: Int!): [VideoProcessingGQL!]!
+  assessVideoProcessing(
+    processingId: Int!
+    groundTruthProcessingId: Int = null
+  ): VideoProcessingAssessmentGQL!
 }
 
 type AggregateResultGQL {
@@ -208,6 +220,7 @@ input FilterInput @oneOf {
   intendedPocketType: [PocketEnum!]
   shotDirection: [ShotDirectionEnum!]
   videoId: [Int!]
+  processingId: [Int!]
   userId: [Int!]
   runId: [Int!]
   username: [String!]
@@ -605,6 +618,8 @@ type IntPoint2D {
 
 type VideoProcessingGQL {
   id: Int!
+  parentProcessingId: Int
+  labels: [String!]!
   errors: [VideoProcessingErrorGQL!]!
   status: ProcessingStatusEnum!
   statuses: [VideoProcessingStatusGQL!]!
@@ -668,6 +683,8 @@ type DeployedConfigGQL {
   minimumAllowedAppVersion: String!
   subscriptionGatingEnabled: Boolean!
   bannerMessages: [BannerGQL!]!
+  defaultAndroidRecordingFormat: StreamSegmentTypeEnum!
+  bucketUrl: String!
 }
 
 type BannerGQL {
@@ -975,6 +992,17 @@ enum StripeSubscriptionStatusEnum {
   PAUSED
 }
 
+type QuotaStatusGQL {
+  tierName: String!
+  periodStart: DateTime!
+  periodEnd: DateTime!
+  durationUsedSeconds: Float!
+  durationLimitSeconds: Int
+  maxVideoDurationSeconds: Int
+  durationRemainingSeconds: Float
+  canUpload: Boolean!
+}
+
 type UserPlayTimeGQL {
   totalSeconds: Float!
 }
@@ -999,6 +1027,37 @@ scalar JSON
     url: "https://ecma-international.org/wp-content/uploads/ECMA-404_2nd_edition_december_2017.pdf"
   )
 
+type VideoProcessingAssessmentGQL {
+  processingId: Int!
+  groundTruthProcessingId: Int!
+  sourceShotCount: Int!
+  sourceFalsePositiveCount: Int!
+  groundTruthShotCount: Int!
+  groundTruthPositiveShotCount: Int!
+  groundTruthFalsePositiveCount: Int!
+  matchedGroundTruthShots: Int!
+  missedGroundTruthShots: Int!
+  unexpectedShots: Int!
+  correctlyFlaggedFalsePositives: Int!
+  unflaggedFalsePositives: Int!
+  extraFlaggedFalsePositives: Int!
+  groundTruthShotsFlaggedFalsePositive: Int!
+  makeDisagreements: Int!
+  matchedGroundTruthPairs: [VideoProcessingAssessmentShotPairGQL!]!
+  missedGroundTruthShotIds: [Int!]!
+  unexpectedShotIds: [Int!]!
+  correctlyFlaggedFalsePositivePairs: [VideoProcessingAssessmentShotPairGQL!]!
+  unflaggedFalsePositivePairs: [VideoProcessingAssessmentShotPairGQL!]!
+  extraFlaggedFalsePositiveShotIds: [Int!]!
+  groundTruthShotsFlaggedFalsePositivePairs: [VideoProcessingAssessmentShotPairGQL!]!
+  makeDisagreementPairs: [VideoProcessingAssessmentShotPairGQL!]!
+}
+
+type VideoProcessingAssessmentShotPairGQL {
+  processingShotId: Int!
+  groundTruthShotId: Int!
+}
+
 type Mutation {
   createBucketSet(params: CreateBucketSetInput!): BucketSetGQL!
   createRuleSet(name: String!, description: String = null): RuleSet!
@@ -1060,6 +1119,19 @@ type Mutation {
     shotId: Int!
     fieldsToEdit: EditableShotFieldInputGQL!
   ): EditShotReturn!
+  createGroundTruthShot(input: CreateGroundTruthShotInputGQL!): ShotGQL!
+  updateGroundTruthShotFrameRange(
+    shotId: Int!
+    startFrame: Int = null
+    endFrame: Int = null
+    notes: String = null
+  ): ShotGQL!
+  splitGroundTruthShot(
+    shotId: Int!
+    splitFrame: Int!
+    notes: String = null
+  ): SplitGroundTruthShotReturn!
+  deleteGroundTruthShot(shotId: Int!): DeleteGroundTruthShotReturn!
   getProfileImageUploadLink(
     fileExt: String = ".png"
   ): GetProfileUploadLinkReturn!
@@ -1080,7 +1152,13 @@ type Mutation {
   findPrerecordTableLayout(b64Image: String!, videoId: Int!): HomographyInfoGQL
   createUploadStream(
     videoMetadata: VideoMetadataInput!
+    expectedDurationSeconds: Float = null
   ): CreateUploadStreamReturn!
+  getOrCreateGroundTruthProcessing(
+    videoId: Int!
+    sourceProcessingId: Int = null
+  ): VideoProcessingGQL!
+  promoteGroundTruthProcessing(processingId: Int!): VideoProcessingGQL!
   getUploadLink(videoId: Int!, segmentIndex: Int!): GetUploadLinkReturn!
   getHlsInitUploadLink(videoId: Int!): GetUploadLinkReturn!
   setSegmentDuration(
@@ -1157,12 +1235,32 @@ input EditableShotFieldInputGQL {
   shotDirection: ShotDirectionEnum
   spinType: SpinTypeEnum
   targetPocketAngleDirection: ShotDirectionEnum
+  intendedBallId: Int
+  intendedPocketId: PocketIdentifier
+  pathMetadataIndex: Int
   make: Boolean
   backcut: Boolean
   excludeFromStats: Boolean
   notes: String
 }
 
+input CreateGroundTruthShotInputGQL {
+  processingId: Int!
+  startFrame: Int!
+  endFrame: Int!
+  make: Boolean
+  notes: String
+}
+
+type SplitGroundTruthShotReturn {
+  originalShot: ShotGQL!
+  createdShot: ShotGQL!
+}
+
+type DeleteGroundTruthShotReturn {
+  deletedShotId: Int!
+}
+
 type GetProfileUploadLinkReturn {
   value: UploadLinkGetProfileUploadLinkErrors!
 }