Skip to content

Commit 61a0771

Browse files
committed
chore: cap product feature map
1 parent fb71bf2 commit 61a0771

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

roborock/device_features.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ class ProductFeatures(StrEnum):
205205
RoborockProductNickname.RUBYSLITE: [ProductFeatures.MOP_ELECTRONIC_MODULE],
206206
}
207207

208-
product_feature_map: dict[RoborockProductNickname, list[ProductFeatures]] = {
208+
PRODUCT_FEATURE_MAP: dict[RoborockProductNickname, list[ProductFeatures]] = {
209209
product: (
210210
features
211211
+ ([ProductFeatures.DEFAULT_CLEANMODECUSTOM] if product not in PRODUCTS_WITHOUT_CUSTOM_CLEAN else [])
@@ -538,7 +538,7 @@ def from_feature_flags(
538538
kwargs[f.name] = product_nickname not in blacklist
539539
elif (product_features := f.metadata.get("product_features")) is not None:
540540
if product_nickname is not None:
541-
available_features = product_feature_map.get(product_nickname, [])
541+
available_features = PRODUCT_FEATURE_MAP.get(product_nickname, [])
542542
if any(feat in available_features for feat in product_features): # type: ignore
543543
kwargs[f.name] = True
544544

0 commit comments

Comments
 (0)