phSrSSKrSSKrSSKJrJrJr SSKJr SSKrSSK r SSK J r SSK JrJr SrSS jrS rS r"S S 5rSrSrSrSSjrSr\RS5rg)zZ Functions used by more than one PyPhi module or class, or that might be of external use. N)chain combinationsproduct)time)comb)config constantsc@^U(a[U4SjU55$S$)z*Return the state-tuple of the given nodes.c3.># UH nTUv M g7fN).0n network_states E/home/james-whalen/.local/lib/python3.13/site-packages/pyphi/utils.py state_of..s15aq!5sr)tuple)nodesrs `rstate_ofrs5:5151 1BBc#j# US:Xag[SUS9HnU(aUv MUSSS2v M g7f)aSReturn all binary states for a system. Args: n (int): The number of elements in the system. big_endian (bool): Whether to return the states in big-endian order instead of little-endian order. Yields: tuple[int]: The next state of an ``n``-element system, in little-endian order unless ``big_endian`` is ``True``. rN)rr)repeat)r)r big_endianstates r all_statesrs8 Av* K"+  +s13c(SURlU$)zMake a NumPy array immutable.F)flags writeableas r np_immutabler$1sAGG HrcUc [S5$[R"U5n[[R "UR UR55R5S5$)zReturn a hash of a NumPy array.N) hashnpascontiguousarrayinthashlibsha1viewdtype hexdigestr"s rnp_hashr07sNyDz QA w||AFF177O,668" ==rc0\rSrSrSrSrSrSrSrSr g) np_hashableBz(A hashable wrapper around a NumPy array.c@[UR55Ulgr )r$copy_array)selfarrays r__init__np_hashable.__init__Fs"5::<0 rc,[UR5$r )r0r6r7s r__hash__np_hashable.__hash__Ist{{##rcX[R"URUR5$r )r( array_equalr6)r7others r__eq__np_hashable.__eq__Ls~~dkk5<<88rc,[UR5$r )reprr6r<s r__repr__np_hashable.__repr__OsDKK  r)r6N) __name__ __module__ __qualname____firstlineno____doc__r9r=rBrF__static_attributes__rrrr2r2Bs21$9!rr2c>[X- 5[R:*$)z%Compare two values up to |PRECISION|.)absr EPSILON)xys reqrSSs qu:** **rclUS:Xa[R"/5$[R"U5nUS:Xa URO%[R"SUR4/U-5n[R"[ X5U5nUR UR5R SU5$)a'NumPy implementation of ``itertools.combinations``. Return successive ``r``-length combinations of elements in the array ``a``. Args: a (np.ndarray): The array from which to get combinations. r (int): The length of the combinations. Returns: np.ndarray: An array of combinations. rr)r(asarrayr.fromiterrr-reshape)r#r data_typebs rcombsr\Ys Avzz"~ 1 A6rxx"agg!0C'DI L& 2A 66!''? " "2q ))rc [XSS9n[R"[R"[ [ U5U55[X!-S9nURSU5$)a``n``-dimensional version of itertools.combinations. Args: a (np.ndarray): The array from which to get combinations. k (int): The desired length of the combinations. Returns: np.ndarray: Indices that give the ``k``-combinations of ``n`` elements. Example: >>> n, k = 3, 2 >>> data = np.arange(6).reshape(2, 3) >>> data[:, comb_indices(n, k)] array([[[0, 1], [0, 2], [1, 2]], [[3, 4], [3, 5], [4, 5]]]) T)exact)countr) rr(rWr from_iterablerranger*rX)rkr_indicess r comb_indicesrdpsR. T "Ekk Lq156 yG ??2q !!rc^[T5mU(aSnOSn[U[T5S-5nU(a[U5nTR 5 [ R "U4SjU55$)afGenerate the power set of an iterable. Args: iterable (Iterable): The iterable from which to generate the power set. Keyword Args: nonempty (boolean): If True, don't include the empty set. reverse (boolean): If True, reverse the order of the powerset. Returns: Iterable: An iterator over the power set. Example: >>> ps = powerset(np.arange(2)) >>> list(ps) [(), (0,), (1,), (0, 1)] >>> ps = powerset(np.arange(2), nonempty=True) >>> list(ps) [(0,), (1,), (0, 1)] >>> ps = powerset(np.arange(2), nonempty=True, reverse=True) >>> list(ps) [(1, 0), (1,), (0,)] rrc3<># UHn[TU5v M g7fr )r)rrYiterables rrpowerset..sL)Q|Ha88)s)listralenreversedreverserr`)rgnonemptyrlstart seq_sizess` rpowersetrps`0H~HeS]Q./IY'    L)L LLrc^^[RR[RR[55mUU4Sjn[ U5Vs/sHn[ R"U"U5SS9PM sn$s snf)zLoad numpy data from the data directory. The files should stored in ``../data/`` and named ``0.npy, 1.npy, ... .npy``. Returns: list: A list of loaded data, such that ``list[i]`` contains the the contents of ``i.npy``. c`>[RRTST[U5S-5$)Ndataz.npy)ospathjoinstr)i directoryroots rget_pathload_data..get_paths$ww||D&)SVf_EErT) allow_pickle)rtruabspathdirname__file__rar(load)rynumr{rxrzs` @r load_datarsX 77??277??84 5DF>C3Z HZBGGHQKd 3Z HH Hs%A?c[5nU"U0UD6n[5n[XS- [R5UlU$)z{Annotate the decorated function or method with the total execution time. The result is annotated with a `time` attribute. )rroundr PRECISION)funcargskwargsrnresultends rtime_annotatedrs< FE 4 "6 "F &C V%5%56FK Mr)F)FF)rLr+rt itertoolsrrrr decoratornumpyr( scipy.specialrrUr r rrr$r0r2rSr\rdrprrrrrrs}   22C , >!!"+ *."D%MPI(   r