From 7f3d90750b598ef778c96b3632ede9fbbc40d1a6 Mon Sep 17 00:00:00 2001 From: edwardchoi Date: Fri, 24 Jan 2020 16:54:50 -0800 Subject: [PATCH] Simple bug fix. --- .../graph_convolutional_transformer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/graph-convolutional-transformer/graph_convolutional_transformer.py b/graph-convolutional-transformer/graph_convolutional_transformer.py index 2482325..ebc38aa 100644 --- a/graph-convolutional-transformer/graph_convolutional_transformer.py +++ b/graph-convolutional-transformer/graph_convolutional_transformer.py @@ -597,7 +597,8 @@ def parse(self, serialized_examples): serialized_examples, context_features=self.context_features_config, sequence_features=self.sequence_features_config) - return (batch_context, batch_sequence) + + return (batch_context, batch_sequence) class EHRTransformer(object):