Fix java tests

Signed-off-by: Kevin Zhang <[email protected]>
feast-dev · Apr 26, 2022 · fbf27be · fbf27be
1 parent 7d3975f
commit fbf27be
Showing 1 changed file with 8 additions and 8 deletions.
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();
   }