|h*SrSSKJr SSKrSSKrSSKrSSKJrJr SSK r Sr \ r S Sjr S Sjr"S S 5rg!\ a Sr N!f=f) a Memory monitoring utilities for measuring memory usage. Example usage: tracker = MemoryTracker("my_function") with tracker.monitor(): my_function() # Access data: tracker.rss_delta, tracker.duration, etc. # Get formatted string: tracker.get_summary() ) annotationsN)DictOptionalTFc~[(a"[R"5nUR$g![a gf=f)z Get current available system memory in bytes. Used for memory threshold checking in parallel test execution. Returns: int or None: Available memory in bytes, or None if unavailable N) _HAS_PSUTILpsutilvirtual_memory available Exception)sys_mems P/home/james-whalen/.local/lib/python3.13/site-packages/numba/misc/memoryutils.pyget_available_memoryrsD{ ++-G$$ $     s / <<cz0n[(am[R"[R"55nUR 5nUR US'[R"5nURUS'SU;aSUS'SU;aSUS'U$![R[R4a N>f=f)a> Get memory usage information needed for monitoring. Returns only RSS and available memory which are the fields actually used by the MemoryTracker. Returns: dict: Memory usage information including: - rss: Current process RSS (physical memory currently used) - available: Available system memory rssr N) rrProcessosgetpid memory_inforr r NoSuchProcess AccessDenied)rprocessmem_infor s r get_memory_usager/sK{ nnRYY[1G**,H!)K ++-G'.'8'8K $  K! E+%#' K $$f&9&9:   sA,B$B:9B:c\rSrSr%SrS\S'S\S'S\S'S\S 'S \S 'S \S 'S\S 'S\S'SSjr\RS5r SSjr Sr g) MemoryTrackerTz A simple memory monitor that tracks RSS delta and timing. Stores monitoring data in instance attributes for later access. Each instance is typically used for monitoring a single operation. intpidstrnamez float | None start_timeend_timezDict[str, int | None] | None start_memory end_memorydurationz int | None rss_deltac[R"5UlXlSUlSUlSUlSUlSUlSUl g)z6Initialize a MemoryTracker with empty monitoring data.N) rrrr r!r"r#r$r%r&)selfr s r __init__MemoryTracker.__init__ds?99;    c## [R"5Ul[5UlUv [R"5Ul[5UlURUR- UlURRSS5nUR RSS5nU(a U(aX!- OSUlg![R"5Ul[5UlURUR- UlURRSS5nUR RSS5nU(aU(a X!- UlfSUlf=f7f)a_ Context manager to monitor memory usage during function execution. Records start/end memory usage and timing, calculates RSS delta, and stores all data in instance attributes. Args: name (str): Name/identifier for the function or operation being monitored Yields: self: The MemoryTracker instance for accessing stored data rrN) timer!rr#r"r$r%getr&)r( start_rssend_rsss r monitorMemoryTracker.monitoros ))+,. ?J!IIKDM.0DO MMDOO;DM))--eQ7Ioo))%3G!*w '2<= N!IIKDM.0DO MMDOO;DM))--eQ7Ioo))%3G!*w '2DN<= Ns*E-C BE- B E**E-c NURb URc [S5eURRS5nSSjn[R "S[R "UR55nURRSS5nURRSS5nSUR3SUR3S U3S URS S 3S U"U53SU"U53SU"URSS93SU"U53/nSRU5$)at Return a formatted summary of the memory monitoring data. Formats the stored monitoring data into a human-readable string containing name, PID, RSS delta, available memory, duration, and start time. Returns: str: Formatted summary string with monitoring results Note: Should be called after monitor() context has completed to ensure all data is available. z$Memory monitoring data not availabler cUcgUS:XagSnU(a US:aSOSn[U5nSHnUS:a UUS S U3s $US-nM UUS S 3$) z&Convert bytes to human readable formatzN/Arz0 B-+)BKBMBGBg@z.2f z TB)abs) bytes_val show_signsignunits r format_bytes/MemoryTracker.get_summary..format_bytess{ A~D'!msII/v%"VIc?!D6::V# 0VIc?#. .r+z%H:%M:%SrrzName: zPID: zStart: z Duration: z.3fsz Start RSS: z End RSS: z RSS delta: T)r?zAvail memory: z | )F) r#r$ ValueErrorr.r-strftime localtimer!r rr%r&join)r(current_availablerBstart_tsr/r0bufs r get_summaryMemoryTracker.get_summarys%    $(?CD D OO// < /$==T^^DOO-LM%%))%3 //%%eQ/TYYK DHH: hZ s+1 -,y12 3 W-. /,t~~FG H\*;<= >  zz#r+)r%r$r"r rr&r#r!N)r r)returnr) __name__ __module__ __qualname____firstlineno____doc____annotations__r) contextlibcontextmanagerr1rL__static_attributes__r+r rrTs\ H I..,, ??@4r+r)rNz Optional[int])rNzDict[str, Optional[int]])rS __future__rrrUr-typingrrrr ImportError IS_SUPPORTEDrrrrXr+r r]s] #  !K  &"JppKs;AA