From 37d40a38bc3c578f942602519575a89a11a62755 Mon Sep 17 00:00:00 2001 From: Shane Smiskol Date: Thu, 8 Aug 2024 20:53:30 -0700 Subject: [PATCH] safe --- selfdrive/car/data_test_kinda_works_chatgpt.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/selfdrive/car/data_test_kinda_works_chatgpt.py b/selfdrive/car/data_test_kinda_works_chatgpt.py index 83804c0670..9b5d6d8d1a 100644 --- a/selfdrive/car/data_test_kinda_works_chatgpt.py +++ b/selfdrive/car/data_test_kinda_works_chatgpt.py @@ -1,13 +1,15 @@ # import attr from dataclasses import dataclass, field -from typing import Any, List -from enum import auto auto_obj = object() -def apply_auto_defaults(cls): - cls_annotations = cls.__annotations__ +def auto_factory(): + return auto_obj + + +def apply_auto_factory(cls): + cls_annotations = cls.__dict__.get('__annotations__', {}) for name, typ in cls_annotations.items(): current_value = getattr(cls, name, None) if current_value is auto_obj: @@ -15,12 +17,8 @@ def apply_auto_defaults(cls): return cls -def auto_factory(): - return auto_obj - - @dataclass -@apply_auto_defaults +@apply_auto_factory class CarControl: enabled: bool = auto_factory() pts: list[int] = auto_factory()