Lint & fix mypy module warning

Signed-off-by: Claudio Spiess <[email protected]>

Author: claudiosv

.pre-commit-config.yaml

@@ -56,6 +56,7 @@ repos:
     rev: 'v1.15.0'
     hooks:
       - id: mypy
+        args: [--explicit-package-bases]
         verbose: true
         additional_dependencies: ['types-PyYAML']
   # type check the Python code using pyright

examples/optimizer/__init__.py

@@ -6,15 +6,16 @@
 
 import yaml
 from datasets import load_from_disk
-from fever_evaluator import FEVEREvaluator
-from gsm8k_evaluator import Gsm8kEvaluator
-from gsmhard_evaluator import GsmHardEvaluator
-from mbpp_dataset import MBPPDataset
-from mbpp_evaluator import MBPPEvaluator
 
 from pdl.optimize.config_parser import OptimizationConfig
 from pdl.optimize.pdl_optimizer import PDLOptimizer
 
+from .fever_evaluator import FEVEREvaluator
+from .gsm8k_evaluator import Gsm8kEvaluator
+from .gsmhard_evaluator import GsmHardEvaluator
+from .mbpp_dataset import MBPPDataset
+from .mbpp_evaluator import MBPPEvaluator
+
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
         description="PDL optimization and benchmarking tool",

examples/optimizer/optimize.py

@@ -47,9 +47,7 @@ def dump_program(program):
 
 
 def resave_pdl(input_path: Path, output_path: Path, state: dict) -> int:
-    with (
-        input_path.open(encoding="utf-8") as pdl,
-    ):
+    with (input_path.open(encoding="utf-8") as pdl,):
         pdl_program = Program.model_validate(yaml.safe_load(pdl))
         if pdl_program.root is None or not isinstance(
             pdl_program.root,
@@ -144,9 +142,7 @@ def parse_budget(self):
                 self.time_budget = duration
 
     def load_pdl(self, path: Path) -> Program:
-        with (
-            path.open(encoding="utf-8") as pdl,
-        ):
+        with (path.open(encoding="utf-8") as pdl,):
             return Program.model_validate(yaml.safe_load(pdl))
 
     def sample_random_indices(self, dataset: list, size: int) -> list[Any]:

src/pdl/optimize/pdl_optimizer.py

@@ -22,11 +22,14 @@ def __init__(
     ) -> None:
         super().__init__()
         try:
-            from multiprocess import (  # pyright: ignore pylint: disable=import-outside-toplevel
-                Manager,
-                Process,
+            # isort: off
+            from multiprocess import (  # pylint: disable=import-outside-toplevel
+                Manager,  # pyright: ignore
+                Process,  # pyright: ignore
             )
 
+            # isort: on
+
             self.manager = Manager
             self.process = Process
         except ImportError as exc: