server/tasks/task1_missing.py

"""
Task 1 — Easy: Fill Missing Values
Objective: Fill all NaN values in the employee records DataFrame.
Score: 1.0 - (remaining_nulls / original_nulls)
"""

from server.data_generator import generate_task1_datasets

TASK_ID = 1
MAX_STEPS = 20
DESCRIPTION = (
    "Task 1 (Easy) — Fill Missing Values\n"
    "You have an employee records dataset with missing values (NaN) in "
    "'age', 'salary', and 'department' columns. "
    "Your goal is to fill all missing values so the dataset is complete.\n\n"
    "Available operation: fill_missing\n"
    "  params.strategy: 'median' | 'mean' | 'mode' | 'constant'\n"
    "  params.value: (required when strategy='constant') the fill value\n"
    "Example action: {\"operation\": \"fill_missing\", \"column\": \"age\", \"params\": {\"strategy\": \"median\"}}"
)


def load():
    """Return (dirty_df, clean_df, original_null_count)."""
    dirty, clean = generate_task1_datasets()
    original_nulls = int(dirty.isnull().sum().sum())
    return dirty.copy(), clean, original_nulls


def score(current_df, original_nulls: int) -> float:
    """Score in [0, 1]: fraction of nulls filled."""
    if original_nulls == 0:
        return 1.0
    remaining = int(current_df.isnull().sum().sum())
    return round(max(0.0, 1.0 - remaining / original_nulls), 4)


def count_errors(current_df) -> int:
    return int(current_df.isnull().sum().sum())