Black format

Author: robmarkcole

.gitignore

@@ -20,3 +20,118 @@ deepstack_python.egg-info*
 # should NOT be excluded as they contain compiler settings and other important
 # information for Eclipse / Flash Builder.
 
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/

deepstack/__pycache__/core.cpython-36.pyc

@@ -6,7 +6,7 @@
 from PIL import Image
 
 ## Const
-CLASSIFIER = 'deepstack'
+CLASSIFIER = "deepstack"
 HTTP_OK = 200
 HTTP_BAD_REQUEST = 400
 HTTP_UNAUTHORIZED = 401
@@ -18,8 +18,11 @@ def get_matched_faces(predictions: dict):
     Get the predicted faces and their confidence.
     """
     try:
-        matched_faces = {face['userid']: round(face['confidence']*100, 1)
-                         for face in predictions if not face['userid'] == 'unknown'}
+        matched_faces = {
+            face["userid"]: round(face["confidence"] * 100, 1)
+            for face in predictions
+            if not face["userid"] == "unknown"
+        }
         return matched_faces
     except:
         return {}
@@ -34,7 +37,7 @@ def is_valid_image(file_path: str):
             pass
 
         image_extension = imghdr.what(file_path)
-        if image_extension in ['jpeg', '.jpg', '.png']:
+        if image_extension in ["jpeg", ".jpg", ".png"]:
             return True
         return False
     except Exception as exc:
@@ -45,11 +48,7 @@ def is_valid_image(file_path: str):
 def post_image(url: str, image: bytes):
     """Post an image to the classifier."""
     try:
-        response = requests.post(
-            url,
-            files={"image": image},
-            timeout=TIMEOUT
-        )
+        response = requests.post(url, files={"image": image}, timeout=TIMEOUT)
         return response
     except requests.exceptions.ConnectionError:
         print("ConnectionError: Is %s running?", CLASSIFIER)
@@ -59,13 +58,14 @@ def post_image(url: str, image: bytes):
         return None
 
 
-class DeepstackFace():
+class DeepstackFace:
     """Work with faces."""
 
     def __init__(self, ip_address: str, port: str):
 
         self._url_check = "http://{}:{}/v1/vision/face/recognize".format(
-            ip_address, port)
+            ip_address, port
+        )
 
         self._faces = None
         self._matched = {}
@@ -78,8 +78,7 @@ def process_file(self, file_path: str):
 
     def process_image_bytes(self, image_bytes: bytes):
         """Process an image."""
-        response = post_image(
-            self._url_check, image_bytes)
+        response = post_image(self._url_check, image_bytes)
         if response:
             if response.status_code == HTTP_OK:
                 predictions_json = response.json()["predictions"]
@@ -94,7 +93,7 @@ def process_image_bytes(self, image_bytes: bytes):
     def attributes(self):
         """Return the classifier attributes."""
         return {
-            'faces': self._faces,
-            'matched_faces': self._matched,
-            'total_matched_faces': len(self._matched),
+            "faces": self._faces,
+            "matched_faces": self._matched,
+            "total_matched_faces": len(self._matched),
         }

deepstack/core.py

@@ -41,8 +41,8 @@
      "output_type": "stream",
      "text": [
       "{'faces': 2, 'matched_faces': {}, 'total_matched_faces': 0}\n",
-      "CPU times: user 362 ms, sys: 47.5 ms, total: 409 ms\n",
-      "Wall time: 12.8 s\n"
+      "CPU times: user 367 ms, sys: 52.6 ms, total: 420 ms\n",
+      "Wall time: 14.9 s\n"
      ]
     },
     {