From 7e7669546578ae959f57e4c80e0ea4329356bc4b Mon Sep 17 00:00:00 2001
From: Naval Chand <chandnaval17@gmail.com>
Date: Mon, 26 Aug 2019 08:45:04 +0530
Subject: [PATCH] Auto annotation (#660)

* Added missing import
* Added OPENVINO_TOOLKIT param in docker-compose.yml for auto annotation
---
 cvat/apps/auto_annotation/inference.py | 2 +-
 docker-compose.yml                     | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/cvat/apps/auto_annotation/inference.py b/cvat/apps/auto_annotation/inference.py
index 607b6b73..8d3442e6 100644
--- a/cvat/apps/auto_annotation/inference.py
+++ b/cvat/apps/auto_annotation/inference.py
@@ -1,6 +1,6 @@
 from .model_loader import ModelLoader
 from cvat.apps.engine.utils import import_modules, execute_python_code
-
+import itertools
 
 def _process_detections(detections, path_to_conv_script, restricted=True):
     results = Results()
diff --git a/docker-compose.yml b/docker-compose.yml
index 2f521520..3fb7b8bf 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -49,6 +49,7 @@ services:
         USER: "django"
         DJANGO_CONFIGURATION: "production"
         WITH_TESTS: "no"
+        OPENVINO_TOOLKIT: "no"
     environment:
       DJANGO_MODWSGI_EXTRA_ARGS: ""
     volumes: