【AI脚本工程师养成】1. 把Python代码封装成库

作为一个乙方的算法研究员，疲于应付各种项目，每天至少要写几百行脚本，虽然绝大多数都是一次性代码，但是也有少量有复用价值的代码，所以准备把日常用到的所有代码整理成一个可以随时调用的库。
今天是2022年10月7日，记录一下怎么把Python代码封装成库。

1. 项目结构：

项目结构

每个模块下面都有一个init.py文件

2. 编写setup.py

from setuptools import setup, find_packages

setup(
    name='dl',
    version='0.1',
    description='deeplearning toolkit',
    author='dalalaa',
    packages=find_packages()
)

3. 编写__init__.py

dl/__init__.py

__all__ = ['module1', 'module2']

dl/module1/__init__.py

__all__ = ['add']

dl/module2/__init__.py

__all__ = ['minus']

4. 打包成whl

python setup.py bdist_wheel

5. 安装whl

pip install dist/dl-0.1-py3-none-any.whl

6. 使用

经过上述步骤，这个dl库就可以在任意目录中使用了

image.png