Apply ruff/Pylint rule PLR5501

PLR5501 Use `elif` instead of `else` then `if`, to reduce indentation

Author: DimitriPapadopoulos

src/zarr/core/codec_pipeline.py

@@ -405,13 +405,12 @@ async def _read_key(
             ):
                 if chunk_array is None:
                     chunk_array_batch.append(None)  # type: ignore[unreachable]
+                elif not chunk_spec.config.write_empty_chunks and chunk_array.all_equal(
+                    fill_value_or_default(chunk_spec)
+                ):
+                    chunk_array_batch.append(None)
                 else:
-                    if not chunk_spec.config.write_empty_chunks and chunk_array.all_equal(
-                        fill_value_or_default(chunk_spec)
-                    ):
-                        chunk_array_batch.append(None)
-                    else:
-                        chunk_array_batch.append(chunk_array)
+                    chunk_array_batch.append(chunk_array)
 
             chunk_bytes_batch = await self.encode_batch(
                 [

src/zarr/core/group.py

@@ -300,19 +300,18 @@ def flatten(
             children: dict[str, ArrayV2Metadata | ArrayV3Metadata | GroupMetadata] = {}
             if isinstance(group, ArrayV2Metadata | ArrayV3Metadata):
                 children[key] = group
+            elif group.consolidated_metadata and group.consolidated_metadata.metadata is not None:
+                children[key] = replace(
+                    group, consolidated_metadata=ConsolidatedMetadata(metadata={})
+                )
+                for name, val in group.consolidated_metadata.metadata.items():
+                    full_key = f"{key}/{name}"
+                    if isinstance(val, GroupMetadata):
+                        children.update(flatten(full_key, val))
+                    else:
+                        children[full_key] = val
             else:
-                if group.consolidated_metadata and group.consolidated_metadata.metadata is not None:
-                    children[key] = replace(
-                        group, consolidated_metadata=ConsolidatedMetadata(metadata={})
-                    )
-                    for name, val in group.consolidated_metadata.metadata.items():
-                        full_key = f"{key}/{name}"
-                        if isinstance(val, GroupMetadata):
-                            children.update(flatten(full_key, val))
-                        else:
-                            children[full_key] = val
-                else:
-                    children[key] = replace(group, consolidated_metadata=None)
+                children[key] = replace(group, consolidated_metadata=None)
             return children
 
         for k, v in self.metadata.items():
@@ -1257,9 +1256,8 @@ async def require_array(
             if exact:
                 if ds.dtype != dtype:
                     raise TypeError(f"Incompatible dtype ({ds.dtype} vs {dtype})")
-            else:
-                if not np.can_cast(ds.dtype, dtype):
-                    raise TypeError(f"Incompatible dtype ({ds.dtype} vs {dtype})")
+            elif not np.can_cast(ds.dtype, dtype):
+                raise TypeError(f"Incompatible dtype ({ds.dtype} vs {dtype})")
         except KeyError:
             ds = await self.create_array(name, shape=shape, dtype=dtype, **kwargs)
 
@@ -3103,22 +3101,21 @@ async def create_hierarchy(
                     else:
                         # Any other exception is a real error
                         raise extant_node
-                else:
-                    # this is a node that already exists, but a node with the same key was specified
-                    #  in nodes_parsed.
-                    if isinstance(extant_node, GroupMetadata):
-                        # a group already exists where we want to create a group
-                        if isinstance(proposed_node, ImplicitGroupMarker):
-                            # we have proposed an implicit group, which is OK -- we will just skip
-                            # creating this particular metadata document
-                            redundant_implicit_groups.append(key)
-                        else:
-                            # we have proposed an explicit group, which is an error, given that a
-                            # group already exists.
-                            raise ContainsGroupError(store, key)
-                    elif isinstance(extant_node, ArrayV2Metadata | ArrayV3Metadata):
-                        # we are trying to overwrite an existing array. this is an error.
-                        raise ContainsArrayError(store, key)
+                # this is a node that already exists, but a node with the same key was specified
+                #  in nodes_parsed.
+                elif isinstance(extant_node, GroupMetadata):
+                    # a group already exists where we want to create a group
+                    if isinstance(proposed_node, ImplicitGroupMarker):
+                        # we have proposed an implicit group, which is OK -- we will just skip
+                        # creating this particular metadata document
+                        redundant_implicit_groups.append(key)
+                    else:
+                        # we have proposed an explicit group, which is an error, given that a
+                        # group already exists.
+                        raise ContainsGroupError(store, key)
+                elif isinstance(extant_node, ArrayV2Metadata | ArrayV3Metadata):
+                    # we are trying to overwrite an existing array. this is an error.
+                    raise ContainsArrayError(store, key)
 
     nodes_explicit: dict[str, GroupMetadata | ArrayV2Metadata | ArrayV3Metadata] = {}
 
@@ -3278,13 +3275,12 @@ def _parse_hierarchy_dict(
             # If a component is not already in the output dict, add ImplicitGroupMetadata
             if subpath not in out:
                 out[subpath] = ImplicitGroupMarker(zarr_format=v.zarr_format)
-            else:
-                if not isinstance(out[subpath], GroupMetadata | ImplicitGroupMarker):
-                    msg = (
-                        f"The node at {subpath} contains other nodes, but it is not a Zarr group. "
-                        "This is invalid. Only Zarr groups can contain other nodes."
-                    )
-                    raise ValueError(msg)
+            elif not isinstance(out[subpath], GroupMetadata | ImplicitGroupMarker):
+                msg = (
+                    f"The node at {subpath} contains other nodes, but it is not a Zarr group. "
+                    "This is invalid. Only Zarr groups can contain other nodes."
+                )
+                raise ValueError(msg)
     return out
 
 
@@ -3492,12 +3488,11 @@ async def _read_metadata_v2(store: Store, path: str) -> ArrayV2Metadata | GroupM
     # return the array metadata.
     if zarray_bytes is not None:
         zmeta = json.loads(zarray_bytes.to_bytes())
+    elif zgroup_bytes is None:
+        # neither .zarray or .zgroup were found results in KeyError
+        raise FileNotFoundError(path)
     else:
-        if zgroup_bytes is None:
-            # neither .zarray or .zgroup were found results in KeyError
-            raise FileNotFoundError(path)
-        else:
-            zmeta = json.loads(zgroup_bytes.to_bytes())
+        zmeta = json.loads(zgroup_bytes.to_bytes())
 
     return _build_metadata_v2(zmeta, zattrs)