Remove builtin override and allow prints (#406)

Author: vishalbollu

pkg/workloads/cortex/lib/log.py

@@ -13,25 +13,23 @@
 # limitations under the License.
 
 import logging
+import sys
+
 from cortex.lib import stringify
 import datetime as dt
 
 
 class MyFormatter(logging.Formatter):
     converter = dt.datetime.fromtimestamp
 
-    def formatTime(self, record, datefmt=None):
+    def formatTime(self, record, datefmt):
         ct = self.converter(record.created)
-        if datefmt:
-            s = ct.strftime(datefmt)
-        else:
-            t = ct.strftime("%Y-%m-%d %H:%M:%S")
-            s = "%s,%03d" % (t, record.msecs)
+        s = ct.strftime(datefmt)
         return s
 
 
 logger = logging.getLogger("cortex")
-handler = logging.StreamHandler()
+handler = logging.StreamHandler(stream=sys.stdout)
 formatter = MyFormatter(
     fmt="%(asctime)s:%(name)s:%(levelname)s:%(message)s", datefmt="%Y-%m-%d,%H:%M:%S.%f"
 )

pkg/workloads/cortex/onnx_serve/api.py

@@ -29,13 +29,6 @@
 from cortex.lib.exceptions import CortexException, UserRuntimeException, UserException
 from cortex.lib.stringify import truncate
 
-
-def cortex_print(*args, **kwargs):
-    logger.info(*args)
-
-
-builtins.print = cortex_print
-
 logger = get_logger()
 logger.propagate = False  # prevent double logging (flask modifies root logger)
 

pkg/workloads/cortex/tf_api/api.py

@@ -34,13 +34,6 @@
 from cortex.lib.exceptions import UserRuntimeException, UserException
 from cortex.lib.stringify import truncate
 
-
-def cortex_print(*args, **kwargs):
-    logger.info(*args)
-
-
-builtins.print = cortex_print
-
 logger = get_logger()
 logger.propagate = False  # prevent double logging (flask modifies root logger)