pydantic models refactor + add to autodoc + fixes

2025-03-18 12:56:12 -04:00
parent 81f2203151
commit a563815fe7
28 changed files with 1014 additions and 951 deletions
--- a/tests/utils/test_models.py
+++ b/tests/utils/test_models.py
@@ -119,7 +119,7 @@ class TestModelsUtils:

    def test_message_property_mapping(self):
        """Test message property mapping configuration validation"""
-        from axolotl.utils.config.models.input.v0_4_1 import SFTDataset
+        from axolotl.utils.schemas.datasets import SFTDataset

        # Test legacy fields are mapped orrectly
        dataset = SFTDataset(