Correctly handle injection into dataclasses

ZechCodes · Nov 1, 2023 · be95663 · be95663
1 parent b38c874
commit be95663
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/bevy/injectors/classes.py b/bevy/injectors/classes.py
@@ -1,4 +1,5 @@
 import sys
+import dataclasses
 from inspect import get_annotations
 from typing import Any, Generic, Type, TypeVar
 
@@ -22,6 +23,9 @@ def __init__(self):
 
     def __get__(self, instance: object, owner: Type):
         if instance is None:
+            if hasattr(owner, dataclasses._PARAMS):
+                return dataclasses.field(default_factory=self._inject_dependency)
+
             return self
 
         return self._inject_dependency()