Fix java tests

kevjumba · kevjumba · commit fbf27be363f1 · 2022-04-26T10:47:34.000-07:00
Signed-off-by: Kevin Zhang &lt;kzhang@tecton.ai&gt;
diff --git a/java/serving/src/test/java/feast/serving/util/DataGenerator.java b/java/serving/src/test/java/feast/serving/util/DataGenerator.java
@@ -126,11 +126,11 @@ public static EntityProto.EntitySpecV2 createEntitySpecV2(
   }
 
   public static FeatureProto.FeatureSpecV2 createFeatureSpecV2(
-      String name, ValueProto.ValueType.Enum valueType, Map<String, String> labels) {
+      String name, ValueProto.ValueType.Enum valueType, Map<String, String> tags) {
     return FeatureProto.FeatureSpecV2.newBuilder()
         .setName(name)
         .setValueType(valueType)
-        .putAllLabels(labels)
+        .putAllTags(tags)
         .build();
   }
 
@@ -140,7 +140,7 @@ public static FeatureTableSpec createFeatureTableSpec(
       List<String> entities,
       Map<String, ValueProto.ValueType.Enum> features,
       int maxAgeSecs,
-      Map<String, String> labels) {
+      Map<String, String> tags) {
 
     return FeatureTableSpec.newBuilder()
         .setName(name)
@@ -152,7 +152,7 @@ public static FeatureTableSpec createFeatureTableSpec(
                         FeatureSpecV2.newBuilder()
                             .setName(entry.getKey())
                             .setValueType(entry.getValue())
-                            .putAllLabels(labels)
+                            .putAllTags(tags)
                             .build())
                 .collect(Collectors.toList()))
         .setMaxAge(Duration.newBuilder().setSeconds(3600).build())
@@ -169,7 +169,7 @@ public static FeatureTableSpec createFeatureTableSpec(
                         .setUri("/dev/null")
                         .build())
                 .build())
-        .putAllLabels(labels)
+        .putAllLabels(tags)
         .build();
   }
 
@@ -178,7 +178,7 @@ public static FeatureTableSpec createFeatureTableSpec(
       List<String> entities,
       ImmutableMap<String, ValueProto.ValueType.Enum> features,
       int maxAgeSecs,
-      Map<String, String> labels) {
+      Map<String, String> tags) {
 
     return FeatureTableSpec.newBuilder()
         .setName(name)
@@ -190,11 +190,11 @@ public static FeatureTableSpec createFeatureTableSpec(
                         FeatureSpecV2.newBuilder()
                             .setName(entry.getKey())
                             .setValueType(entry.getValue())
-                            .putAllLabels(labels)
+                            .putAllTags(tags)
                             .build())
                 .collect(Collectors.toList()))
         .setMaxAge(Duration.newBuilder().setSeconds(maxAgeSecs).build())
-        .putAllLabels(labels)
+        .putAllLabels(tags)
         .build();
   }
 
