data_juicer.ops.deduplicator.ray_basic_deduplicator module

class data_juicer.ops.deduplicator.ray_basic_deduplicator.DedupSet

Bases: object

__init__()

is_unique(key)

data_juicer.ops.deduplicator.ray_basic_deduplicator.get_remote_dedup_set()

Get the remote version of DedupSet with Ray decorator applied at runtime.

class data_juicer.ops.deduplicator.ray_basic_deduplicator.Backend(*args, **kwargs)

Bases: ABC

Backend for deduplicator.

abstractmethod __init__(*args, **kwargs)

abstractmethod is_unique(md5_value: str)

class data_juicer.ops.deduplicator.ray_basic_deduplicator.ActorBackend(dedup_set_num: int, RemoteDedupSet=None)

Bases: Backend

Ray actor backend for deduplicator.

__init__(dedup_set_num: int, RemoteDedupSet=None)

is_unique(md5_value: str)

class data_juicer.ops.deduplicator.ray_basic_deduplicator.RedisBackend(redis_address: str)

Bases: Backend

Redis backend for deduplicator.

__init__(redis_address: str)

is_unique(md5_value: str)

class data_juicer.ops.deduplicator.ray_basic_deduplicator.RayBasicDeduplicator(backend: str = 'ray_actor', redis_address: str = 'redis://localhost:6379', *args, **kwargs)

Bases: Filter

A basic exact matching deduplicator for RAY. Although its functionality is deduplication, it is implemented as Filter sub-class.

EMPTY_HASH_VALUE = 'EMPTY'

__init__(backend: str = 'ray_actor', redis_address: str = 'redis://localhost:6379', *args, **kwargs)

Initialization. :param backend: the backend for dedup, either ‘ray_actor’ or ‘redis’ :param redis_address: the address of redis server :param args: extra args :param kwargs: extra args

calculate_hash(sample, context=False)

Calculate hash value for the sample.

compute_stats_single(sample, context=False)

Compute stats for the sample which is used as a metric to decide whether to filter this sample.

Parameters:

• sample – input sample.

• context – whether to store context information of intermediate vars in the sample temporarily.

Returns:

sample with computed stats

process_single(sample)

For sample level, sample –> Boolean.

Parameters:

sample – sample to decide whether to filter

Returns:

true for keeping and false for filtering