Lazy-loading the required modules to avoid loading the global config (#…

…10)

src/aiida_pythonjob/launch.py

@@ -6,7 +6,6 @@
 
 from aiida import orm
 
-from .data.serializer import serialize_to_aiida_nodes
 from .utils import build_function_data, get_or_create_code
 
 
@@ -23,8 +22,8 @@ def prepare_pythonjob_inputs(
     function_data: dict | None = None,
     **kwargs: Any,
 ) -> Dict[str, Any]:
-    pass
     """Prepare the inputs for PythonJob"""
+    from .data.serializer import serialize_to_aiida_nodes
 
     if function is None and function_data is None:
         raise ValueError("Either function or function_data must be provided")

src/aiida_pythonjob/parsers/pythonjob.py

@@ -3,8 +3,6 @@
 from aiida.engine import ExitCode
 from aiida.parsers.parser import Parser
 
-from aiida_pythonjob.data.serializer import general_serializer
-
 
 class PythonJobParser(Parser):
     """Parser for an `PythonJob` job."""
@@ -95,6 +93,7 @@ def find_output(self, name):
 
     def serialize_output(self, result, output):
         """Serialize outputs."""
+        from aiida_pythonjob.data.serializer import general_serializer
 
         name = output["name"]
         if output.get("identifier", "Any").upper() == "NAMESPACE":

tests/test_data.py

@@ -1,12 +1,11 @@
 import aiida
-from aiida_pythonjob.data import general_serializer
-from aiida_pythonjob.utils import get_required_imports
 
 
 def test_typing():
     """Test function with typing."""
     from typing import List
 
+    from aiida_pythonjob.utils import get_required_imports
     from numpy import array
 
     def generate_structures(
@@ -57,6 +56,8 @@ def test_atoms_data():
 
 
 def test_only_data_with_value():
+    from aiida_pythonjob.data import general_serializer
+
     try:
         general_serializer(aiida.orm.List([1]))
     except ValueError as e:

tests/test_parser.py

@@ -5,7 +5,6 @@
 from aiida import orm
 from aiida.cmdline.utils.common import get_workchain_report
 from aiida.common.links import LinkType
-from aiida_pythonjob.parsers import PythonJobParser
 
 
 def create_retrieved_folder(result: dict, output_filename="results.pickle"):
@@ -32,6 +31,8 @@ def create_process_node(result: dict, function_data: dict, output_filename: str
 
 
 def create_parser(result, function_data, output_filename="results.pickle"):
+    from aiida_pythonjob.parsers import PythonJobParser
+
     node = create_process_node(result, function_data, output_filename=output_filename)
     parser = PythonJobParser(node=node)
     return parser