The update to pydantic V2 deleted a feature of recipe export that would resolve clashing stage names. This caused tests/sparseml/transformers/obcq/test_repeats.py:test_consecutive_runs() to fail because we apply two recipes with stages labeled test_stage, and the stage name keys must be unique for serialization.
Fixed by adding the old indexing code back in to the refactored export code and the unit test now passes
The update to pydantic V2 deleted a feature of recipe export that would resolve clashing stage names. This caused
tests/sparseml/transformers/obcq/test_repeats.py:test_consecutive_runs()to fail because we apply two recipes with stages labeledtest_stage, and the stage name keys must be unique for serialization.Fixed by adding the old indexing code back in to the refactored export code and the unit test now passes