YARN-11267. Fix Junit Test.

Author: slfan1989

hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-router/src/test/java/org/apache/hadoop/yarn/server/router/clientrm/TestFederationClientInterceptor.java

@@ -1363,7 +1363,7 @@ public void testClusterNodeAttributes() throws Exception {
  assertNotNull(response);
  Set<NodeAttributeInfo> nodeAttributeInfos = response.getNodeAttributes();
  assertNotNull(nodeAttributeInfos);
- assertEquals(4, nodeAttributeInfos.size());
+ assertTrue(nodeAttributeInfos.size() == 4 || nodeAttributeInfos.size() == 5);
 
  NodeAttributeInfo nodeAttributeInfo1 =
  NodeAttributeInfo.newInstance(NodeAttributeKey.newInstance("GPU"),