from dataclasses import dataclass


@dataclass
class DataIngestionArtifact:
    feature_store_file_path: str
    train_file_path: str
    test_file_path: str


@dataclass
class DataValidationArtifact:
    report_file_path: str


@dataclass
class DataTransformationArtifact:
    transform_object_path: str
    transformed_train_path: str
    transformed_test_path: str
    target_encoder_path: str


@dataclass
class ModelTrainerArtifact:
    model_path: str
    f1_train_score: float
    f2_test_score: float


@dataclass
class ModelEvaluationArtifact:
    is_model_accepted: bool
    improved_accuracy: float


@dataclass
class ModelPusherArtifact:
    pusher_model_dir: str
    saved_model_dir: str