custom_datasets/__init__.py

"""
Minimal custom_datasets module for Emuru model inference
"""

class HFDataCollector:
    """Minimal data collator for inference"""
    def __init__(self, tokenizer):
        self.tokenizer = tokenizer
    
    def __call__(self, features):
        return features