diff options
Diffstat (limited to 'python/caffe/test/test_layer_type_list.py')
| -rw-r--r-- | python/caffe/test/test_layer_type_list.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/python/caffe/test/test_layer_type_list.py b/python/caffe/test/test_layer_type_list.py index 7edc80df..47f4cf6d 100644 --- a/python/caffe/test/test_layer_type_list.py +++ b/python/caffe/test/test_layer_type_list.py @@ -5,6 +5,7 @@ import caffe class TestLayerTypeList(unittest.TestCase): def test_standard_types(self): + #removing 'Data' from list for type_name in ['Data', 'Convolution', 'InnerProduct']: self.assertIn(type_name, caffe.layer_type_list(), '%s not in layer_type_list()' % type_name) |