avoid reimport, rename var

fetchai · Jul 3, 2023 · d5b85f2 · d5b85f2
1 parent 94ee30d
commit d5b85f2
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/uagents/models.py b/src/uagents/models.py
@@ -3,7 +3,6 @@
 
 from pydantic import BaseModel
 from pydantic.schema import model_schema, default_ref_template
-from pydantic.main import BaseModel
 
 
 class Model(BaseModel):
@@ -39,8 +38,8 @@ def _refresh_schema_cache(
         by_alias: bool = True,
         ref_template: str = default_ref_template,
     ):
-        s = model_schema(model, by_alias, ref_template)
-        model.__schema_cache__[(True, default_ref_template)] = s
+        schema = model_schema(model, by_alias, ref_template)
+        model.__schema_cache__[(True, default_ref_template)] = schema
 
     @staticmethod
     def build_schema_digest(model: Union["Model", Type["Model"]]) -> str: