iI:ddlmZmZmZddlmZGddeZy))AnyCallableOptional)overridec eZdZdZ ddedeedeedededef d Z e dded eedefd Z deddfd Z de fd ZdefdZy)_StrategyRegistrya>This class is a Registry that stores information about the Training Strategies. The Strategies are mapped to strings. These strings are names that identify a strategy, e.g., "deepspeed". It also returns Optional description and parameters to initialize the Strategy, which were defined durng the registration. The motivation for having a StrategyRegistry is to make it convenient for the Users to try different Strategies by passing just strings to the strategy flag to the Trainer. Example:: @StrategyRegistry.register("lightning", description="Super fast", a=1, b=True) class LightningStrategy: def __init__(self, a, b): ... or StrategyRegistry.register("lightning", LightningStrategy, description="Super fast", a=1, b=True) Nnamestrategy descriptionr init_paramsreturnc ttstdvr|stddi||ndd<|d<dtdtffd }|||S|S) aRegisters a strategy mapped to a name and with required metadata. Args: name : the name that identifies a strategy, e.g. "deepspeed_stage_3" strategy : strategy class description : strategy description override : overrides the registered strategy, if True init_params: parameters to initialize the strategy z`name` must be a str, found 'z@' is already present in the registry. HINT: Use `override=True`.r r r r c&|d<d<<|S)Nr strategy_name)r datar selfs y/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/lightning/fabric/strategies/registry.py do_registerz/_StrategyRegistry.register..do_registerIs%'D $(D !DJO) isinstancestr TypeError ValueErrorr)rr r r rr rrs`` @rregisterz_StrategyRegistry.register,s$ 4 5:4&AB B 4<q&fgh h!-8-Dk"])] ( x   x( (rdefaultc||vr||}|ddi|dS||Sd}djt|jxsd}t|j ||)zCalls the registered strategy with the required parameters and returns the strategy object. Args: name (str): the name that identifies a strategy, e.g. "deepspeed_stage_3" r r z/'{}' not found in registry. Available names: {}, noner)joinsortedkeysKeyErrorformat)rr rrerr_msgavailable_namess rgetz_StrategyRegistry.getTsr 4<:D#4 #:d=&9: :  NC))F499;$78BFw~~dO<==rc&|j|y)z(Removes the registered strategy by name.N)pop)rr s rremovez_StrategyRegistry.removegs  rc4t|jS)z(Returns a list of registered strategies.)listr$rs ravailable_strategiesz&_StrategyRegistry.available_strategiesksDIIK  rc^djdj|jS)NzRegistered Strategies: {}r )r&r"r$r/s r__str__z_StrategyRegistry.__str__os"*11$))DIIK2HIIr)NNF)N)__name__ __module__ __qualname____doc__rrrboolrrrr)r,r.r0r2rrrrrs6(,%) &&8$&c] &  &  & &P>>hsm>s>>$34!d!JJrrN)typingrrrtyping_extensionsrdictrrrrr;s +*&]J]Jr