utils/multimodalutils.py

from enum import IntEnum
import numpy as np
import gym.spaces as spaces
import torch

raise DeprecationWarning("Do not use this. Grammar is defined in the env class; SocialAIGrammar is socialaigrammar.py")

# class Grammar(object):
#
#     templates = ["Where is ", "Who is"]
#     things = ["me", "exit", "you", "him", "task"]
#
#     grammar_action_space = spaces.MultiDiscrete([len(templates), len(things)])
#
#     @classmethod
#     def construct_utterance(cls, action):
#         return cls.templates[int(action[0])] + " " + cls.things[int(action[1])] + ". "