added extra metrics with apf 1.0.1

alercebroker · Oct 14, 2020 · a992e6b · a992e6b
1 parent 375060c
commit a992e6b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -2,7 +2,7 @@ numpy
 Cython
 matplotlib
 GPy
-apf-base==1.0.0
+apf-base==1.0.1
 -e git+https://github.com/alercebroker/mhps.git#egg=mhps
 -e git+https://github.com/alercebroker/db-plugins.git#egg=db-plugins
 -e git+https://github.com/alercebroker/late_classifier.git#egg=late-classifier

diff --git a/settings.py b/settings.py
@@ -262,6 +262,12 @@
 
 METRICS_CONFIG = {
     "CLASS": "apf.metrics.KafkaMetricsProducer",
+    "EXTRA_METRICS": [
+        {"key": "candid", "format": lambda x: str(x)},
+        "oid",
+        {"key": "detections", "format": lambda x: len(x), "alias": "n_det"},
+        {"key": "non_detections", "format": lambda x: len(x), "alias": "n_non_det"}
+    ],
     "PARAMS": {
         "PARAMS": {
             "bootstrap.servers": os.environ["METRICS_HOST"],