From 0e0df1012fb20e0fe1b9f1963de2bbb74c03b0d7 Mon Sep 17 00:00:00 2001 From: Marek Wolan Date: Sun, 31 Mar 2024 23:39:24 +0100 Subject: [PATCH] #2417 update observations init to autoimport all obs types --- src/primaite/game/agent/observations/__init__.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/primaite/game/agent/observations/__init__.py b/src/primaite/game/agent/observations/__init__.py index b9d97ae6..15fdf7ed 100644 --- a/src/primaite/game/agent/observations/__init__.py +++ b/src/primaite/game/agent/observations/__init__.py @@ -1,4 +1,5 @@ # flake8: noqa +# Pre-import all the observations when we load up the observations module so that they can be resolved by the parser. 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 @@ -10,3 +11,10 @@ from primaite.game.agent.observations.observation_manager import NestedObservati 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",] +# fmt: on