please

Author: paramkpr

src/cli/01_train_teacher.py

@@ -1,8 +1,6 @@
 import typer
-import os
 import yaml
 import logging
-from pathlib import Path
 
 import torch
 from transformers import DataCollatorWithPadding, IntervalStrategy, TrainingArguments, Trainer
@@ -22,7 +20,7 @@
 
 
 @app.command()
-def main(config_path: Path = type.Argument(..., help="Path to YAML config")):
+def main(config_path):
     cfg = yaml.safe_load(config_path.read_text())
 
     # --- SETUP W&B ---