# Python imports.
import numpy as np

# Other imports.
from ..StateClass import State

class D4RLAntMazeState(State):
    def __init__(self, position, others, done):
        """
        Args:
            position (np.ndarray)
            others (np.ndarray)
            done (bool)
        """
        self.position = position
        self.others = others
        features = self.position.tolist() + self.others.tolist()

        State.__init__(self, data=features, is_terminal=done)

    def __str__(self):
        return "x: {}\ty: {}\tothers:{}\tterminal: {}\n".format(self.position[0],
                                                                self.position[1],
                                                                self.others,
                                                                self.is_terminal())

    def __repr__(self):
        return str(self)

6847fff27c97f199992923f974046d9b3239c1bf
<mask token>


class D4RLAntMazeState(State):

    def __init__(self, position, others, done):
        """
        Args:
            position (np.ndarray)
            others (np.ndarray)
            done (bool)
        """
        self.position = position
        self.others = others
        features = self.position.tolist() + self.others.tolist()
        State.__init__(self, data=features, is_terminal=done)
    <mask token>
    <mask token>

<mask token>


class D4RLAntMazeState(State):

    def __init__(self, position, others, done):
        """
        Args:
            position (np.ndarray)
            others (np.ndarray)
            done (bool)
        """
        self.position = position
        self.others = others
        features = self.position.tolist() + self.others.tolist()
        State.__init__(self, data=features, is_terminal=done)

    def __str__(self):
        return 'x: {}\ty: {}\tothers:{}\tterminal: {}\n'.format(self.
            position[0], self.position[1], self.others, self.is_terminal())
    <mask token>

<mask token>


class D4RLAntMazeState(State):

    def __init__(self, position, others, done):
        """
        Args:
            position (np.ndarray)
            others (np.ndarray)
            done (bool)
        """
        self.position = position
        self.others = others
        features = self.position.tolist() + self.others.tolist()
        State.__init__(self, data=features, is_terminal=done)

    def __str__(self):
        return 'x: {}\ty: {}\tothers:{}\tterminal: {}\n'.format(self.
            position[0], self.position[1], self.others, self.is_terminal())

    def __repr__(self):
        return str(self)

import numpy as np
from ..StateClass import State


class D4RLAntMazeState(State):

    def __init__(self, position, others, done):
        """
        Args:
            position (np.ndarray)
            others (np.ndarray)
            done (bool)
        """
        self.position = position
        self.others = others
        features = self.position.tolist() + self.others.tolist()
        State.__init__(self, data=features, is_terminal=done)

    def __str__(self):
        return 'x: {}\ty: {}\tothers:{}\tterminal: {}\n'.format(self.
            position[0], self.position[1], self.others, self.is_terminal())

    def __repr__(self):
        return str(self)

# Python imports.
import numpy as np

# Other imports.
from ..StateClass import State

class D4RLAntMazeState(State):
    def __init__(self, position, others, done):
        """
        Args:
            position (np.ndarray)
            others (np.ndarray)
            done (bool)
        """
        self.position = position
        self.others = others
        features = self.position.tolist() + self.others.tolist()

        State.__init__(self, data=features, is_terminal=done)

    def __str__(self):
        return "x: {}\ty: {}\tothers:{}\tterminal: {}\n".format(self.position[0],
                                                                self.position[1],
                                                                self.others,
                                                                self.is_terminal())

    def __repr__(self):
        return str(self)
