<commit_before>package org.corundummc.entities.living.animals;

import net.minecraft.entity.passive.*;
import org.corundummc.entities.Entity;
import org.corundummc.world.Location;

/**
 * The base for all animal entities.
 */
public class Animal extends Entity {
    public Animal(AnimalType animalType, Location location) {
        super(animalType, location);
        // TODO
    }

    // TODO constructors, utilities etc.

    public static class AnimalType<T extends AnimalType<T>> extends Entity.EntityType<T> {
        public static final AnimalType COW = new AnimalType(EntityCow.class),
                PIG = new AnimalType(EntityPig.class),
                CHICKEN = new AnimalType(EntityChicken.class),
                SHEEP = new AnimalType(EntitySheep.class),
                HORSE = new AnimalType(EntityHorse.class),
                MOOSHROOM = new AnimalType(EntityMooshroom.class);

        private Class<? extends EntityAnimal> mcAnimalClass;

        public AnimalType(Class<? extends EntityAnimal> clazz) {
            this.mcAnimalClass = clazz;
        }


    }
}<commit_msg>Added two missing animals from AnimalType.<commit_after>package org.corundummc.entities.living.animals;

import net.minecraft.entity.passive.*;
import org.corundummc.entities.Entity;
import org.corundummc.world.Location;

/**
 * The base for all animal entities.
 */
public class Animal extends Entity {
    public Animal(AnimalType animalType, Location location) {
        super(animalType, location);
        // TODO
    }

    // TODO constructors, utilities etc.

    public static class AnimalType<T extends AnimalType<T>> extends Entity.EntityType<T> {
        public static final AnimalType COW = new AnimalType(EntityCow.class),
                PIG = new AnimalType(EntityPig.class),
                CHICKEN = new AnimalType(EntityChicken.class),
                SHEEP = new AnimalType(EntitySheep.class),
                HORSE = new AnimalType(EntityHorse.class),
                MOOSHROOM = new AnimalType(EntityMooshroom.class),
                OCELOT = new AnimalType(EntityOcelot.class),
                WOLF = new AnimalType(EntityWolf.class);

        private Class<? extends EntityAnimal> mcAnimalClass;

        public AnimalType(Class<? extends EntityAnimal> clazz) {
            this.mcAnimalClass = clazz;
        }


    }
}
{'fraction_non_alphanumeric': 0.0892018779342723, 'fraction_numerical': 0.0, 'mean_word_length': 2.703001579778831, 'pattern_counts': {'":': 0, '<': 13, '<?xml version=': 0, '>': 13, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '6538711', 'n_tokens_mistral': 674, 'n_tokens_neox': 635, 'n_words': 157}