Added null check for arguments

myui · Oct 18, 2013 · e7759d9 · e7759d9
1 parent 138cd80
commit e7759d9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/main/hivemall/ftvec/hashing/ArrayPrefixedHashValuesUDF.java b/src/main/hivemall/ftvec/hashing/ArrayPrefixedHashValuesUDF.java
@@ -28,6 +28,13 @@
 public class ArrayPrefixedHashValuesUDF extends UDF {
 
     public List<String> evaluate(List<String> values, String prefix) {
+        if(values == null) {
+            return null;
+        }
+        if(prefix == null) {
+            prefix = "";
+        }
+
         List<Integer> hashValues = ArrayHashValuesUDF.hashValues(values, null, MurmurHash3UDF.DEFAULT_NUM_FEATURES);
         final int len = hashValues.size();
         final String[] stringValues = new String[len];

diff --git a/target/hivemall.jar b/target/hivemall.jar