github/home-llm
1
1
Fork 0
mirror of https://github.com/acon96/home-llm.git synced 2026-01-08 05:14:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

Annotate generate_random_parameter and get_random_response

Browse Source
This commit is contained in:
Evian Schlenz
2025-12-31 01:12:22 +01:00
parent b3df3d5346
commit c1b5d912d2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
data/utils.py
View File

@@ -45,7 +45,7 @@ def get_included_vars(response: str):
return ",".join(sorted(result))
def generate_random_parameter(param_name, piles_of_data):
def generate_random_parameter(param_name: str, piles_of_data: "DatasetPiles"):
RANDOM_PARAMETER_GENERATORS = {
"rgb_color": lambda: (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)),
"brightness": lambda: random.randint(0, 100),
@@ -68,7 +68,7 @@ def generate_random_parameter(param_name, piles_of_data):
return param_generator()
def get_random_response(pile_of_responses, *, service: str, persona: str, question_template: str, short: bool) -> tuple[str, str]:
def get_random_response(pile_of_responses: pandas.DataFrame, *, service: str, persona: str, question_template: str, short: bool) -> tuple[str, str]:
required_vars = list(set([var for var in var_pattern.findall(question_template) if "device_name" not in var]))