Admin User 420f351a06
All checks were successful
Build and Push Docker Image / test (push) Successful in 8s
Build and Push Docker Image / build_and_push (push) Successful in 21s
PD v2 Post-Processing block
2025-02-05 19:24:24 +00:00

32 lines
909 B
Python

import logging
# Configure logging
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s [%(levelname)s] %(name)s - %(message)s",
)
logger = logging.getLogger(__name__)
def __main__(probability:float):
logger.info("Received input: probability=%.8f", float(probability))
if not isinstance(probability, (int, float)):
logger.error("Invalid input type: probability=%s", type(probability).__name__)
raise ValueError("Input probability must be a number (int or float)")
if probability <= 0.26:
grade = "A1"
elif 0.26 < probability <= 0.38:
grade = "A2"
elif 0.38 < probability <= 0.52:
grade = "B1"
elif 0.52 < probability <= 0.652:
grade = "B2"
elif 0.652 < probability <= 0.7:
grade = "C1"
else:
grade = "C2"
logger.info(f"PD V2 Grade: {grade}")
return {"grade":grade}