1 files changed, 7 insertions, 1 deletions
diff --git a/cli/app/sql/models/graph.py b/cli/app/sql/models/graph.py
index 59d55c8..ef5a817 100644
--- a/cli/app/sql/models/graph.py
+++ b/cli/app/sql/models/graph.py
@@ -5,7 +5,6 @@ from wtforms_alchemy import ModelForm
 
 from app.sql.common import db, Base, Session
 
-from app.utils.file_utils import sha256_tree
 from app.settings import app_cfg
 
 from os.path import join
@@ -33,3 +32,10 @@ class Graph(Base):
       'created_at': self.created_at,
       'updated_at': self.updated_at,
     }
+
+class GraphForm(ModelForm):
+  class Meta:
+    model = Graph
+    exclude = ['settings', 'created_at', 'updated_at']
+  def get_session():
+    return Session()