Fix dependencies

Signed-off-by: Beat Buesser <beat.buesser@ibm.com>

Author: beat-buesser

examples/get_started_yolo.py

@@ -208,10 +208,8 @@ def forward(self, x, targets=None):
             else:
                 return self.model(x)
 
-    # model_path = "./yolov3.cfg"
-    # weights_path = "./yolov3.weights"
-    model_path = "/tmp/PyTorch-YOLOv3/config/yolov3.cfg"
-    weights_path = "/tmp/PyTorch-YOLOv3/weights/yolov3.weights"
+    model_path = "./yolov3.cfg"
+    weights_path = "./yolov3.weights"
     model = load_model(model_path=model_path, weights_path=weights_path)
 
     model = Yolo(model)

tests/estimators/object_detection/conftest.py

@@ -16,6 +16,7 @@
 # TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 import logging
+
 import numpy as np
 import pytest