def get_allowed_categories(version):
    """
    Modelnet40 categories
    0  -  airplane
    1  -  bathtub
    2  -  bed
    3  -  bench
    4  -  bookshelf
    5  -  bottle
    6  -  bowl
    7  -  car
    8  -  chair
    9  -  cone
    10 -  cup
    11 -  curtain
    12 -  desk
    13 -  door
    14 -  dresser
    15 -  flower_pot
    16 -  glass_box
    17 -  guitar
    18 -  keyboard
    19 -  lamp
    20 -  laptop
    21 -  mantel
    22 -  monitor
    23 -  night_stand
    24 -  person
    25 -  piano
    26 -  plant
    27 -  radio
    28 -  range_hood
    29 -  sink
    30 -  sofa
    31 -  stairs
    32 -  stool
    33 -  table
    34 -  tent
    35 -  toilet
    36 -  tv_stand
    37 -  vase
    38 -  wardrobe
    39 -  xbox
    """
    assert version in ["small", "big", "airplane", "chair", "sofa", "toilet"]
    if version == "big":
        return [0, 2, 5, 6, 7, 8, 17, 30, 35]
    if version == "small":
        return [0, 7, 8, 35]
    if version == "airplane":
        return [0]
    if version == "chair":
        return [8]
    if version == "sofa":
        return [20]
    if version == "toilet":
        return [35]
    return 0