Simple bug fix.

Google-Health · Jan 25, 2020 · 7f3d907 · 7f3d907
1 parent 56ecb60
commit 7f3d907
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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):