From 879e91a8f9e3d65893a357ffe469cd9335009dd2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=B9=E5=89=8D=E6=98=8E?= <909355014@qq.com>
Date: Thu, 22 Sep 2022 18:33:45 +0800
Subject: [PATCH] helmet2 bug

---
 apps/ai/main.py       | 1 +
 apps/develop/views.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/apps/ai/main.py b/apps/ai/main.py
index 6ad09b19..9cf04977 100644
--- a/apps/ai/main.py
+++ b/apps/ai/main.py
@@ -7,6 +7,7 @@ myLogger = logging.getLogger('log')
 
 algo_dict = {
     "helmet": "apps.ai.client.helmet",
+    "helmet2": "apps.ai.client.helmet2",
     "fire1": "apps.ai.client.fire1"
 }
 
diff --git a/apps/develop/views.py b/apps/develop/views.py
index f5f1f203..47113eda 100755
--- a/apps/develop/views.py
+++ b/apps/develop/views.py
@@ -214,7 +214,7 @@ class TestViewSet(CustomGenericViewSet):
         """
         code = request.data['code']
         vc = TDevice.objects.get(code=code)
-        snap_and_analyse(vc, ['fire1'], Opl.objects.get(id='1568880208688320512'))
+        snap_and_analyse(vc, ['fire1', 'helmet'], Opl.objects.get(id='1572195171456651264'))
         return Response()
 
     @action(methods=['post'], detail=False, serializer_class=Serializer)