Feature/add buffers to detail view (#65)

Re-adds buffers / stack trace to Operation detail view.

Author: GregHattJr

backend/models.py

@@ -117,7 +117,7 @@ class Tensor(db.Model):
 
 class Buffer(db.Model):
     __table__ = buffers
-
+    device= relationship("Device")
 
 class InputTensor(db.Model):
     __table__ = input_tensors
@@ -139,6 +139,7 @@ class Operation(db.Model):
     inputs = db.relationship("InputTensor", lazy="joined")
     outputs = db.relationship("OutputTensor", lazy="joined")
     stack_trace = db.relationship("StackTrace")
+    buffers = db.relationship("Buffer")
 
 
 class OperationArgument(db.Model):

backend/schemas.py

@@ -81,7 +81,6 @@ class Meta:
     max_size_per_bank = ma.auto_field()
     buffer_type = ma.auto_field()
     device_id = ma.Function(lambda obj: obj.device.device_id)
-    operation_id = ma.Function(lambda obj: obj.operation.operation_id)
 
 
 class OperationSchema(ma.SQLAlchemySchema):
@@ -96,7 +95,12 @@ class Meta:
     outputs = ma.List(ma.Nested(OutputTensorSchema()))
     inputs = ma.List(ma.Nested(InputTensorSchema()))
     arguments = ma.List(ma.Nested(OperationArgumentsSchema()))
+    stack_trace = ma.Method("get_stack_trace")
 
+    def get_stack_trace(self, obj):
+        if obj.stack_trace and len(obj.stack_trace):
+            return next(x for x in obj.stack_trace).stack_trace
+        return ""
 
 # Filesystem Schemas
 class RemoteConnectionSchema(ma.Schema):