feat: support key py_modules in root of the config.yml file (#46)

* feat: support key py_modules in config.yml file

* fix: merge metas_py_modules and root_py_modules of the config.yml

* fix: add test case into test_core

* test: optimize test

* fix: fix the test of core case 5 and 6

* fix: fix py_modules generation method and optimize test

normalizer/core.py

@@ -396,7 +396,6 @@ def normalize(
     logger.debug(f'=> The environment variables: ')
     for k, v in env.items():
         logger.debug('%20s: -> %20s' % (k, v))
-
     if not work_path.exists():
         raise FileNotFoundError(
             f'The folder "{work_path}" does not exist, can not normalize'
@@ -421,9 +420,14 @@ def normalize(
 
         if class_name is None:
             raise Exception('Not found jtype in config.yml')
+
+        metas_py_modules = config.get('metas', {}).get('py_modules', None);
+        root_py_modules = config.get('py_modules', None)
 
-        py_modules = config.get('metas', {}).get('py_modules', None)
+        if metas_py_modules and root_py_modules:
+            raise Exception('The parameter py_modules can only be appear in one of metas and root in config.yml')
 
+        py_modules = metas_py_modules if metas_py_modules else root_py_modules
         if isinstance(py_modules, str):
             py_glob = [work_path.joinpath(py_modules)]
         elif isinstance(py_modules, list):

tests/cases/executor_5/config.yml

@@ -0,0 +1,4 @@
+jtype: Executor5
+py_modules:
+    - foo.py
+

tests/cases/executor_5/foo.py

@@ -0,0 +1,7 @@
+from jina import Executor, requests
+
+class Executor5(Executor):
+    @requests
+    def foo(self):
+        pass
+

tests/cases/executor_6/config.yml

@@ -0,0 +1,2 @@
+jtype: Executor6
+py_modules: foo.py

tests/cases/executor_6/foo.py

@@ -0,0 +1,7 @@
+from jina import Executor, requests
+
+class Executor6(Executor):
+    @requests
+    def foo(self):
+        pass
+

tests/test_core.py

@@ -54,6 +54,14 @@ def test_inspect_dummy_execs():
             Path(__file__).parent / 'cases' / 'executor_4',
             Path(__file__).parent / 'cases' / 'executor_4.json',
         ),
+        (
+            Path(__file__).parent / 'cases' / 'executor_5',
+            None,
+        ),
+        (
+            Path(__file__).parent / 'cases' / 'executor_6',
+            None,
+        ),
         (
             Path(__file__).parent / 'cases' / 'nested',
             Path(__file__).parent / 'cases' / 'nested.json',
@@ -86,8 +94,6 @@ def test_get_executor_args(package_path, expected_path):
             assert executor == expected_executor
     else:
         core.normalize(package_path, dry_run=True)
-
-
 def test_prelude():
     imports = [
         deps.Package(name='tensorflow', version='2.5.0'),