Lint

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

Author: claudiosv

src/pdl/optimize/PDLThread.py

@@ -1,4 +1,9 @@
 # pylint: disable=too-many-instance-attributes
+import time
+from pathlib import Path
+from threading import Thread
+from typing import Any
+
 from pdl.optimize.config_parser import OptimizationConfig
 from pdl.optimize.util import RETRY_COUNT, TrialOutput, console
 from pdl.pdl_ast import Program, ScopeType
@@ -7,12 +12,6 @@
 from pdl.pdl_parser import PDLParseError
 
 
-import time
-from pathlib import Path
-from threading import Thread
-from typing import Any
-
-
 class PDLThread(Thread):
     """Evaluates a candidate (configuration, i.e. fewshots, style) against **one** test example."""
 
@@ -122,13 +121,13 @@ def run(
                 else:
                     message = get_loc_string(exc.loc) + exc.message
                 console.log(message)
-                retry = True # tries < RETRY_COUNT
+                retry = True  # tries < RETRY_COUNT
                 if tries >= RETRY_COUNT:
                     retry = False
                 console.log("Retrying: ", retry)
                 exception = exc
             except TimeoutError as exc:
-                retry = True # tries < RETRY_COUNT
+                retry = True  # tries < RETRY_COUNT
                 if tries >= RETRY_COUNT:
                     retry = False
                 exception = exc
@@ -163,4 +162,4 @@ def run(
             example=self.example,
             total_tokens=total_tokens,
             index=self.index,
-        )
+        )

src/pdl/optimize/gsmhard_thread.py

@@ -81,6 +81,4 @@ def extract_answer(self, document: PdlApply) -> Any:
     def answer_correct(self, document: str, answer: Any, truth: Any) -> bool:
         answerf = is_float(answer)
         truthf = is_float(truth)
-        return (
-            answer == truth or answerf == truthf or document.endswith(f" {truth}")
-        )
+        return answer == truth or answerf == truthf or document.endswith(f" {truth}")

src/pdl/optimize/mbpp_dataset.py

@@ -24,7 +24,9 @@ def __init__(self):
             MBPP_OUTPUT_NOT_NONE_TASKS,
         )
 
-        self.mbpp = load_from_disk("../prompt-declaration-language-merge/var/mbpp_trajectified").rename_column(
+        self.mbpp = load_from_disk(
+            "../prompt-declaration-language-merge/var/mbpp_trajectified"
+        ).rename_column(
             "code",
             "canonical_solution",
         )

src/pdl/optimize/optimize.py

@@ -1,14 +1,14 @@
 import argparse
+import traceback
 from enum import Enum
 from pathlib import Path
-import traceback
 
 import yaml
 from datasets import concatenate_datasets, load_dataset, load_from_disk
 
+from pdl.optimize.config_parser import OptimizationConfig
 from pdl.optimize.gsmhard_thread import GsmHardTrialThread
 from pdl.optimize.mbpp_dataset import MBPPDataset
-from pdl.optimize.config_parser import OptimizationConfig
 from pdl.optimize.mbpp_thread import MBPPTrialThread
 
 from .fever_thread import FEVERTrialThread

src/pdl/optimize/pdl_optimizer.py

@@ -20,12 +20,7 @@
 
 from pdl.optimize.config_parser import OptimizationConfig
 from pdl.optimize.PDLThread import PDLThread
-from pdl.optimize.util import (
-    CandidateResult,
-    TrialOutput,
-    console,
-    execute_threads,
-)
+from pdl.optimize.util import CandidateResult, TrialOutput, console, execute_threads
 from pdl.pdl_ast import DataBlock, Program
 from pdl.pdl_dumper import dump_yaml
 
@@ -58,9 +53,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))
 
     for variable, value in state.items():
@@ -149,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:

src/pdl/pdl_ast.py

@@ -250,7 +250,7 @@ class Block(BaseModel):
     """Current context
     """
     # Fields for internal use
-    pdl__id: Optional[str]  = Field(default="", exclude=True)
+    pdl__id: Optional[str] = Field(default="", exclude=True)
     """Unique identifier for this block
     """
     pdl__result: Optional[Any] = Field(default=None, exclude=True)
@@ -475,7 +475,8 @@ class CodeBlock(BaseCodeBlock):
     """
 
     lang: Annotated[
-        Literal["python", "command", "jinja", "pdl", "ipython"], BeforeValidator(_ensure_lower)
+        Literal["python", "command", "jinja", "pdl", "ipython"],
+        BeforeValidator(_ensure_lower),
     ]
     """Programming language of the code.
     """