Update pydantic support

neuralmagic / sparsezoo

Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes

Apache License 2.0

366 stars 25 forks source link

Closed rahul-tuli closed 5 months ago

rahul-tuli commented 5 months ago

This PR upgrades pydantic support from v1 to v2

The changes include:

Using pydantic's bump-pydantic tool for the initial migration
Updating typing of class vars
Left FeatureStatus table to use BaseModel from pydantic.v1 (shipped as a part of v2) as it is a low importance feature

Test method:

===================== 345 passed, 126 warnings in 656.59s (0:10:56) ======================

rahul-tuli commented 5 months ago

Note: We should wait until sparseml pydantic upgrade is ready to land https://github.com/neuralmagic/sparseml/pull/2248