ci(torchvision): install torchvision dependency to enable tests

gnes-ai · Jul 25, 2019 · 83276f9 · 83276f9
1 parent 7fcd50d
commit 83276f9
Showing 1 changed file with 3 additions and 12 deletions.
diff --git a/tests/test_preprocessor.py b/tests/test_preprocessor.py
@@ -16,18 +16,12 @@ def setUp(self):
         self.dirname = os.path.dirname(__file__)
 
     def test_preprocessor_service_empty(self):
-        args = set_preprocessor_service_parser().parse_args([
-            '--port_in', str(YamlComposer._get_random_port()),
-            '--port_out', str(YamlComposer._get_random_port())
-        ])
+        args = set_preprocessor_service_parser().parse_args([])
         with PreprocessorService(args):
             pass
 
     def test_preprocessor_service_echo(self):
-        args = set_preprocessor_service_parser().parse_args([
-            '--port_in', str(YamlComposer._get_random_port()),
-            '--port_out', str(YamlComposer._get_random_port())
-        ])
+        args = set_preprocessor_service_parser().parse_args([])
         c_args = _set_client_parser().parse_args([
             '--port_in', str(args.port_out),
             '--port_out', str(args.port_in)
@@ -44,10 +38,7 @@ def test_preprocessor_service_echo(self):
             print(r)
 
     def test_preprocessor_service_realdata(self):
-        args = set_preprocessor_service_parser().parse_args([
-            '--port_in', str(YamlComposer._get_random_port()),
-            '--port_out', str(YamlComposer._get_random_port())
-        ])
+        args = set_preprocessor_service_parser().parse_args([])
         c_args = _set_client_parser().parse_args([
             '--port_in', str(args.port_out),
             '--port_out', str(args.port_in)