Merge 6aa4f49 into 34676c5

zigpy · Aug 12, 2019 · f4c3e00 · f4c3e00
2 parents 34676c5 + 6aa4f49
commit f4c3e00
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/zigpy/__init__.py b/zigpy/__init__.py
@@ -1,6 +1,6 @@
 # coding: utf-8
 MAJOR_VERSION = 0
 MINOR_VERSION = 7
-PATCH_VERSION = '0'
+PATCH_VERSION = '1'
 __short_version__ = '{}.{}'.format(MAJOR_VERSION, MINOR_VERSION)
 __version__ = '{}.{}'.format(__short_version__, PATCH_VERSION)
diff --git a/zigpy/quirks/registry.py b/zigpy/quirks/registry.py
@@ -1,6 +1,7 @@
 import collections
 import itertools
 import logging
+import zigpy.quirks
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -35,6 +36,8 @@ def remove(self, custom_device):
 
     def get_device(self, device):
         """Get a CustomDevice object, if one is available"""
+        if isinstance(device, zigpy.quirks.CustomDevice):
+            return device
         dev_ep = set(device.endpoints) - set([0])
         _LOGGER.debug("Checking quirks for %s %s (%s)",
                       device.manufacturer, device.model, device.ieee)