predict: fix double periodic restarts

koush · Apr 19, 2024 · ffc9ca1 · ffc9ca1
1 parent 9b349fd
commit ffc9ca1
Show file tree

Hide file tree

Showing 7 changed files with 13 additions and 12 deletions.
diff --git a/plugins/coreml/package-lock.json b/plugins/coreml/package-lock.json
diff --git a/plugins/coreml/package.json b/plugins/coreml/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.40"
+   "version": "0.1.41"
 }
diff --git a/plugins/openvino/package-lock.json b/plugins/openvino/package-lock.json
diff --git a/plugins/openvino/package.json b/plugins/openvino/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.73"
+   "version": "0.1.74"
 }
diff --git a/plugins/tensorflow-lite/package-lock.json b/plugins/tensorflow-lite/package-lock.json
diff --git a/plugins/tensorflow-lite/package.json b/plugins/tensorflow-lite/package.json
@@ -53,5 +53,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.55"
+   "version": "0.1.56"
 }
diff --git a/plugins/tensorflow-lite/src/predict/__init__.py b/plugins/tensorflow-lite/src/predict/__init__.py
@@ -40,9 +40,10 @@ class PredictPlugin(DetectPlugin):
     def __init__(self, nativeId: str | None = None):
         super().__init__(nativeId=nativeId)
 
-        # periodic restart because there seems to be leaks in tflite or coral API.
-        loop = asyncio.get_event_loop()
-        loop.call_later(4 * 60 * 60, lambda: self.requestRestart())
+        # periodic restart of main plugin because there seems to be leaks in tflite or coral API.
+        if not nativeId:
+            loop = asyncio.get_event_loop()
+            loop.call_later(4 * 60 * 60, lambda: self.requestRestart())
 
         self.batch: List[Tuple[Any, asyncio.Future]] = []
         self.batching = 0