fix release bug of AbstractInferenceModel (#1556)

* fix release bug of AbstractInferenceModel * fix release bug of AbstractInferenceModel * fix bug
intel-analytics · Aug 8, 2019 · b98d97a · b98d97a
1 parent 5e3f3ef
commit b98d97a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/zoo/src/main/java/com/intel/analytics/zoo/pipeline/inference/AbstractInferenceModel.java b/zoo/src/main/java/com/intel/analytics/zoo/pipeline/inference/AbstractInferenceModel.java
@@ -117,7 +117,7 @@ public void reload(String modelPath, String weightPath) {
   }
 
   public void release() {
-    release();
+    doRelease();
   }
 
   @Deprecated

diff --git a/zoo/src/main/scala/com/intel/analytics/zoo/pipeline/inference/InferenceModel.scala b/zoo/src/main/scala/com/intel/analytics/zoo/pipeline/inference/InferenceModel.scala
@@ -523,7 +523,7 @@ class InferenceModel(private var autoScalingEnabled: Boolean = true,
   /**
    * release original model and all the cloned ones in the queue
    */
-  def release(): Unit = {
+  def doRelease(): Unit = {
     clearModelQueue()
   }
 

diff --git a/zoo/src/test/scala/com/intel/analytics/zoo/pipeline/inference/OpenVINOModelSuite.scala b/zoo/src/test/scala/com/intel/analytics/zoo/pipeline/inference/OpenVINOModelSuite.scala
@@ -157,7 +157,7 @@ class OpenVINOModelSuite extends FunSuite with Matchers with BeforeAndAfterAll
     threads2.foreach(_.join())
 
     fasterrcnnModel.release()
-    fasterrcnnInferenceModel.release()
+    fasterrcnnInferenceModel.doRelease()
   }
 
   def almostEqual(x: Float, y: Float, precision: Float): Boolean = {