IR-eHddlmZddlZddlmZdgZGddeZdS)) OrderedDictN) isiterableFlagCollectionceZdZdZdZdZdS)ra The purpose of this class is to provide a dictionary for containing arrays of flags for the `NDData` class. Flags should be stored in Numpy arrays that have the same dimensions as the parent data, so the `FlagCollection` class adds shape checking to an ordered dictionary class. The `FlagCollection` should be initialized like an `~collections.OrderedDict`, but with the addition of a ``shape=`` keyword argument used to pass the NDData shape. cd|vr>|d|_t|jstdnt dt j|g|Ri|dS)Nshapez1FlagCollection shape should be an iterable objectz?FlagCollection should be initialized with the shape of the data)poprr ValueError Exceptionr__init__)selfargskwargss >lib/python3.11/site-packages/astropy/nddata/flag_collection.pyr zFlagCollection.__init__s f  G,,DJdj)) V !TUUU VQ  T3D333F33333c t|tjrS|j|jkrt j|||fi|dSt d|j|jtd)Nz1flags array shape {} does not match data shape {}z&flags should be given as a Numpy array) isinstancenpndarrayrr __setitem__r format TypeError)r itemvaluers rrzFlagCollection.__setitem__&s eRZ ( ( F{dj(('dEDDVDDDDD GNN TZ DEE ErN)__name__ __module__ __qualname____doc__r rrrrr sA   4 4 4 F F F F Fr) collectionsrnumpyrastropy.utils.miscr__all__rrrrr$s}$#####))))))  $F$F$F$F$F[$F$F$F$F$Fr