o Th @sFddlZddlmZddZddddZdd d Zdd d d ZdS)N) get_namespacec Csht|tjs|Sz d|jjvr|WSWn ttfyYnwt|}tj|jd|id}|j|dS)zAttach unique values of y to y and return the result. The result is a view of y, and the metadata (unique) is not attached to y. unique)metadata)dtype) isinstancenpndarrayrrAttributeError TypeErrorrview)yr unique_dtyperU/home/air/segue/gpt/backup/venv/lib/python3.10/site-packages/sklearn/utils/_unique.py_attach_unique s    rF) return_tuplecGs.tdd|D}t|dkr|s|dS|S)aAttach unique values of ys to ys and return the results. The result is a view of y, and the metadata (unique) is not attached to y. IMPORTANT: The output of this function should NEVER be returned in functions. This is to avoid this pattern: .. code:: python y = np.array([1, 2, 3]) y = attach_unique(y) y[1] = -1 # now np.unique(y) will be different from cached_unique(y) Parameters ---------- *ys : sequence of array-like Input data arrays. return_tuple : bool, default=False If True, always return a tuple even if there is only one array. Returns ------- ys : tuple of array-like or array-like Input data with unique values attached. css|]}t|VqdSN)r.0r rrr 8sz attach_unique..rtuplelen)rysresrrr attach_uniquesrcCsXz|jjdurd|jjvr|jjdWSWn tyYnwt||d\}}||S)zReturn the unique values of y. Use the cached values from dtype.metadata if present. This function does NOT cache the values in y, i.e. it doesn't change y. Call `attach_unique` to attach the unique values to y. Nrxp)rrr r unique_values)r r_rrr_cached_unique>s   r!rcs.tfdd|D}t|dkr|dS|S)aReturn the unique values of ys. Use the cached values from dtype.metadata if present. This function does NOT cache the values in y, i.e. it doesn't change y. Call `attach_unique` to attach the unique values to y. Parameters ---------- *ys : sequence of array-like Input data arrays. xp : module, default=None Precomputed array namespace module. When passed, typically from a caller that has already performed inspection of its own inputs, skips array namespace inspection. Returns ------- res : tuple of array-like or array-like Unique values of ys. c3s|] }t|dVqdS)rN)r!rrrrrisz cached_unique..rrr)rrrrrr cached_uniqueQs r"r)numpyrsklearn.utils._array_apirrrr!r"rrrrs   "