o qoi<@sxdZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZGd d d eZd S) z Sweeper plugin interface )abstractmethod)AnyListSequenceOptional) TaskFunction) DictConfig)Launcher)Plugin) HydraContextc @seZdZUdZeeed<eeed<eeed<e dede deddfdd Z e d e e defd d Zd eee ddfddZdS)Sweeperz An abstract sweeper interface Sweeper takes the command line arguments, generates a and launches jobs (where each job typically takes a different command line arguments) hydra_contextconfiglauncher task_functionreturnNcCst)N)NotImplementedError)selfrrrrI/home/ubuntu/.local/lib/python3.10/site-packages/hydra/plugins/sweeper.pysetupsz Sweeper.setup argumentscCsdS)a. Execute a sweep :param arguments: list of strings describing what this sweeper should do. exact structure is determine by the concrete Sweeper class. :return: the return objects of all thy launched jobs. structure depends on the Sweeper implementation. Nr)rrrrrsweep%s z Sweeper.sweepbatchcCs^t|dr|jdur|jjn|j}|dusJ|jdusJ|D] }|j|jt|dq dS)z Ensures that the given batch can be composed. This repeat work the launcher will do, but as the launcher may be performing this in a different process/machine it's important to do it here as well to detect failures early. rN) master_configsweep_overrides)hasattrr config_loaderrload_sweep_configlist)rrr overridesrrrvalidate_batch_is_legal0s   zSweeper.validate_batch_is_legal)__name__ __module__ __qualname____doc__rr __annotations__rr rrrrstrrrrr"rrrrr s$       r N)r&abcrtypingrrrr hydra.typesr omegaconfrrr pluginr r r rrrrs