Implement review feedback.

BlueBrain · Mar 25, 2024 · 8902618 · 8902618
1 parent 0068f4e
commit 8902618
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/atlas_densities/densities/fitting.py b/atlas_densities/densities/fitting.py
@@ -18,9 +18,10 @@
 """
 from __future__ import annotations
 
+import itertools as it
 import logging
 import warnings
-from typing import TYPE_CHECKING, Dict, List, Optional, Set, Union
+from typing import TYPE_CHECKING, Dict, List, Optional, Union
 
 import numpy as np
 import pandas as pd
@@ -623,10 +624,7 @@ def _get_group_region_names(groups):
         A list of region names.
     """
 
-    group_regions_names: Set[str] = set()
-    for v in groups.values():
-        group_regions_names = group_regions_names.union(v)
-    return list(group_regions_names)
+    return list(set(it.chain.from_iterable(groups.values())))
 
 
 def linear_fitting(  # pylint: disable=too-many-arguments