Spaces:
Sleeping
Sleeping
File size: 1,610 Bytes
6e079c7 061507f 6e079c7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | import pandas as pd
from src.utils.schema import User
def validate_rules(row:User):
total = getattr(row, "TotalCharges")
monthly = getattr(row, "MonthlyCharges")
if total < monthly:
raise ValueError(f"TotalCharges ({total}) must be >= MonthlyCharges ({monthly})")
allowed_gender =["Male","Female"]
allowed_sen_citizen =[0,1]
if row.gender not in allowed_gender:
raise ValueError (f"Invalid Gender: {row.gender}")
if not (0 <= row.tenure <= 72):
raise ValueError (f"Tenure must be between 0 & 72: {row.tenure}")
# if row.Dependents =="Yes" and row.Partner=="No":
# raise ValueError("Dependents cannot be 'Yes' when Partner is 'No'")
if row.SeniorCitizen not in [0, 1]:
raise ValueError(f"Invalid SeniorCitizen value: {row.SeniorCitizen}")
# if row.InternetService == "No":
# internet_cols =[
# "OnlineSecurity", "OnlineBackup", "DeviceProtection",
# "TechSupport", "StreamingTV", "StreamingMovies"
# ]
# for col in internet_cols:
# if getattr(row,col) not in ["No",None]:
# raise ValueError(f"Invalid, Can't have NO internet and have {col}")
numeric_cols = ["tenure", "MonthlyCharges", "TotalCharges"]
for col in numeric_cols:
value = getattr(row,col)
if value <0:
raise ValueError(f"{col} Can't be negative")
binary_cols = ["Partner", "Dependents", "PhoneService"]
for col in binary_cols:
if getattr(row, col) not in ["Yes", "No"]:
raise ValueError(f"{col} must be Yes or No")
|