[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

machine_learning/apriori_algorithm.py

@@ -73,7 +73,9 @@ def find_frequent_itemsets(self) -> List[Dict[frozenset, float]]:
 
         total: int = len(self.transactions)
         current_itemsets: Dict[frozenset, float] = {
-            k: v / total for k, v in item_counts.items() if v / total >= self.min_support
+            k: v / total
+            for k, v in item_counts.items()
+            if v / total >= self.min_support
         }
         if current_itemsets:
             self.itemsets.append(current_itemsets)
@@ -92,7 +94,9 @@ def find_frequent_itemsets(self) -> List[Dict[frozenset, float]]:
                         candidates.add(union)
 
             freq_candidates: Dict[frozenset, float] = {
-                c: self._get_support(c) for c in candidates if self._get_support(c) >= self.min_support
+                c: self._get_support(c)
+                for c in candidates
+                if self._get_support(c) >= self.min_support
             }
             if not freq_candidates:
                 break
@@ -103,7 +107,9 @@ def find_frequent_itemsets(self) -> List[Dict[frozenset, float]]:
 
         return self.itemsets
 
-    def generate_association_rules(self) -> List[Tuple[frozenset, frozenset, float, float]]:
+    def generate_association_rules(
+        self,
+    ) -> List[Tuple[frozenset, frozenset, float, float]]:
         """Generate association rules with min confidence and lift."""
         for level in self.itemsets:
             for itemset in level:
@@ -121,7 +127,11 @@ def generate_association_rules(self) -> List[Tuple[frozenset, frozenset, float,
                             conf,
                             lft,
                         )
-                        if rule not in self.rules and conf >= self.min_confidence and lft >= self.min_lift:
+                        if (
+                            rule not in self.rules
+                            and conf >= self.min_confidence
+                            and lft >= self.min_lift
+                        ):
                             self.rules.append(rule)
         return self.rules