o ̳i@sUddlZddlmZmZmZmZmZmZddlm Z m Z ddl m Z e r.ddl mZmZnedZGdddeeZGd d d eZeeeefZe ed <d d ZdedefddZdS)N)AnyCallableIterableIteratorMappingTypeVar)_TORCHDATA_INSTALLED_TORCHDATA_MIN_VERSION) TypeAlias)BaseNodeLoaderTc@eZdZddZdS)r cOsdSNselfargskwargsrrM/home/ubuntu/.local/lib/python3.10/site-packages/torchtune/data/_torchdata.py__init__szBaseNode.__init__N__name__ __module__ __qualname__rrrrrr  r c@r)r cOs tdSrassert_torchdata_installedrrrrrs zLoader.__init__Nrrrrrr rr DatasetTypecCsts tdtddS)Nzstorchdata is not installed, or the current version is too old. Please (re-)install it with `pip install torchdata>=z`. )r ImportErrorr rrrrr!srfuncreturncstfdd}|S)zW Decorator to check if torchdata is installed and raise an ImportError if not. cst|i|Srr)rrr rrwrapper.sz#requires_torchdata..wrapper) functoolswraps)r r#rr"rrequires_torchdata)sr&)r$typingrrrrrrtorchtune.utils._import_guardrr typing_extensionsr torchdata.nodesr r r strr__annotations__rr&rrrrs