重构结构

Author: bobo0810

.github/workflows/pytest.yaml

@@ -15,7 +15,8 @@ jobs:
       uses: actions/checkout@main
     - name: Installation depends # 安装依赖
       run: | 
-        pip install -r requirements.txt
+        pip install -r bobotools/requirements.txt
     - name: Start test # 自动化测试
       run: | 
+        cd bobotools/
         pytest -s test.py 

bobotools/common/tools.py bobotools/com.pybobotools/common/tools.py renamed to bobotools/com.py

@@ -1,6 +1,8 @@
 import torchvision.models as models
-from bobotools.torch_tools import Torch_Tools
-
+from .torch_tools import Torch_Tools
+'''
+pytest自动化测试
+'''
 #=================Torch工具类==============================
 def test_get_model_info():
     model = models.resnet18(pretrained=False)

requirements.txt bobotools/requirements.txtrequirements.txt renamed to bobotools/requirements.txt

@@ -1,4 +1,4 @@
-from .common.tools import get_model_size,get_model_time
+from .com import get_model_size,get_model_time
 class Torch_Tools(object):
     """
     Pytorch操作

test.py bobotools/test.pytest.py renamed to bobotools/test.py

@@ -14,7 +14,7 @@
     description="bobotools",
     long_description=open("README.md", "r").read(),
     long_description_content_type="text/markdown",
-    version="0.4.5",  # 版本
+    version="0.4.6",  # 版本
     install_requires=install_requires,
     python_requires=">=3.6",
     include_package_data=True,