from polyfactory.factories.pydantic_factory import ModelFactory

from app.adapters.ows_dmp import AudienceExport
from app.schemas import S3Record


class S3RecordFactory(ModelFactory[S3Record]):
    __model__ = S3Record


class AudienceExportFactory(ModelFactory[AudienceExport]):
    __model__ = AudienceExport