rename HorovodRunnerBase to HorovodRunner (#165)

mengxr · web-flow · commit a20740c17862 · 2018-11-13T17:45:42.000-08:00
diff --git a/python/docs/sparkdl.horovod.rst b/python/docs/sparkdl.horovod.rst
diff --git a/python/sparkdl/__init__.py b/python/sparkdl/__init__.py
@@ -20,7 +20,7 @@
 from .transformers.tf_tensor import TFTransformer
 from .transformers.utils import imageInputPlaceholder
 from .estimators.keras_image_file_estimator import KerasImageFileEstimator
-from .horovod.runner_base import HorovodRunnerBase as HorovodRunner
+from .horovod.runner_base import HorovodRunner
 
 __all__ = [
     'TFImageTransformer', 'TFInputGraph', 'TFTransformer', 'DeepImagePredictor',
diff --git a/python/sparkdl/horovod/runner_base.py b/python/sparkdl/horovod/runner_base.py
@@ -22,7 +22,7 @@
 
 from sparkdl.param import keyword_only
 
-class HorovodRunnerBase(object):
+class HorovodRunner(object):
     """
     HorovodRunner runs distributed deep learning training jobs using Horovod.
 
diff --git a/python/tests/horovod/runner_base_test.py b/python/tests/horovod/runner_base_test.py
@@ -18,7 +18,7 @@
 import inspect
 import unittest
 
-from sparkdl.horovod.runner_base import HorovodRunnerBase as HorovodRunner
+from sparkdl.horovod.runner_base import HorovodRunner
 
 
 class HorovodRunnerBaseTestCase(unittest.TestCase):
