From 1d8b81079321808acbed057ed311fdbb9694f606 Mon Sep 17 00:00:00 2001
From: Kaipeng Deng <dengkaipeng@baidu.com>
Date: Mon, 3 Aug 2020 15:42:51 +0800
Subject: [PATCH] fix test_architectures (#1147)

---
 ppdet/modeling/tests/test_architectures.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/ppdet/modeling/tests/test_architectures.py b/ppdet/modeling/tests/test_architectures.py
index ef8ebfc16..ab99f3088 100644
--- a/ppdet/modeling/tests/test_architectures.py
+++ b/ppdet/modeling/tests/test_architectures.py
@@ -19,6 +19,7 @@ from __future__ import print_function
 import unittest
 import numpy as np
 
+import paddle
 import paddle.fluid as fluid
 import os
 import sys
@@ -70,6 +71,10 @@ class TestCascadeRCNN(TestFasterRCNN):
         self.cfg_file = 'configs/cascade_rcnn_r50_fpn_1x.yml'
 
 
+@unittest.skipIf(
+    paddle.version.major < "2",
+    "Paddle 2.0 should be used for YOLOv3 takes scale_x_y as inputs, "
+    "disable this unittest for Paddle major version < 2")
 class TestYolov3(TestFasterRCNN):
     def set_config(self):
         self.cfg_file = 'configs/yolov3_darknet.yml'
-- 
GitLab