more consistent logging; use logger. instead of logging. where appropriate

1 files changed, 7 insertions, 4 deletions

diff --git a/lib/automata.py b/lib/automata.py
index b3318e0..69b3969 100755
--- a/lib/automata.py
+++ b/lib/automata.py
@@ -3,11 +3,14 @@
 from .functions import AnalyticFunction, NormalizationFunction
 from .utils import is_numeric
 import itertools
+import logging
 import numpy as np
 import json
 import queue
 import yaml
 
+logger = logging.getLogger(__name__)
+
 
 def _dict_to_list(input_dict: dict) -> list:
     return [input_dict[x] for x in sorted(input_dict.keys())]
@@ -1305,8 +1308,8 @@ class PTA:
                                 "power"
                             ]
                 except KeyError:
-                    print(
-                        "[W] skipping model update of state {} due to missing data".format(
+                    logger.warning(
+                        "skipping model update of state {} due to missing data".format(
                             state.name
                         )
                     )
@@ -1353,8 +1356,8 @@ class PTA:
                         "timeout"
                     ]
             except KeyError:
-                print(
-                    "[W] skipping model update of transition {} due to missing data".format(
+                logger.warning(
+                    "skipping model update of transition {} due to missing data".format(
                         transition.name
                     )
                 )