make changes to support latency in trainer

Signed-off-by: ssrigiri1 <[email protected]>
numaproj · Nov 17, 2023 · 3b3ba9a · 3b3ba9a
1 parent 4970457
commit 3b3ba9a
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 9 deletions.
diff --git a/numalogic/__init__.py b/numalogic/__init__.py
@@ -12,6 +12,6 @@
 
 import logging
 
-print("init in numalogic")
+
 LOGGER = logging.getLogger(__name__)
 LOGGER.addHandler(logging.NullHandler())
diff --git a/numalogic/connectors/_config.py b/numalogic/connectors/_config.py
@@ -51,12 +51,12 @@ def __post_init__(self):
 
         if not self.aggregations:
             self.aggregations={
-                "agg_out": _agg.quantiles_doubles_sketch("valuesDoublesSketch", "agg0", 64)
+                "agg_out": _agg.quantiles_doubles_sketch("valuesDoublesSketch", "agg0", 128)
             }
         if  not self.post_aggregations:
             self.post_aggregations={
-                "p90": _post_agg.QuantilesDoublesSketchToQuantile(
-                    output_name="agg_out", field=postaggregator.Field("agg_out"), fraction=0.90
+                "p75": _post_agg.QuantilesDoublesSketchToQuantile(
+                    output_name="agg_out", field=postaggregator.Field("agg_out"), fraction=0.75
                 )
             }
 

diff --git a/numalogic/connectors/druid/_druid.py b/numalogic/connectors/druid/_druid.py
@@ -13,7 +13,7 @@
 from typing import Optional
 
 _LOGGER = logging.getLogger(__name__)
-TIMEOUT = 10000
+TIMEOUT = 25000
 
 
 # TODO: pass dictionary of keys and values as dict

diff --git a/numalogic/udfs/__init__.py b/numalogic/udfs/__init__.py
@@ -11,7 +11,6 @@
 from numalogic.udfs.preprocess import PreprocessUDF
 from numalogic.udfs.trainer import TrainerUDF
 
-print("__init__ in udfs")
 
 def set_logger() -> None:
     """Sets the logger for the UDFs."""
@@ -23,7 +22,6 @@ def set_logger() -> None:
         logging.getLogger("root").setLevel(logging.DEBUG)
 
 
-
 __all__ = [
     "NumalogicUDF",
     "PreprocessUDF",

diff --git a/numalogic/udfs/__main__.py b/numalogic/udfs/__main__.py
@@ -42,6 +42,4 @@ def start_server():
 
 
 if __name__ == "__main__":
-    print("main in udfs")
-    print("sys.argv", sys.argv)
     start_server()