Improve logging

shepherd/runner/base_runner.py

@@ -43,8 +43,6 @@ def __init__(self, config_path: str, port: int, stream_name: str):
         """Create new :py:class:`Runner`."""
         logging.info('Creating runner from `%s` listening on port %s', config_path, port)
 
-        self._logger = logging
-
         # bind to the socket
         self._port = port
         self._socket = None

shepherd/runner/runner_entry_point.py

@@ -1,10 +1,12 @@
 import asyncio
+import importlib
 import logging
 import os
 import sys
 from argparse import ArgumentParser
 
 __all__ = ['main']
+from typing import Tuple, Optional
 
 from shepherd.runner import BaseRunner
 
@@ -33,7 +35,11 @@ def main() -> None:
 
     args = create_argparser().parse_args()
 
-    runner = BaseRunner(args.config_path, args.port, args.stream)
+    module_name = "model.isletnet_runner"
+    class_name = "IsletnetRunner"
+
+    _module = importlib.import_module(module_name)
+    runner = getattr(_module, class_name)(args.config_path, args.port, args.stream)
 
     # listen for input messages
     asyncio.run(runner.process_all())