Upgrade Pydantic to >= 2.0.0 (#607)

bigscience-workshop · Aug 25, 2024 · 22afba6 · 22afba6 · adnanbaqi · Nov 25, 2024
1 parent c68c1c3
commit 22afba6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -39,7 +39,7 @@ install_requires =
     tokenizers>=0.13.3
     transformers==4.43.1  # if you change this, please also change version assert in petals/__init__.py
     speedtest-cli==2.1.3
-    hivemind @ git+https://github.com/learning-at-home/hivemind.git@f98acba7aa3cf93717c40ca95d0b4e5b5742414c
+    hivemind @ git+https://github.com/learning-at-home/hivemind.git@213bff98a62accb91f254e2afdccbf1d69ebdea9
     tensor_parallel==1.0.23
     humanfriendly
     async-timeout>=4.0.2

diff --git a/src/petals/data_structures.py b/src/petals/data_structures.py
@@ -2,7 +2,7 @@
 from enum import Enum
 from typing import Any, Dict, Optional, Sequence, Tuple
 
-import pydantic
+import pydantic.v1 as pydantic
 from hivemind import PeerID
 from hivemind.moe.expert_uid import ExpertUID