classes in python

classes.py

class Foo(object):

    def __init__(self, data):
        self.data = data
        self.shape = self.data.shape

    def __getitem__(self, index):
        return self.data[index] ** 2

    def __dir__(self):
        return ['data']