from pathlib import Path
from loguru import logger
from tqdm import tqdm
import typer
from practice_sdoml.config import MODELS_DIR, PROCESSED_DATA_DIR
app = typer.Typer()
[docs]
@app.command()
def main(
# ---- REPLACE DEFAULT PATHS AS APPROPRIATE ----
features_path: Path = PROCESSED_DATA_DIR / "test_features.csv",
model_path: Path = MODELS_DIR / "model.pkl",
predictions_path: Path = PROCESSED_DATA_DIR / "test_predictions.csv",
# -----------------------------------------
):
# ---- REPLACE THIS WITH YOUR OWN CODE ----
logger.info("Performing inference for model...")
for i in tqdm(range(10), total=10):
if i == 5:
logger.info("Something happened for iteration 5.")
logger.success("Inference complete.")
# -----------------------------------------
if __name__ == "__main__":
app()