23 lines
1.6 KiB
Python
23 lines
1.6 KiB
Python
# © Crown-owned copyright 2025, Defence Science and Technology Laboratory UK
|
|
# flake8: noqa
|
|
# Pre-import all the observations when we load up the observations module so that they can be resolved by the parser.
|
|
"""Converts PrimAITE simulation data into numerical RL-ready observations."""
|
|
from primaite.game.agent.observations.acl_observation import ACLObservation
|
|
from primaite.game.agent.observations.file_system_observations import FileObservation, FolderObservation
|
|
from primaite.game.agent.observations.firewall_observation import FirewallObservation
|
|
from primaite.game.agent.observations.host_observations import HostObservation
|
|
from primaite.game.agent.observations.link_observation import LinkObservation, LinksObservation
|
|
from primaite.game.agent.observations.nic_observations import NICObservation, PortObservation
|
|
from primaite.game.agent.observations.node_observations import NodesObservation
|
|
from primaite.game.agent.observations.observation_manager import NestedObservation, NullObservation, ObservationManager
|
|
from primaite.game.agent.observations.observations import AbstractObservation
|
|
from primaite.game.agent.observations.router_observation import RouterObservation
|
|
from primaite.game.agent.observations.software_observation import ApplicationObservation, ServiceObservation
|
|
|
|
# fmt: off
|
|
__all__ = [
|
|
"ACLObservation", "FileObservation", "FolderObservation", "FirewallObservation", "HostObservation",
|
|
"LinksObservation", "NICObservation", "PortObservation", "NodesObservation", "NestedObservation",
|
|
"ObservationManager", "ApplicationObservation", "ServiceObservation", "RouterObservation", "LinkObservation",]
|
|
# fmt: on
|