great-ai/great_ai/utilities/parallel_map/parallel_map_configuration.py
2022-07-04 19:31:15 +02:00

15 lines
302 B
Python

from typing import Optional
from pydantic import BaseModel
from ..logger import get_logger
logger = get_logger("parallel_map")
class ParallelMapConfiguration(BaseModel):
concurrency: int
chunk_count: Optional[int]
chunk_size: int
input_length: Optional[int]
function_name: str