o oi@sddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZdZejedZejed Zejed Zejd d d vZdedede fddZedejedddZde ee ffddZde ee ffddZdS)N)module_from_specspec_from_file_locationPath) ModuleType)AnyDict)find_namespace_packages.src lightning requirementsFREEZE_REQUIREMENTS0)1truenamelocationreturncCsTt||}|sJd|d|t|}|js"Jd|d||j||S)NzFailed to load module z from zModuleSpec.loader is None for )rrloader exec_module)rrspecpyrG/home/ubuntu/.local/lib/python3.10/site-packages/lightning/__setup__.py_load_py_modules  r assistantz.actionsz assistant.py)rrcsLddttjtddD}dtrdndifdd |D}|d |d |d <|d |d |d<|d|d|d|d<|d|d|d<t|D]}d|ddd}||g||||<qQt| D]#}|d}t |dkr|d|dksqrt||||d<||=qrdd | D}t d||S)NcSsg|]}t|qSrr.0prrr "sz#_prepare_extras..*z*.txtunfreezenonemajorcsPi|]$}|jdvr|jjds|jjd|jtjd|j|jdqS))zdocs.txtzbase.txt_-) file_namepath_dirr)rparent startswithstem _ASSISTANTload_requirementsr common_argsrr %s *z#_prepare_extras..zfabric-strategieszfabric-examplesz fabric-allz fabric-testz fabric-devz pytorch-extrazpytorch-strategieszpytorch-examplesz pytorch-allz pytorch-testz pytorch-devr&rcSsi|] \}}|tt|qSr)sortedset)rrreqsrrrr0=szThe extras are: ) globospathjoin_PATH_REQUIREMENTS_FREEZE_REQUIREMENTSlistsplitgetkeyslenitemsprint) req_filesextrasextrarkkkrr.r_prepare_extrass*     rHcCs tdtjtd}tdtjtd}tjt|j|j d}tj t t r%dndd}|d g7}id d d|j d |j d |jd|jd|jddd|jdtdd dgddddid|ddddddd gd!d"d#d$d%d&d'gig|td(d)dd*gd+d,S)-Naboutz __about__.pyversionz__version__.py)homepagerJr#r$)r"zpytorch-lightningrr descriptionauthor author_emailurl download_urlz)https://github.com/Lightning-AI/lightninglicensepackagesr z lightning.*)whereinclude package_dirlong_descriptionlong_description_content_typez text/markdowninclude_package_dataTzip_safeFkeywords)z deep learningpytorchAIpython_requiresz>=3.9 entry_pointsconsole_scriptsz#fabric = lightning.fabric.cli:_mainz-lightning = lightning.fabric.cli:_legacy_mainz0https://github.com/Lightning-AI/lightning/issuesz#https://lightning.ai/lightning-docs)z Bug Tracker Documentationz Source Code) zEnvironment :: ConsolezNatural Language :: EnglishzDevelopment Status :: 4 - BetazIntended Audience :: Developersz:Topic :: Scientific/Engineering :: Artificial Intelligencez7Topic :: Scientific/Engineering :: Information Analysisz2License :: OSI Approved :: Apache Software Licensez"Operating System :: OS Independentz#Programming Language :: Python :: 3z%Programming Language :: Python :: 3.9z&Programming Language :: Python :: 3.10z&Programming Language :: Python :: 3.11z&Programming Language :: Python :: 3.12)setup_requiresinstall_requiresextras_require project_urls classifiers)rr7r8r9 _PACKAGE_ROOTr,load_readme_description _PROJECT_ROOT __homepage__rJr-r:r;__docs__ __author____author_email__ __license__r rH)rIrJrWrcrrr _setup_argsBst        ro)r6os.pathr7importlib.utilrrpathlibrtypesrtypingrr setuptoolsr rir8r9 _SOURCE_ROOTrgr:environr>lowerr;strrr,rHrorrrrs     %