+ QV-j7V^RIt^RIHtHt^RIHt^RIHtHtH t H t H t H t ^RI t^RIHt^RIHt^RIHt^RIHtHtHt^R IHtHt]'d ^RIt^R IHt]P@'d$^RIt!R R ]PBPD4t#]!R R44t$RRRllt%R#)N) dataclassfield)Path) TYPE_CHECKINGAnyClassVarDictOptionalUnion)config)DownloadConfig) array_cast) is_local_path is_remote_urlxopen)no_op_if_value_is_nullstring_to_dict) FeatureTypecHaa]tRt^toRtV3RlV3RlltRtRtVtV;t #)Nifti1ImageWrapperzI A wrapper around nibabel's Nifti1Image to customize its representation. cH<V^8dQhRS[PP/#) nifti_image)nibnifti1 Nifti1Image)format __classdict__s"h/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/datasets/features/nifti.py __annotate__Nifti1ImageWrapper.__annotate__s + + (>(> +c <\SV`VPVPVPVP VP VP4R7WnR#))dataobjaffineheaderextrafile_mapdtypeN) super__init__r$r%r&r'r(get_data_dtyper)selfr __class__s&&rr+Nifti1ImageWrapper.__init__sV G #++"))"))!''$--!002   + r"c^RIHpHpHpHp^RIHpVPP4pV!4pVPVP4VPVPn RVPnRVPnRp\!VPR4'dmRVPP"9dR\%VPP"R,RR4e(VPP"R,P&pVfRpV!WR 7p VP)V .4V!V4R#) r)NiiVue ShowRender SliceTypeVolume)displayTNr(imagefilenamez volume.nii.gz)namedata) ipyniivuer1r2r3r4IPython.displayr5rto_bytesset_slice_type MULTIPLANARALWAYSoptsmultiplanar_show_rendershow_3d_crosshairmultiplanar_force_renderhasattrr(getattrr7 load_volumes) r-r1r2r3r4r5bytes_nvr8volumes & r _repr_html_Nifti1ImageWrapper._repr_html_'s G G /%%..0FB   i33 4.8.?.?BGG +(,BGG %/3BGG ,Dt''44t//888 0 0 9 9' BJPTUa++44W=FFD|&3F OOVH % BKr")r) __name__ __module__ __qualname____firstlineno____doc__r+rJ__static_attributes____classdictcell__ __classcell__)r.rs@@rrrs  + +  r"rc2a]tRt^?toRtRt]!RRR7tRt] P!R] P!4R] P!4/4t ]!RRRR 7tR tV3R lR ltRV3R lRlltRV3RlRlltV3RlRltV3RlRltV3RltRtVtR#)Niftiaq **Experimental.** Nifti [`Feature`] to read NIfTI neuroimaging files. Input: The Nifti feature accepts as input: - A `str`: Absolute path to the NIfTI file (i.e. random access is allowed). - A `pathlib.Path`: path to the NIfTI file (i.e. random access is allowed). - A `dict` with the keys: - `path`: String with relative path of the NIfTI file in a dataset repository. - `bytes`: Bytes of the NIfTI file. This is useful for archived files with sequential access. - A `nibabel` image object (e.g., `nibabel.nifti1.Nifti1Image`). Args: decode (`bool`, defaults to `True`): Whether to decode the NIfTI data. If `False` a string with the bytes is returned. `decode=False` is not supported when decoding examples. Examples: ```py >>> from datasets import Dataset, Nifti >>> ds = Dataset.from_dict({"nifti": ["path/to/file.nii.gz"]}).cast_column("nifti", Nifti()) >>> ds.features["nifti"] Nifti(decode=True, id=None) >>> ds[0]["nifti"] >>> ds = ds.cast_column("nifti", Nifti(decode=False)) >>> ds[0]["nifti"] {'bytes': None, 'path': 'path/to/file.nii.gz'} ``` TNF)defaultreprznibabel.nifti1.Nifti1Imagebytespath)rVinitrWcVP#N)pa_type)r-s&r__call__Nifti.__call__ks ||r"cF<V^8dQhRS[S[S[S[S[R3,RS[/#)rvaluenib.Nifti1Imagereturn)r strrX bytearraydict)rrs"rr Nifti.__annotate__ns-&&E#uiGX*X$Y&^b&r"c <\P'd^RIpMRp\V\4'dRVRR/#\V\ 4'dR\ VP 44RR/#\V\\34'dRRRV/#Ve1\WPP4'd \V4#\V\4'dVPR4eB\PP!VR,4'dRRRVPR4/#VPR4fVPR4e%RVPR4RVPR4/#\#RV R24h\#R\%V4 R24h)zEncode example into a format for Arrow. Args: value (`str`, `bytes`, `nibabel.Nifti1Image` or `dict`): Data passed as input to Nifti feature. Returns: `dict` with "path" and "bytes" fields NrYrXzTA nifti sample should have one of 'path' or 'bytes' but they are missing or None in .zPA nifti sample should be a string, bytes, Path, nibabel image, or dict, but got )r NIBABEL_AVAILABLEnibabel isinstancerdrabsoluterXre spatialimages SpatialImageencode_nibabel_imagerfgetosrYisfile ValueErrortype)r-rars&& rencode_exampleNifti.encode_examplensk  # # # !C eS ! !E7D1 1 t $ $C 017DA A y1 2 2D'51 1 _E3D3D3Q3Q!R!R'. . t $ $yy ,f 1N1Nvuyy/@AA7#/599V3D3P7!3VUYYv=NOO jkpjqqrsbcghmcnboopq r"c$<V^8dQhRS[RR/#)rrarcr)rf)rrs"rr rgs=)=)D=)EY=)r"c \P'd^RIpM \R4hVf/pVR,VR,rTVEfSVf\ RV R24hVP R4'd@\ VPR4R,4'dVPR4R,p\ V4'dVP!V4pEMVPR4R,pVP \P4'd\PM\Pp\Wx4p V eVPV R ,4MRp \V R 7p \VR V R 7;_uu_4p VP!V 4pRRR4M>^RIp VR ,R8XdV P#V4pVP$P'V4p\)X4# +'giL;i)a Decode example NIfTI file into nibabel image object. Args: value (`str` or `dict`): A string with the absolute NIfTI file path, a dictionary with keys: - `path`: String with absolute or relative NIfTI file path. - `bytes`: The bytes of the NIfTI file. token_per_repo_id (`dict`, *optional*): To access and decode NIfTI files from private repositories on the Hub, you can pass a dictionary repo_id (`str`) -> token (`bool` or `str`). Returns: `nibabel.Nifti1Image` objects Nz:To support decoding NIfTI files, please install 'nibabel'.rYrXzBA nifti should have one of 'path' or 'bytes' but both are None in rizgzip://::repo_idtokenrbdownload_config:NrNs)r rjrk ImportErrorrt startswithrsplitload HF_ENDPOINTHUB_DATASETS_URLHUB_DATASETS_HFFS_URLrrqr rgzip decompressr from_bytesr)r-ratoken_per_repo_idrrYrGnifti source_urlpatternsource_url_fieldsr}rfrs&&& rdecode_exampleNifti.decode_examples&  # # # !Z[ [  $ " V}eGnf >| #efkellm!noo??9--- 4@PQS@T2U2U::d+B/D &&HHTNE!%D!1"!5J&001C1CDD//#99 )7z(K%O`Ol)--.? .JKrv'55&AOtT?KKq # LK r k)0OO..v6E!%((LKs 9G G. cZ<V^8dQhRS[PRS[RS[RS[P/#)rstorage local_files remote_filesrc)pa StructArraybool)rrs"rr rgs;C1C1~~C1LPC1gkC1 C1r"c  aSf/o\V3Rl4p\P!VP4Uu.uFlpVedVR,fPV'd\ VR,4'g!V'd(\ VR,4'dV!VR,4M VR,MRNKn up\P !4R7p\P!VPR4P4Uu.uF\pVeTV'd\ V4'gV'd1\ V4'd \PPV4MTMRNK^ up\P!4R7p \PPWy.RR.VP4R7p\WP 4#uupiuupi)aEmbed NifTI files into the Arrow array. Args: storage (`pa.StructArray`): PyArrow array to embed. token_per_repo_id (`dict`, optional): Dictionary repo_id -> token to fetch the files bytes. local_files (`bool`, defaults to `True`) Whether to embed local files data in the array remote_files (`bool`, defaults to `True`) Whether to embed remote files data in the array. E.g. files with paths that start with hf:// or https:// Returns: `pa.StructArray`: Array in the NifTI arrow storage type, that is `pa.struct({"bytes": pa.binary(), "path": pa.string()})`. Nc<VPR4R,pVP\P4'd\PM\P p\ W4pVeSPVR,4MRp\VR7p\VRVR7;_uu_4pVP4uuRRR4# +'giR#;i)rzNr{r|r~rr) rrr rrrrrqr rread)rYrrrr}rrrs& r path_to_bytes*Nifti.embed_storage..path_to_bytessD)"-J+5+@+@ASAS+T+T''Z`ZvZv !/z C K\Kh%))*;I*FGnrE,59OtT?CCqvvxDCCCs *C C rXrYrumask)rrarray to_pylistrrbinaryrrrrYbasenamestringr from_arraysis_nullrr]) r-rrrrrx bytes_arrayrY path_arrays &&f&& r embed_storageNifti.embed_storages0  $ "     hh!**, -A= z)%-& *B*B YfghiogpYqYq"!F),7-   XX$MM&1;;= >D#$ d(;(;R_`dReReGG$$T* >   ..,,k-FRXHY`g`o`o`q,r'<<007  s#F>F>56F>8G G"8GcH<V^8dQhRS[RS[S[R3,3,/#)rrcr)r r rd)rrs"rr rgs(   }d3 3E.FFG  r"c Z^RIHpVP'dV#RV!R4RV!R4/#)zfIf in the decodable state, return the feature itself, otherwise flatten the feature into a dictionary.)ValuerXrrYr)featuresrdecode)r-rs& rflatten Nifti.flattens8#{{{  xh r"c<V^8dQhRS[S[PS[PS[P3,RS[P/#)rrrc)r r StringArrayr BinaryArray)rrs"rr rg's:"1"1E"..".."..*X$Y"1^`^l^l"1r"c *\PPVP4'dr\P!R.\ V4,\P !4R7p\PPW!.RR.VP4R7pEM\PPVP4'dr\P!R.\ V4,\P!4R7p\PPW.RR.VP4R7pEM?\PPVP4'EdVPPR4^8dVPR4pM<\P!R.\ V4,\P !4R7pVPPR4^8dVPR4pM<\P!R.\ V4,\P!4R7p\PPW#.RR.VP4R7p\WP 4#)aCast an Arrow array to the Nifti arrow storage type. The Arrow types that can be converted to the Nifti pyarrow storage type are: - `pa.string()` - it must contain the "path" data - `pa.binary()` - it must contain the NIfTI bytes - `pa.struct({"bytes": pa.binary()})` - `pa.struct({"path": pa.string()})` - `pa.struct({"bytes": pa.binary(), "path": pa.string()})` - order doesn't matter Args: storage (`Union[pa.StringArray, pa.StructArray, pa.BinaryArray]`): PyArrow array to cast. Returns: `pa.StructArray`: Array in the Nifti arrow storage type, that is `pa.struct({"bytes": pa.binary(), "path": pa.string()})`. NrrXrYr)rtypes is_stringrurlenrrrr is_binaryr is_structget_field_indexrrr])r-rrrs&& r cast_storageNifti.cast_storage's$ 88  gll + +((D6CL#8ryy{KKnn00+1G'SYIZahapapar0sG XX   - -4&3w<"7biikJJnn00'1FRXHY`g`o`o`q0rG XX   - -||++G49%mmG4  hhvG '<299;O ||++F3q8$]]62 XXtfs7|&;"))+N nn00+1JWV\L]dkdsdsdu0vG'<<00r"c<V^8dQh/S[;R&S[S[,;R&S[S[,;R&S[S[,;R&S[;R&#)rridr)r]_type)rr rdrr)rrs"rr rg?s\H IJ  7KP C=7QRc]SST ?Ur"r\)NTT)rLrMrNrOrPrrrr)rstructrrr]rr^rvrrrr__annotate_func__rQrR)rs@rrUrU?s DFd7B8EYYfbiik'RSGwU?E&&P=)=)~C1C1J   "1"1Qr"rUc V^8dQhRRR\R\\\\\\ 3,,3,/#)rimgrb force_bytesrc)rrfrdr r rX)rs"rr r Ls@////d/tTWYabghkmrhrbsYtTtOu/r"c\VR4'd:VPe,V'g$VPR,PpRVRR/#VP4pRRRV/#)a Encode a nibabel image object into a dictionary. If the image has an associated file path, returns the path. Otherwise, serializes the image content into bytes. Args: img: A nibabel image object (e.g., Nifti1Image). force_bytes: If `True`, always serialize to bytes even if a file path exists. Needed to upload bytes properly. Returns: dict: A dictionary with "path" or "bytes" field. r(Nr6rYrX)rDr(r7r<)rrr7 bytes_datas&& rrprpLsYsJCLL$<[<<(11'400J D': ..r")F)&rr dataclassesrrpathlibrtypingrrrr r r pyarrowrr download.download_configr tablerutils.file_utilsrrrutils.py_utilsrrrkrrrrjrrrrUrprr"rrs (FF5BBC% %SZZ33%P I1I1 I1X//r"