summaryrefslogtreecommitdiff
path: root/astroid/brain/brain_numpy_ndarray.py
diff options
context:
space:
mode:
authorNick Drozd <nicholasdrozd@gmail.com>2022-11-22 09:14:21 -0600
committerGitHub <noreply@github.com>2022-11-22 16:14:21 +0100
commit7a0b5626ae7011c682a1fdee2362823cb36adc69 (patch)
tree36afd4cf0882a6eb65730e9182d676c8321d998c /astroid/brain/brain_numpy_ndarray.py
parent595c8bb416153ef3470b6d2ec41de0316fd5a769 (diff)
downloadastroid-git-7a0b5626ae7011c682a1fdee2362823cb36adc69.tar.gz
Add some InferenceContext annotations (#1883)
Diffstat (limited to 'astroid/brain/brain_numpy_ndarray.py')
-rw-r--r--astroid/brain/brain_numpy_ndarray.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/astroid/brain/brain_numpy_ndarray.py b/astroid/brain/brain_numpy_ndarray.py
index e2429985..48db84eb 100644
--- a/astroid/brain/brain_numpy_ndarray.py
+++ b/astroid/brain/brain_numpy_ndarray.py
@@ -3,14 +3,17 @@
# Copyright (c) https://github.com/PyCQA/astroid/blob/main/CONTRIBUTORS.txt
"""Astroid hooks for numpy ndarray class."""
+from __future__ import annotations
+
from astroid.brain.brain_numpy_utils import numpy_supports_type_hints
from astroid.builder import extract_node
+from astroid.context import InferenceContext
from astroid.inference_tip import inference_tip
from astroid.manager import AstroidManager
from astroid.nodes.node_classes import Attribute
-def infer_numpy_ndarray(node, context=None):
+def infer_numpy_ndarray(node, context: InferenceContext | None = None):
ndarray = """
class ndarray(object):
def __init__(self, shape, dtype=float, buffer=None, offset=0,