fix PYTHONPATH

models/mobilenetv1.py

@@ -255,7 +255,7 @@ class MobileNetV1(Model):
         if self.with_pool:
             y = self.pool2d_avg(y)
 
-        if self.num_classes > -1:
+        if self.num_classes > 0:
             y = fluid.layers.reshape(y, shape=[-1, 1024])
             y = self.out(y)
         return y

tests/test_callbacks.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# when test, you should add hapi root path to the PYTHONPATH,
+# export PYTHONPATH=PATH_TO_HAPI:$PYTHONPATH
 import unittest
 import time
 import random

tests/test_datasets.py

@@ -12,9 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# when test, you should add hapi root path to the PYTHONPATH,
+# export PYTHONPATH=PATH_TO_HAPI:$PYTHONPATH
 import unittest
-import sys
-sys.path.append('..')
 
 from datasets.folder import DatasetFolder
 

tests/test_model.py

@@ -15,13 +15,13 @@
 from __future__ import division
 from __future__ import print_function
 
+# when test, you should add hapi root path to the PYTHONPATH,
+# export PYTHONPATH=PATH_TO_HAPI:$PYTHONPATH
+
 import unittest
 
 import os
 
-import sys
-sys.path.append('../')
-
 import numpy as np
 import contextlib
 

tests/test_progressbar.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# when test, you should add hapi root path to the PYTHONPATH,
+# export PYTHONPATH=PATH_TO_HAPI:$PYTHONPATH
 import unittest
 import random
 import time

tests/test_transforms.py

@@ -12,9 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# when test, you should add hapi root path to the PYTHONPATH,
+# export PYTHONPATH=PATH_TO_HAPI:$PYTHONPATH
 import unittest
-import sys
-sys.path.append('..')
 
 from datasets.folder import DatasetFolder
 from transform import transforms