From 9be3d96fa7e65f53d9ffc0f6f3fdb2274c1fa078 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 23 Jan 2021 12:51:04 -0800
Subject: [PATCH] Update requirements.txt (#2021)

* Update requirements.txt

* Update ci-testing.yml

* Update hubconf.py
---
 .github/workflows/ci-testing.yml | 1 +
 hubconf.py                       | 2 +-
 requirements.txt                 | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
index 5999ee6a9055..df508474a955 100644
--- a/.github/workflows/ci-testing.yml
+++ b/.github/workflows/ci-testing.yml
@@ -74,6 +74,7 @@ jobs:
           python test.py --img 128 --batch 16 --weights weights/${{ matrix.model }}.pt --device $di
           python test.py --img 128 --batch 16 --weights runs/train/exp/weights/last.pt --device $di
 
+          python hubconf.py  # hub
           python models/yolo.py --cfg models/${{ matrix.model }}.yaml  # inspect
           python models/export.py --img 128 --batch 1 --weights weights/${{ matrix.model }}.pt  # export
         shell: bash
diff --git a/hubconf.py b/hubconf.py
index c4485a42e335..2a34813310e8 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -137,5 +137,5 @@ def custom(path_or_model='path/to/model.pt', autoshape=True):
 
     imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')]
     results = model(imgs)
-    results.show()
     results.print()
+    results.save()
diff --git a/requirements.txt b/requirements.txt
index 3c23f2b750a2..2b447aef4d39 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ matplotlib>=3.2.2
 numpy>=1.18.5
 opencv-python>=4.1.2
 Pillow
-PyYAML>=5.3
+PyYAML>=3.13
 scipy>=1.4.1
 tensorboard>=2.2
 torch>=1.7.0