INNER CODE UNIT · Python
_load_model
antigenomics/vdjtools · python/vdjtools/cli/__init__.py:74
def _load_model(model: Optional[str], source: str, model_path: Optional[Path]):
from vdjtools.model import load_bundled, load_model
if model_path is not None:
return load_model(model_path)
if model is None:
_err("give a model: --model <LOCUS> (built-in) or --model-path <DIR>")
try:
return load_bundled(model, source=source)
except (ValueError, FileNotFoundError) as e: # unknown source / locus not bundled
_err(str(e))
def _write(df: pl.DataFrame, out: Optional[Path]) -> None:
"""Write a result frame. ``.parquet`` / ``.pq`` → Parquet; anything else (or stdout) → TSV."""
if out is None:
sys.stdout.write(df.write_csv(separator="\t"))
return