Skip to content

Commit 0301f4e

Browse files
committed
add comment to scenario, explaining model config
1 parent 56533db commit 0301f4e

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/rai_bench/rai_bench/benchmark_model.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -126,9 +126,11 @@ class Scenario(BaseModel):
126126

127127
task: Task
128128
scene_config: SimulationConfig
129-
model_config = ConfigDict(
130-
arbitrary_types_allowed=True
131-
) # pydantic does not support ABC classes
129+
# TODO pydantic does not support ABC classes
130+
# model_config field is used to define additional configurations to a model.
131+
# Param arbitrary_types_allowed=True, allows a field with
132+
# an abstract class like SimulationConfig.Without it pydantic will throw validation error
133+
model_config = ConfigDict(arbitrary_types_allowed=True)
132134

133135

134136
class Benchmark:

0 commit comments

Comments
 (0)