From 06b71e985094f5e0bb2de60194890bdae74b5aef Mon Sep 17 00:00:00 2001
From: Andrew Wen <2060andrew@gmail.com>
Date: Fri, 18 Feb 2022 08:34:50 -0600
Subject: [PATCH] Use base values for note_id to not explicitly typecast

---
 pom.xml                                                         | 2 +-
 .../ohnlp/medtagger/backbone/MedTaggerBackboneTransform.java    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/pom.xml b/pom.xml
index 63560ec..d8d3439 100644
--- a/pom.xml
+++ b/pom.xml
@@ -6,7 +6,7 @@
 
     <groupId>org.ohnlp.medtagger</groupId>
     <artifactId>medtagger</artifactId>
-    <version>1.0.28</version>
+    <version>1.0.29</version>
     <description>The MedTagger biomedical information extraction pipeline</description>
 
 
diff --git a/src/main/java/org/ohnlp/medtagger/backbone/MedTaggerBackboneTransform.java b/src/main/java/org/ohnlp/medtagger/backbone/MedTaggerBackboneTransform.java
index 05862c0..bf36529 100644
--- a/src/main/java/org/ohnlp/medtagger/backbone/MedTaggerBackboneTransform.java
+++ b/src/main/java/org/ohnlp/medtagger/backbone/MedTaggerBackboneTransform.java
@@ -168,7 +168,7 @@ public void processElement(@Element Row input, OutputReceiver<Row> output) {
             List<Schema.Field> fields = new LinkedList<>(input.getSchema().getFields());
             fields.add(Schema.Field.of("nlp_output_json", Schema.FieldType.STRING));
             Schema schema = Schema.of(fields.toArray(new Schema.Field[0]));
-            int id = input.getInt32("note_id");
+            Object id = input.getBaseValue("note_id");
             String text = input.getString(this.textField);
             cas.reset();
             cas.setDocumentText(text);