From dfe620eafcc16abb847a2e7531be508e2385257a Mon Sep 17 00:00:00 2001
From: Victor Sonck <victor.sonck@gmail.com>
Date: Wed, 1 Jun 2022 17:14:31 +0200
Subject: [PATCH] Fix bug where custom buckets are not passed through correctly

---
 clearml_serving/statistics/metrics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/clearml_serving/statistics/metrics.py b/clearml_serving/statistics/metrics.py
index befb49e..0eed08b 100644
--- a/clearml_serving/statistics/metrics.py
+++ b/clearml_serving/statistics/metrics.py
@@ -323,7 +323,7 @@ class StatisticsController(object):
         metric_cls = self._metric_type_class.get(metric_.type)
         if not metric_cls:
             return None
-        if metric_cls in (Histogram, EnumHistogram):
+        if metric_cls in (ScalarHistogram, EnumHistogram):
             return metric_cls(
                 name=name,
                 documentation="User defined metric {}".format(metric_.type),