Update file that was modified in #5234

microsoft · Apr 4, 2024 · 91789b5 · 91789b5
1 parent cf7bee9
commit 91789b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/inference/v2/ragged/test_manager_configs.py b/tests/unit/inference/v2/ragged/test_manager_configs.py
@@ -5,7 +5,7 @@
 
 import pytest
 
-from deepspeed.pydantic_v1 import ValidationError
+from pydantic import ValidationError
 
 from deepspeed.inference.v2.ragged import DSStateManagerConfig