Remotemixin to operator (towhee-io#1875)

Signed-off-by: shiyu22 <[email protected]> Signed-off-by: shiyu22 <[email protected]>
CV-IP · Aug 26, 2022 · 77d1e54 · 77d1e54
1 parent 776f4b9
commit 77d1e54
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 78 deletions.
diff --git a/tests/unittests/functional/test_mixins.py b/tests/unittests/functional/test_mixins.py
@@ -31,7 +31,6 @@
 import towhee.functional.mixins.state
 import towhee.functional.mixins.serve
 import towhee.functional.mixins.config
-import towhee.functional.mixins.remote
 import towhee.functional.mixins.data_processing
 import towhee.functional.mixins.safe
 import towhee.functional.mixins.list
@@ -57,7 +56,6 @@ def load_tests(loader, tests, ignore):
             towhee.functional.mixins.serve,
             towhee.functional.mixins.column,
             towhee.functional.mixins.config,
-            towhee.functional.mixins.remote,
             towhee.functional.mixins.list,
             towhee.functional.mixins.data_processing,
             towhee.functional.mixins.stream,
@@ -314,20 +312,5 @@ def inner_distance1(query, data):
             )
 
 
-class TestRemoteMixin(unittest.TestCase):
-    """
-    Unit test for RemoteMixin
-    """
-    def test_remote(self):
-        dc_1 = DataCollection(((1, 2, 3, 4, 5, 6), (2, 3, 4, 5, 6, 7)))
-        remote = dc_1.remote('127.0.0.1:8001', mode='infer', protocol='grpc')
-        self.assertEqual(remote[0], None)
-
-    def test_remote_error(self):
-        dc_2 = DataCollection(((1, 2, 3, 4, 5, 6), (2, 3, 4, 5, 6, 7)))
-        remote = dc_2.remote('127.0.0.1:8000', mode='async_infer', protocol='http')
-        self.assertEqual(remote[0], None)
-
-
 if __name__ == '__main__':
     unittest.main()
diff --git a/towhee/functional/mixins/__init__.py b/towhee/functional/mixins/__init__.py
@@ -26,7 +26,6 @@
 from .dag import DagMixin
 from .config import ConfigMixin
 from .compile import CompileMixin
-from .remote import RemoteMixin
 from .list import ListMixin
 from .data_processing import DataProcessingMixin
 from .stream import StreamMixin
@@ -38,7 +37,7 @@
 class DCMixins(DatasetMixin, DispatcherMixin, DisplayMixin, ParallelMixin,
                ComputerVisionMixin, StateMixin, MetricMixin, RayMixin,
                ServeMixin, MilvusMixin, DagMixin, FaissMixin, ConfigMixin,
-               CompileMixin, RemoteMixin, ListMixin, DataProcessingMixin,
+               CompileMixin, ListMixin, DataProcessingMixin,
                SafeMixin, StreamMixin, FormatPriorityMixin, AudioMixin):
 
     def __init__(self) -> None:  # pylint: disable=useless-super-delegation

diff --git a/towhee/functional/mixins/remote.py b/towhee/functional/mixins/remote.py