updates

Author: alexandraBara

nodescraper/cli/cli.py

@@ -354,6 +354,7 @@ def main(arg_input: Optional[list[str]] = None):
         arg_input = sys.argv[1:]
 
     plugin_reg = PluginRegistry(plugin_pkg=extra_pkgs)
+
     config_reg = ConfigRegistry()
     parser, plugin_subparser_map = build_parser(plugin_reg, config_reg)
 
@@ -447,6 +448,7 @@ def main(arg_input: Optional[list[str]] = None):
         connections=parsed_args.connection_config,
         system_info=system_info,
         log_path=log_path,
+        plugin_registry=plugin_reg,
     )
 
     try:

nodescraper/pluginexecutor.py

@@ -50,17 +50,15 @@ def __init__(
         connections: Optional[dict[str, dict | BaseModel]] = None,
         system_info: Optional[SystemInfo] = None,
         logger: Optional[logging.Logger] = None,
-        plugin_registry: Optional[PluginRegistry] = None,
+        plugin_registry: PluginRegistry | None = None,
         log_path: Optional[str] = None,
     ):
 
         if logger is None:
             logger = logging.getLogger(DEFAULT_LOGGER)
         self.logger = logger
 
-        if plugin_registry is None:
-            plugin_registry = PluginRegistry()
-        self.plugin_registry = plugin_registry
+        self.plugin_registry = plugin_registry or PluginRegistry()
 
         if system_info is None:
             system_info = SystemInfo()