From cc90134403b6133901561a0f2e6dad2cc4c0d266 Mon Sep 17 00:00:00 2001 From: mgrover1 Date: Fri, 29 Sep 2023 09:59:46 -0500 Subject: [PATCH] FIX: Cleanup the type hinting --- pyart/xradar/accessor.py | 18 ++---------------- 1 file changed, 2 insertions(+), 16 deletions(-) diff --git a/pyart/xradar/accessor.py b/pyart/xradar/accessor.py index c767170f24..2f236b799f 100644 --- a/pyart/xradar/accessor.py +++ b/pyart/xradar/accessor.py @@ -5,14 +5,12 @@ import copy -from collections.abc import Hashable, Mapping -from typing import Any, overload import numpy as np import pandas as pd from datatree import DataTree, formatting, formatting_html from datatree.treenode import NodePath -from xarray import DataArray, Dataset, concat +from xarray import concat from xarray.core import utils from ..core.transforms import antenna_vectors_to_cartesian @@ -67,19 +65,7 @@ def __repr__(self): def _repr_html_(self): return formatting_html.datatree_repr(self.xradar) - @overload - def __getitem__(self, key: Mapping) -> Dataset: # type: ignore[misc] - ... - - @overload - def __getitem__(self, key: Hashable) -> DataArray: # type: ignore[misc] - ... - - @overload - def __getitem__(self, key: Any) -> Dataset: - ... - - def __getitem__(self: DataTree, key: str) -> DataTree | DataArray: + def __getitem__(self: DataTree, key): """ Access child nodes, variables, or coordinates stored anywhere in this tree.