fix(gms): make get return entity type (#2739)

datahub-project · Jun 23, 2021 · 7615ddb · 7615ddb
1 parent 91f5d4f
commit 7615ddb
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/gms/client/src/main/java/com/linkedin/entity/client/EntityClient.java b/gms/client/src/main/java/com/linkedin/entity/client/EntityClient.java
@@ -8,7 +8,6 @@
 import com.linkedin.data.template.DataTemplateUtil;
 import com.linkedin.data.template.DynamicRecordMetadata;
 import com.linkedin.data.template.FieldDef;
-import com.linkedin.data.template.RecordTemplate;
 import com.linkedin.data.template.StringArray;
 import com.linkedin.entity.EntitiesDoAutocompleteRequestBuilder;
 import com.linkedin.entity.EntitiesDoBrowseRequestBuilder;
@@ -78,7 +77,7 @@ private <T> Response<T> sendClientRequest(Request<T> request) throws RemoteInvoc
     }
 
     @Nonnull
-    public RecordTemplate get(@Nonnull final Urn urn) throws RemoteInvocationException {
+    public Entity get(@Nonnull final Urn urn) throws RemoteInvocationException {
         final GetRequest<Entity> getRequest = ENTITIES_REQUEST_BUILDERS.get()
                 .id(urn.toString())
                 .build();