Fixed _get_item for Series (#50)

Author: zkrolikowski-vl

setup.py

@@ -21,7 +21,7 @@ def list_packages(source_path: str = src_path) -> None:
 setup(
     name="pandas-stubs",
     package_dir={"": src_path},
-    version="1.1.0.8",
+    version="1.1.0.9",
     description="Type annotations for Pandas",
     long_description=(open("README.md").read()
                       if os.path.exists("README.md") else ""),

tests/snippets/test_series.py

@@ -2,6 +2,7 @@
 
 import tempfile
 from pathlib import Path
+from typing import List
 
 import pandas as pd
 import numpy as np
@@ -406,3 +407,10 @@ def test_types_resample() -> None:
     s.resample('3T').sum()
     # origin and offset params added in 1.1.0 https://pandas.pydata.org/docs/whatsnew/v1.1.0.html
     s.resample('20min', origin='epoch', offset=pd.Timedelta(value=2, unit='minutes'))
+
+
+def test_types_getitem() -> None:
+    s = pd.Series({'key': [0, 1, 2, 3]})
+    key: List[int] = s['key']
+    s2 = pd.Series([0, 1, 2, 3])
+    value: int = s2[0]

third_party/3/pandas/core/series.pyi

@@ -47,7 +47,7 @@ class Series(base.IndexOpsMixin, generic.NDFrame):
     def __init__(self, data: Optional[Any] = ..., index: Optional[Any] = ..., dtype: Optional[Dtype] = ..., name: Optional[Any] = ..., copy: bool = ..., fastpath: bool = ...) -> None: ...
     def __len__(self) -> int: ...
     @overload
-    def __getitem__(self, key: Label) -> Scalar: ...
+    def __getitem__(self, key: Label) -> Any: ...
     @overload
     def __getitem__(self, key: slice) -> DataFrame: ...
     def __setitem__(self, key: Label, value: Scalar) -> None: ...