Skip to content

Commit f1d58ab

Browse files
committed
feat: agent visualization
1 parent 0752cbf commit f1d58ab

File tree

3 files changed

+530
-32
lines changed

3 files changed

+530
-32
lines changed

dgp/constants.py

+10
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
Common DGP constants. Constants here only depened on dgp.proto to avoid circular imports.
44
"""
55
from collections import OrderedDict
6+
from dataclasses import dataclass
67

78
from dgp.proto import annotations_pb2, dataset_pb2, features_pb2
89

@@ -34,3 +35,12 @@
3435
FEATURE_TYPE_ID_TO_KEY = OrderedDict({v: k for k, v in FEATURE_KEY_TO_TYPE_ID.items()})
3536
# String identifiers for feature types
3637
ALL_FEATURE_TYPES = tuple(FEATURE_KEY_TO_TYPE_ID.keys())
38+
39+
40+
@dataclass
41+
class Vehicle:
42+
vehicle_name: str
43+
vehicle_length: float
44+
vehicle_width: float
45+
vehicle_height: float
46+
vehicle_applanix_origin_to_r_bumper: float

0 commit comments

Comments
 (0)