Fix unit test failure

pytorch · Nov 8, 2023 · 7565730 · 7565730
1 parent 69574b7
commit 7565730
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/frontend/server/src/test/java/org/pytorch/serve/metrics/MetricCacheTest.java b/frontend/server/src/test/java/org/pytorch/serve/metrics/MetricCacheTest.java
@@ -76,7 +76,6 @@ public void testMetricCacheAddAutoDetectMetricBackendWithoutType() {
                         new Dimension("Level", "model"));
         metricCache.addAutoDetectMetricBackend(metric);
         IMetric cachedMetric = metricCache.getMetricBackend("TestMetricWithoutType");
-        Assert.assertEquals(PrometheusCounter.class, cachedMetric.getClass());
         Assert.assertEquals(cachedMetric.type, MetricBuilder.MetricType.COUNTER);
         Assert.assertEquals(cachedMetric.name, "TestMetricWithoutType");
         Assert.assertEquals(cachedMetric.unit, "count");
@@ -98,7 +97,6 @@ public void testMetricCacheAddAutoDetectMetricBackendWithType() {
                         new Dimension("Level", "model"));
         metricCache.addAutoDetectMetricBackend(metric);
         IMetric cachedMetric = metricCache.getMetricBackend("TestMetricWithType");
-        Assert.assertEquals(PrometheusGauge.class, cachedMetric.getClass());
         Assert.assertEquals(cachedMetric.type, MetricBuilder.MetricType.GAUGE);
         Assert.assertEquals(cachedMetric.name, "TestMetricWithType");
         Assert.assertEquals(cachedMetric.unit, "count");