init
This commit is contained in:
commit
2da206d368
17 changed files with 955 additions and 0 deletions
5
pipeline/generators/__init__.py
Normal file
5
pipeline/generators/__init__.py
Normal file
|
@ -0,0 +1,5 @@
|
|||
from .estimate_histplot import generate_estimate_histplot
|
||||
|
||||
GENERATORS = [
|
||||
generate_estimate_histplot
|
||||
]
|
6
pipeline/generators/estimate_histplot.py
Normal file
6
pipeline/generators/estimate_histplot.py
Normal file
|
@ -0,0 +1,6 @@
|
|||
from ..run import Run
|
||||
from pathlib import Path
|
||||
from typing import Generator
|
||||
|
||||
def generate_estimate_histplot(run: Run, output_dir: Path) -> Generator[Path]:
|
||||
raise NotImplementedError
|
6
pipeline/generators/helpers.py
Normal file
6
pipeline/generators/helpers.py
Normal file
|
@ -0,0 +1,6 @@
|
|||
import pandas as pd
|
||||
from typings import List
|
||||
|
||||
def must_have_columns(df: pd.DataFrame, columns: List[str]):
|
||||
if not all(col in df.columns for col in columns):
|
||||
raise ValueError(f"DataFrame is missing required columns: {columns}")
|
Loading…
Add table
Add a link
Reference in a new issue