Add validated inference model

92cea10 verified 19 days ago

272 Bytes

import abc
from typing import Dict


class BasePolicy(abc.ABC):

    @abc.abstractmethod
    def infer(self, obs: Dict) -> Dict:
        """Infer actions from observations."""

    def reset(self) -> None:
        """Reset the policy to its initial state."""
        pass

	import abc
	from typing import Dict


	class BasePolicy(abc.ABC):

	@abc.abstractmethod
	def infer(self, obs: Dict) -> Dict:
	"""Infer actions from observations."""

	def reset(self) -> None:
	"""Reset the policy to its initial state."""
	pass