[develop < T0034-GA] Add Field to models (#122)

* Add Field to models * Remove Field from gqlalchemy init
memgraph · May 20, 2022 · 082ca9a · 082ca9a
1 parent 479e137
commit 082ca9a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/gqlalchemy/__init__.py b/gqlalchemy/__init__.py
@@ -23,6 +23,7 @@
     Node,
     Path,
     Relationship,
+    Field,
 )
 from .disk_storage import SQLitePropertyDatabase  # noqa F401
 from .query_builder import (  # noqa F401
@@ -47,7 +48,7 @@
 )
 
 from .exceptions import GQLAlchemyWarning, GQLAlchemyError  # noqa F401
-from pydantic import Field, validator  # noqa F401
+from pydantic import validator  # noqa F401
 import warnings
 
 warnings.filterwarnings("once", category=GQLAlchemyWarning)

diff --git a/gqlalchemy/models.py b/gqlalchemy/models.py
@@ -18,7 +18,7 @@
 from collections import defaultdict
 from dataclasses import dataclass
 from typing import Any, Dict, List, Iterable, Optional, Set, Tuple, Union
-from pydantic import BaseModel, PrivateAttr, Extra
+from pydantic import BaseModel, PrivateAttr, Extra, Field  # noqa F401
 
 from .exceptions import (
     GQLAlchemyError,