Compartilhar via


InputPartition

Uma classe base que representa uma partição de entrada retornada pelo partitions() método de DataSourceReader.

Adicionado no Databricks Runtime 14.3 LTS

Sintaxe

InputPartition(value)

Parâmetros

Parâmetro Tipo Descrição
value qualquer O valor que identifica essa partição.

Observações

Essa classe deve ser picklable.

Exemplos

Use a implementação de partição de entrada padrão:

def partitions(self):
    return [InputPartition(1)]

Subclasse a classe de partição de entrada:

from dataclasses import dataclass

@dataclass
class RangeInputPartition(InputPartition):
    start: int
    end: int

def partitions(self):
    return [RangeInputPartition(1, 3), RangeInputPartition(4, 6)]