optimum_benchmark/generators/task_generator.py [101:111]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def labels(self):
        self.assert_not_missing_shapes(["batch_size"])

        return self.generate_random_integers(
            min_value=0,
            max_value=self.shapes.get("num_labels", DEFAULT_NUM_LABELS),
            shape=(self.shapes["batch_size"],),
        )

    def __call__(self):
        dummy = {}
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



optimum_benchmark/generators/task_generator.py [282:292]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def labels(self):
        self.assert_not_missing_shapes(["batch_size"])

        return self.generate_random_integers(
            min_value=0,
            max_value=self.shapes.get("num_labels", DEFAULT_NUM_LABELS),
            shape=(self.shapes["batch_size"],),
        )

    def __call__(self):
        dummy = {}
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -