Do not check torchx.components.base module for components (#258)

Summary:
Pull Request resolved: #258

Do not check `torchx.components.base` module for components

Reviewed By: kiukchung

Differential Revision: D31664832

fbshipit-source-id: 3de72047810ff8c2478e036ce5626459d4c073af

Author: aivanou

torchx/specs/finder.py

@@ -119,9 +119,13 @@ def _get_components_from_dir(
         search_pattern = os.path.join(search_dir, "**", "*.py")
         component_defs = []
         for filepath in glob.glob(search_pattern, recursive=True):
-            module = self._try_load_module(
-                self._get_module_name(filepath, search_dir, base_module)
-            )
+            module_name = self._get_module_name(filepath, search_dir, base_module)
+            # TODO(aivanou): move `torchx.components.base` to `torchx.specs`, since
+            # there is nothing related to components in `torchx.components.base`
+            # see https://github.com/pytorch/torchx/issues/261
+            if module_name.startswith("torchx.components.base"):
+                continue
+            module = self._try_load_module(module_name)
             defs = self._get_components_from_module(base_module, module)
             component_defs += defs
         return component_defs