search

AllenNeuralDynamics / aind-behavior-curriculum

Starter repository for behavior base primitives.
https://aind-behavior-curriculum.readthedocs.io
MIT License
1 stars 0 forks source link

Fix Union[Stage] signature in inherited curriculum model #19

Closed bruno-f-cruz closed 2 months ago

bruno-f-cruz commented 2 months ago

https://github.com/AllenNeuralDynamics/aind-behavior-curriculum/blob/9d77be9216982518391e5de8dce538426fcf2b93/examples/example_project/curriculum.py#L163

Worst case scenario this should be refactored to be a discriminated union (see more details here https://github.com/AllenNeuralDynamics/aind-data-schema/issues/518).

Best case, see if there is a way to not have to pass the types.

bruno-f-cruz commented 2 months ago

Fixed by #20