Cleanup custom check overhead (#1625)

hacs · Nov 5, 2020 · 445e8ac · 445e8ac
1 parent 1a0437b
commit 445e8ac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/custom_components/hacs/helpers/properties/custom.py b/custom_components/hacs/helpers/properties/custom.py
@@ -6,7 +6,7 @@ class RepositoryPropertyCustom(ABC):
     @property
     def custom(self):
         """Return flag if the repository is custom."""
-        if str(self.data.id) in [str(x) for x in self.hacs.common.default]:
+        if str(self.data.id) in self.hacs.common.default:
             return False
         if self.data.full_name == "hacs/integration":
             return False

diff --git a/tests/repositories/helpers/test_properties.py b/tests/repositories/helpers/test_properties.py
@@ -16,7 +16,7 @@ def test_repository_helpers_properties_custom():
     assert repository.custom
 
     repository.data.id = 1337
-    repository.hacs.common.default.append(repository.data.id)
+    repository.hacs.common.default.append(str(repository.data.id))
     assert not repository.custom
 
     repository.hacs.common.default = []