dorn/sprint-econtai
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

progress

Browse source
This commit is contained in:
Félix Dorn 2025-07-03 19:40:35 +02:00
parent b7c94590f9
commit f9f9825abb
9 changed files with 941 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

16
pipeline/enrichments.py
View file

@ -7,8 +7,22 @@ from .run import Run
import pandas as pd
def enrich_with_task_estimateability(run: Run) -> pd.DataFrame:
run.metadata.
"""
TODO: check run.cache_dir / computed_task_estimateability.parquet, if it exists, load it, return it, and don't compute this
call enrich with the right parameters, save the output to cache dir,
return it
"""
raise NotImplementedError
def enrich_with_task_estimates(run: Run) -> pd.DataFrame:
"""
TODO: check run.cache_dir / computed_task_estimates.parquet, if it exists, load it, return it, and don't compute this
call enrich with the right parameters, save the output to cache dir,
return it
"""
raise NotImplementedError
def enrich(model: str, system_prompt: str, schema: Any, rpm: int, chunk_size: int = 100, messages: Any):
raise NotImplementedError