深度阅读

使用pypi将脚本打包成命令行

作者
作者
2023年08月22日
更新时间
10.12 分钟
阅读时间
0
阅读量

核心发布包时候,添加如下

setup.py

setup(
    name=name, #修改包名字-
    version=version,
    description=description,
    author=author,
    author_email=author_email,
    url=url,
    # install_requires=read_requirements('requirements.txt'),  # 指定需要安装的依赖
    long_description=long_description,
    long_description_content_type="text/markdown",
    install_requires=[
        'pytorch-lightning>=1.4.0',
        'tkitAutoTokenizerPosition>=0.0.0.3',
        'pytorch-crf>=0.7.2',
        'transformers>=4.9.2',
        'tkitJson>=0.0.0.3',

    ],
    packages=['tkitTagger'],
    # 这里封装命令行操作
    entry_points = { 
        'console_scripts': ['taggerTrainer=tkitTagger.TaggerTrainer:main'], 
    }
    # scripts=['tkitTagger/TaggerTrainer.py'],
    )

 ```

# 脚本文件
./tkitTagger/TaggerTrainer.py

```python
from pytorch_lightning.utilities.cli import LightningCLI
from tkitTagger.model import autoModel

def main():
    cli = LightningCLI(autoModel,save_config_overwrite=True)

if __name__ == "__main__": 
  main()

博客作者

热爱技术,乐于分享,持续学习。专注于Web开发、系统架构设计和人工智能领域。