Add tests before pushing to Docker Hub (#3924)

2023-07-24 23:56:16 +02:00
parent 86b3c001c4
commit 2ee147838a
7 changed files with 71 additions and 69 deletions
--- a/docker/Dockerfile-python
+++ b/docker/Dockerfile-python
@ -28,7 +28,7 @@ ADD https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt /u

 # Install pip packages
 RUN python3 -m pip install --upgrade pip wheel
-RUN pip install --no-cache -e '.[export]' thop py-cpuinfo --extra-index-url https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache -e '.[export]' thop --extra-index-url https://download.pytorch.org/whl/cpu

 # Run exports to AutoInstall packages
 WORKDIR /tmp_exports