Skip to content

action_primitive_set_base

BaseActionPrimitiveSet

Bases: with_metaclass(ABCMeta, object)

Source code in omnigibson/action_primitives/action_primitive_set_base.py 
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
class BaseActionPrimitiveSet(with_metaclass(ABCMeta, object)):
    def __init_subclass__(cls, **kwargs):
        """
        Registers all subclasses as part of this registry. This is useful to decouple internal codebase from external
        user additions. This way, users can add their custom primitive set by simply extending this class,
        and it will automatically be registered internally. This allows users to then specify their primitive set
        directly in string-from in e.g., their config files, without having to manually set the str-to-class mapping
        in our code.
        """
        if not inspect.isabstract(cls):
            REGISTERED_PRIMITIVE_SETS[cls.__name__] = cls

    def __init__(self, env):
        self.env : Environment = env

    @property
    def robot(self):
        # Currently returns the first robot in the environment, but can be scaled to multiple robots
        # by creating multiple action generators and passing in a robot index etc.
        return self.env.robots[0]

    @abstractmethod
    def get_action_space(self):
        """Get the higher-level action space as an OpenAI Gym Space object."""
        pass

    @abstractmethod
    def apply(self, action):
        """
        Apply a primitive action.

        Given a higher-level action in the same format as the action space (e.g. as a number),
        generates a sequence of lower level actions (or raise ActionPrimitiveError). The action
        will get resolved and passed into apply_ref.
        """
        pass

    @abstractmethod
    def apply_ref(self, action, *args):
        """
        Apply a primitive action by reference.

        Given a higher-level action from the corresponding action set enum and any necessary arguments,
        generates a sequence of lower level actions (or raise ActionPrimitiveError)
        """
        pass

__init_subclass__(**kwargs)

Registers all subclasses as part of this registry. This is useful to decouple internal codebase from external user additions. This way, users can add their custom primitive set by simply extending this class, and it will automatically be registered internally. This allows users to then specify their primitive set directly in string-from in e.g., their config files, without having to manually set the str-to-class mapping in our code.

Source code in omnigibson/action_primitives/action_primitive_set_base.py 
54
55
56
57
58
59
60
61
62
63
def __init_subclass__(cls, **kwargs):
    """
    Registers all subclasses as part of this registry. This is useful to decouple internal codebase from external
    user additions. This way, users can add their custom primitive set by simply extending this class,
    and it will automatically be registered internally. This allows users to then specify their primitive set
    directly in string-from in e.g., their config files, without having to manually set the str-to-class mapping
    in our code.
    """
    if not inspect.isabstract(cls):
        REGISTERED_PRIMITIVE_SETS[cls.__name__] = cls

apply(action) abstractmethod

Apply a primitive action.

Given a higher-level action in the same format as the action space (e.g. as a number), generates a sequence of lower level actions (or raise ActionPrimitiveError). The action will get resolved and passed into apply_ref.

Source code in omnigibson/action_primitives/action_primitive_set_base.py 
79
80
81
82
83
84
85
86
87
88
@abstractmethod
def apply(self, action):
    """
    Apply a primitive action.

    Given a higher-level action in the same format as the action space (e.g. as a number),
    generates a sequence of lower level actions (or raise ActionPrimitiveError). The action
    will get resolved and passed into apply_ref.
    """
    pass

apply_ref(action, *args) abstractmethod

Apply a primitive action by reference.

Given a higher-level action from the corresponding action set enum and any necessary arguments, generates a sequence of lower level actions (or raise ActionPrimitiveError)

Source code in omnigibson/action_primitives/action_primitive_set_base.py 
90
91
92
93
94
95
96
97
98
@abstractmethod
def apply_ref(self, action, *args):
    """
    Apply a primitive action by reference.

    Given a higher-level action from the corresponding action set enum and any necessary arguments,
    generates a sequence of lower level actions (or raise ActionPrimitiveError)
    """
    pass

get_action_space() abstractmethod

Get the higher-level action space as an OpenAI Gym Space object.

Source code in omnigibson/action_primitives/action_primitive_set_base.py 
74
75
76
77
@abstractmethod
def get_action_space(self):
    """Get the higher-level action space as an OpenAI Gym Space object."""
    pass