Repair policy types in TOSCA service template

[policy/models.git] / models-tosca / src / main / java / org / onap / policy / models / tosca / authorative / concepts / ToscaServiceTemplate.java

diff --git a/models-tosca/src/main/java/org/onap/policy/models/tosca/authorative/concepts/ToscaServiceTemplate.java b/models-tosca/src/main/java/org/onap/policy/models/tosca/authorative/concepts/ToscaServiceTemplate.java
index 21b15a8..1739196 100644 (file)
--- a/models-tosca/src/main/java/org/onap/policy/models/tosca/authorative/concepts/ToscaServiceTemplate.java
+++ b/models-tosca/src/main/java/org/onap/policy/models/tosca/authorative/concepts/ToscaServiceTemplate.java
@@ -25,7 +25,6 @@ package org.onap.policy.models.tosca.authorative.concepts;
 
 import com.google.gson.annotations.SerializedName;
 import io.swagger.annotations.ApiModelProperty;
-import java.util.List;
 import java.util.Map;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
@@ -49,17 +48,17 @@ public class ToscaServiceTemplate extends ToscaEntity {
 
     @ApiModelProperty(name = "policy_types")
     @SerializedName("policy_types")
-    private List<Map<String, ToscaPolicyType>> policyTypes;
+    private Map<String, ToscaPolicyType> policyTypes;
 
     @ApiModelProperty(name = "data_types")
     @SerializedName("data_types")
-    private List<Map<String, ToscaDataType>> dataTypes;
+    private Map<String, ToscaDataType> dataTypes;
 
     public Map<ToscaEntityKey, ToscaPolicyType> getPolicyTypesAsMap() {
-        return ToscaEntity.getEntityListMapAsMap(policyTypes);
+        return ToscaEntity.getEntityMapAsMap(policyTypes);
     }
 
     public Map<ToscaEntityKey, ToscaDataType> getDataTypesAsMap() {
-        return ToscaEntity.getEntityListMapAsMap(dataTypes);
+        return ToscaEntity.getEntityMapAsMap(dataTypes);
     }
 }
\ No newline at end of file