remove references to worker in constructor and imports (aws#262)

Author: rahul003

tornasole/tensorflow/keras.py

@@ -2,7 +2,6 @@
 from .collection import *
 from tornasole.core.hook import BaseHook
 from tornasole.core.save_config import SaveConfig
-from tornasole.core.json_config import TORNASOLE_CONFIG_DEFAULT_WORKER_NAME
 
 
 DEFAULT_INCLUDE_COLLECTIONS=['weights', 'gradients', 'metrics']
@@ -11,7 +10,6 @@
 class TornasoleHook(keras.callbacks.Callback, BaseHook):
     def __init__(self, out_dir,
                  dry_run=False,
-                 worker=TORNASOLE_CONFIG_DEFAULT_WORKER_NAME,
                  reduction_config=None,
                  save_config=SaveConfig(),
                  include_regex=None,
@@ -28,7 +26,6 @@ def __init__(self, out_dir,
                          default_include_collections=DEFAULT_INCLUDE_COLLECTIONS,
                          out_dir=out_dir,
                          dry_run=dry_run,
-                         worker=worker,
                          reduction_config=reduction_config,
                          save_config=save_config,
                          include_regex=None,

tornasole/xgboost/hook.py

@@ -8,9 +8,7 @@
 from tornasole.core.save_config import SaveConfig
 from tornasole.core.hook import CallbackHook
 from tornasole.core.access_layer.utils import training_has_ended
-from tornasole.core.json_config import (
-    TORNASOLE_CONFIG_DEFAULT_WORKER_NAME,
-    create_hook_from_json_config)
+from tornasole.core.json_config import create_hook_from_json_config
 
 from .collection import get_collection_manager
 from .utils import validate_data_file_path, get_content_type, get_dmatrix