Skip to content

Commit

Permalink
🚨 Fix linting issues
Browse files Browse the repository at this point in the history
Signed-off-by: Aakanksha Duggal <[email protected]>
  • Loading branch information
aakankshaduggal committed Jun 27, 2024
1 parent abf8598 commit 2f08bd6
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/instructlab/sdg/llmblock.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
logger = setup_logger(__name__)


# pylint: disable=dangerous-default-value
class LLMBlock(Block):
# pylint: disable=too-many-instance-attributes
def __init__(
Expand Down Expand Up @@ -185,7 +186,7 @@ def _generate(self, samples, **gen_kwargs) -> str:
)
return [choice.text.strip() for choice in response.choices]

def _validate(self, prompt_template: str, input_dict: Dict[str, Any]) -> bool:
def validate(self, prompt_template: str, input_dict: Dict[str, Any]) -> bool:
if isinstance(prompt_template, dict):
prompt_template = prompt_template[input_dict[self.selector_column_name]]
return super()._validate(prompt_template, input_dict)

0 comments on commit 2f08bd6

Please sign in to comment.