Source code for nv_ingest.schemas.image_dedup_schema

# SPDX-FileCopyrightText: Copyright (c) 2024, NVIDIA CORPORATION & AFFILIATES.
# All rights reserved.
# SPDX-License-Identifier: Apache-2.0


import logging

from pydantic import ConfigDict, BaseModel
from pydantic import StrictBool

logger = logging.getLogger(__name__)



[docs]
class ImageDedupSchema(BaseModel):
    raise_on_failure: StrictBool = False
    cpu_only: StrictBool = False
    model_config = ConfigDict(extra="forbid")