def __getitem__()

in data/habitat_data.py [0:0]

• 21 lines of code
• 9 McCabe index (conditional complexity)

    def __getitem__(self, item):
        if not (self.train) and (self.val_index < len(self.fixed_val_images)):
            if self.fixed_val_images[self.val_index]:
                data = self.fixed_val_images[self.val_index]
                self.val_index += 1
                return data

        if self.image_generator is None:
            print(
                "Restarting image_generator.... with seed %d in train mode? %s"
                % (self.seed, self.train),
                flush=True,
            )
            self.__restart__()

        if self.restarted:
            self.restart(self.train)
            self.restarted = False

        # Ignore the item and just generate an image
        data = self.image_generator.get_sample(item, self.num_views, self.train)

        if not (self.train) and (self.val_index < len(self.fixed_val_images)):
            self.fixed_val_images[self.val_index] = data

            self.val_index += 1

        return data