diff --git a/docs/_templates/custom-class-template.rst b/docs/_templates/custom-class-template.rst index 8a539bc9..b3f43787 100644 --- a/docs/_templates/custom-class-template.rst +++ b/docs/_templates/custom-class-template.rst @@ -1,3 +1,7 @@ +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + .. Credit to https://github.com/JamesALeedham/Sphinx-Autosummary-Recursion for the custom templates. .. diff --git a/docs/_templates/custom-module-template.rst b/docs/_templates/custom-module-template.rst index e6ecabd1..689d0d13 100644 --- a/docs/_templates/custom-module-template.rst +++ b/docs/_templates/custom-module-template.rst @@ -1,3 +1,7 @@ +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + .. Credit to https://github.com/JamesALeedham/Sphinx-Autosummary-Recursion for the custom templates. .. diff --git a/docs/api.rst b/docs/api.rst index df2bc193..d3db0a9c 100644 --- a/docs/api.rst +++ b/docs/api.rst @@ -1,3 +1,7 @@ +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + .. DO NOT DELETE THIS FILE! It contains the all-important `.. autosummary::` directive with `:recursive:` option, without which API documentation wouldn't get extracted from docstrings by the `sphinx.ext.autosummary` engine. It is hidden diff --git a/docs/conf.py b/docs/conf.py index 51b745cf..b14e5937 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Configuration file for the Sphinx documentation builder. # # For the full list of built-in configuration values, see the documentation: diff --git a/docs/index.rst b/docs/index.rst index cba573d6..5ba94976 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -1,3 +1,7 @@ +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + Welcome to PrimAITE's documentation ==================================== diff --git a/docs/source/about.rst b/docs/source/about.rst index a7135fc0..e237da41 100644 --- a/docs/source/about.rst +++ b/docs/source/about.rst @@ -1,4 +1,8 @@ -.. _about: +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + +.. _about: About PrimAITE ============== diff --git a/docs/source/config.rst b/docs/source/config.rst index af590a24..fa173772 100644 --- a/docs/source/config.rst +++ b/docs/source/config.rst @@ -1,3 +1,7 @@ +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + .. _config: The Config Files Explained diff --git a/docs/source/custom_agent.rst b/docs/source/custom_agent.rst index b4552d64..7d426856 100644 --- a/docs/source/custom_agent.rst +++ b/docs/source/custom_agent.rst @@ -1,4 +1,8 @@ -Custom Agents +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + +Custom Agents ============= diff --git a/docs/source/dependencies.rst b/docs/source/dependencies.rst index bbca3fce..fda95267 100644 --- a/docs/source/dependencies.rst +++ b/docs/source/dependencies.rst @@ -1,3 +1,7 @@ +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + .. role:: raw-html(raw) :format: html diff --git a/docs/source/getting_started.rst b/docs/source/getting_started.rst index e0254cdb..bb2b4bde 100644 --- a/docs/source/getting_started.rst +++ b/docs/source/getting_started.rst @@ -1,3 +1,7 @@ +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + .. _getting-started: Getting Started diff --git a/docs/source/glossary.rst b/docs/source/glossary.rst index 58b4cd5e..6748c415 100644 --- a/docs/source/glossary.rst +++ b/docs/source/glossary.rst @@ -1,3 +1,7 @@ +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + Glossary ============= diff --git a/docs/source/migration_1.2_-_2.0.rst b/docs/source/migration_1.2_-_2.0.rst index 2adf9656..072bdaa6 100644 --- a/docs/source/migration_1.2_-_2.0.rst +++ b/docs/source/migration_1.2_-_2.0.rst @@ -1,3 +1,7 @@ +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + v1.2 to v2.0 Migration guide ============================ diff --git a/docs/source/primaite_session.rst b/docs/source/primaite_session.rst index bfb66332..3569b29b 100644 --- a/docs/source/primaite_session.rst +++ b/docs/source/primaite_session.rst @@ -1,3 +1,7 @@ +.. only:: comment + + Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. + .. _run a primaite session: Run a PrimAITE Session @@ -44,7 +48,8 @@ For example, when running a session at 17:30:00 on 31st January 2023, the sessio ``~/primaite/sessions/2023-01-31/2023-01-31_17-30-00/``. Loading a session -------- +----------------- + A previous session can be loaded by providing the **directory** of the previous session to either the ``primaite session`` command from the cli (See :func:`primaite.cli.session`), or by calling :func:`primaite.main.run` with session_path. diff --git a/src/primaite/agents/__init__.py b/src/primaite/agents/__init__.py index 89580145..71f63d3a 100644 --- a/src/primaite/agents/__init__.py +++ b/src/primaite/agents/__init__.py @@ -1 +1,2 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. """Common interface between RL agents from different libraries and PrimAITE.""" diff --git a/src/primaite/agents/agent_abc.py b/src/primaite/agents/agent_abc.py index 515adfd0..fd9fbe9c 100644 --- a/src/primaite/agents/agent_abc.py +++ b/src/primaite/agents/agent_abc.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. from __future__ import annotations import json diff --git a/src/primaite/agents/hardcoded_abc.py b/src/primaite/agents/hardcoded_abc.py index cfee3e16..d900bc97 100644 --- a/src/primaite/agents/hardcoded_abc.py +++ b/src/primaite/agents/hardcoded_abc.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import time from abc import abstractmethod from pathlib import Path diff --git a/src/primaite/agents/hardcoded_acl.py b/src/primaite/agents/hardcoded_acl.py index e08a1d6d..4ed81693 100644 --- a/src/primaite/agents/hardcoded_acl.py +++ b/src/primaite/agents/hardcoded_acl.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. from typing import Any, Dict, List, Union import numpy as np diff --git a/src/primaite/agents/hardcoded_node.py b/src/primaite/agents/hardcoded_node.py index 113f622a..6857b251 100644 --- a/src/primaite/agents/hardcoded_node.py +++ b/src/primaite/agents/hardcoded_node.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import numpy as np from primaite.agents.hardcoded_abc import HardCodedAgentSessionABC diff --git a/src/primaite/agents/rllib.py b/src/primaite/agents/rllib.py index 1707cb81..4bc8e4af 100644 --- a/src/primaite/agents/rllib.py +++ b/src/primaite/agents/rllib.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. from __future__ import annotations import json diff --git a/src/primaite/agents/sb3.py b/src/primaite/agents/sb3.py index 862a0116..9bd895a4 100644 --- a/src/primaite/agents/sb3.py +++ b/src/primaite/agents/sb3.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. from __future__ import annotations import json diff --git a/src/primaite/agents/simple.py b/src/primaite/agents/simple.py index f81163ea..ec965a26 100644 --- a/src/primaite/agents/simple.py +++ b/src/primaite/agents/simple.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. from primaite.agents.hardcoded_abc import HardCodedAgentSessionABC from primaite.agents.utils import get_new_action, transform_action_acl_enum, transform_action_node_enum diff --git a/src/primaite/agents/utils.py b/src/primaite/agents/utils.py index 8858fa6a..85ba6f83 100644 --- a/src/primaite/agents/utils.py +++ b/src/primaite/agents/utils.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. from typing import Dict, List, Union import numpy as np diff --git a/src/primaite/common/custom_typing.py b/src/primaite/common/custom_typing.py index 37b10efe..6a6f1408 100644 --- a/src/primaite/common/custom_typing.py +++ b/src/primaite/common/custom_typing.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. from typing import Type, Union from primaite.nodes.active_node import ActiveNode diff --git a/src/primaite/config/__init__.py b/src/primaite/config/__init__.py index 03ed4cf1..5e9211be 100644 --- a/src/primaite/config/__init__.py +++ b/src/primaite/config/__init__.py @@ -1 +1,2 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. """Configuration parameters for running experiments.""" diff --git a/src/primaite/data_viz/__init__.py b/src/primaite/data_viz/__init__.py index db6ce6c8..7aa49525 100644 --- a/src/primaite/data_viz/__init__.py +++ b/src/primaite/data_viz/__init__.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. """Utility to generate plots of sessions metrics after PrimAITE.""" from enum import Enum diff --git a/src/primaite/data_viz/session_plots.py b/src/primaite/data_viz/session_plots.py index 245b9774..4d1984a8 100644 --- a/src/primaite/data_viz/session_plots.py +++ b/src/primaite/data_viz/session_plots.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. from pathlib import Path from typing import Dict, Optional, Union diff --git a/src/primaite/environment/observations.py b/src/primaite/environment/observations.py index 53c173fd..55446be9 100644 --- a/src/primaite/environment/observations.py +++ b/src/primaite/environment/observations.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. """Module for handling configurable observation spaces in PrimAITE.""" import logging from abc import ABC, abstractmethod diff --git a/src/primaite/notebooks/__init__.py b/src/primaite/notebooks/__init__.py index 6ca1d3f6..8cf1a0c5 100644 --- a/src/primaite/notebooks/__init__.py +++ b/src/primaite/notebooks/__init__.py @@ -1,5 +1,6 @@ -"""Contains default jupyter notebooks which demonstrate PrimAITE functionality.""" # Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. +"""Contains default jupyter notebooks which demonstrate PrimAITE functionality.""" + import importlib.util import os import subprocess diff --git a/src/primaite/pol/__init__.py b/src/primaite/pol/__init__.py index cba4b28b..c630d5d5 100644 --- a/src/primaite/pol/__init__.py +++ b/src/primaite/pol/__init__.py @@ -1,2 +1,2 @@ -"""Pattern of Life- Represents the actions of users on the network.""" # Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. +"""Pattern of Life- Represents the actions of users on the network.""" diff --git a/src/primaite/primaite_session.py b/src/primaite/primaite_session.py index 76134238..bc997c18 100644 --- a/src/primaite/primaite_session.py +++ b/src/primaite/primaite_session.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. """Main entry point to PrimAITE. Configure training/evaluation experiments and input/output.""" from __future__ import annotations diff --git a/src/primaite/setup/__init__.py b/src/primaite/setup/__init__.py index 3c0bfe14..68b78767 100644 --- a/src/primaite/setup/__init__.py +++ b/src/primaite/setup/__init__.py @@ -1,2 +1,2 @@ -"""Utilities to prepare the user's data folders.""" # Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. +"""Utilities to prepare the user's data folders.""" diff --git a/src/primaite/setup/reset_example_configs.py b/src/primaite/setup/reset_example_configs.py index 599de8dc..120bc0d8 100644 --- a/src/primaite/setup/reset_example_configs.py +++ b/src/primaite/setup/reset_example_configs.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import filecmp import os import shutil diff --git a/src/primaite/transactions/__init__.py b/src/primaite/transactions/__init__.py index 45315b22..c86c3b57 100644 --- a/src/primaite/transactions/__init__.py +++ b/src/primaite/transactions/__init__.py @@ -1,2 +1,2 @@ -"""Record data of the system's state and agent's observations and actions.""" # Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. +"""Record data of the system's state and agent's observations and actions.""" diff --git a/src/primaite/utils/__init__.py b/src/primaite/utils/__init__.py index 55e8a6ba..c56bbdf0 100644 --- a/src/primaite/utils/__init__.py +++ b/src/primaite/utils/__init__.py @@ -1 +1,2 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. """Utilities for PrimAITE.""" diff --git a/src/primaite/utils/session_metadata_parser.py b/src/primaite/utils/session_metadata_parser.py index 936d3269..2434a812 100644 --- a/src/primaite/utils/session_metadata_parser.py +++ b/src/primaite/utils/session_metadata_parser.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import json from pathlib import Path from typing import Union diff --git a/src/primaite/utils/session_output_reader.py b/src/primaite/utils/session_output_reader.py index 2ff4a16a..6dd685e6 100644 --- a/src/primaite/utils/session_output_reader.py +++ b/src/primaite/utils/session_output_reader.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. from pathlib import Path from typing import Any, Dict, Tuple, Union diff --git a/src/primaite/utils/session_output_writer.py b/src/primaite/utils/session_output_writer.py index 104acc62..ca152be7 100644 --- a/src/primaite/utils/session_output_writer.py +++ b/src/primaite/utils/session_output_writer.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import csv from logging import Logger from typing import Final, List, Tuple, TYPE_CHECKING, Union diff --git a/tests/config/legacy_conversion/legacy_training_config.yaml b/tests/config/legacy_conversion/legacy_training_config.yaml index 5c2025a2..e7e244de 100644 --- a/tests/config/legacy_conversion/legacy_training_config.yaml +++ b/tests/config/legacy_conversion/legacy_training_config.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Main Config File # Generic config values diff --git a/tests/config/legacy_conversion/new_training_config.yaml b/tests/config/legacy_conversion/new_training_config.yaml index c57741f7..2380dcb0 100644 --- a/tests/config/legacy_conversion/new_training_config.yaml +++ b/tests/config/legacy_conversion/new_training_config.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Main Config File # Generic config values diff --git a/tests/config/obs_tests/laydown.yaml b/tests/config/obs_tests/laydown.yaml index ef77ce83..25da9de3 100644 --- a/tests/config/obs_tests/laydown.yaml +++ b/tests/config/obs_tests/laydown.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. - item_type: PORTS ports_list: - port: '80' diff --git a/tests/config/obs_tests/main_config_LINK_TRAFFIC_LEVELS.yaml b/tests/config/obs_tests/main_config_LINK_TRAFFIC_LEVELS.yaml index 2ac8f59a..11904ddf 100644 --- a/tests/config/obs_tests/main_config_LINK_TRAFFIC_LEVELS.yaml +++ b/tests/config/obs_tests/main_config_LINK_TRAFFIC_LEVELS.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/obs_tests/main_config_NODE_LINK_TABLE.yaml b/tests/config/obs_tests/main_config_NODE_LINK_TABLE.yaml index a9986d5b..522686df 100644 --- a/tests/config/obs_tests/main_config_NODE_LINK_TABLE.yaml +++ b/tests/config/obs_tests/main_config_NODE_LINK_TABLE.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/obs_tests/main_config_NODE_STATUSES.yaml b/tests/config/obs_tests/main_config_NODE_STATUSES.yaml index a129712c..0521d1b3 100644 --- a/tests/config/obs_tests/main_config_NODE_STATUSES.yaml +++ b/tests/config/obs_tests/main_config_NODE_STATUSES.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/obs_tests/main_config_without_obs.yaml b/tests/config/obs_tests/main_config_without_obs.yaml index 03d11b82..88895bd3 100644 --- a/tests/config/obs_tests/main_config_without_obs.yaml +++ b/tests/config/obs_tests/main_config_without_obs.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/one_node_states_on_off_lay_down_config.yaml b/tests/config/one_node_states_on_off_lay_down_config.yaml index aadbd449..93538f0c 100644 --- a/tests/config/one_node_states_on_off_lay_down_config.yaml +++ b/tests/config/one_node_states_on_off_lay_down_config.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. - item_type: PORTS ports_list: - port: '21' diff --git a/tests/config/one_node_states_on_off_main_config.yaml b/tests/config/one_node_states_on_off_main_config.yaml index db7399aa..2cb025c0 100644 --- a/tests/config/one_node_states_on_off_main_config.yaml +++ b/tests/config/one_node_states_on_off_main_config.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/ppo_not_seeded_training_config.yaml b/tests/config/ppo_not_seeded_training_config.yaml index 14b3f087..9d8e6986 100644 --- a/tests/config/ppo_not_seeded_training_config.yaml +++ b/tests/config/ppo_not_seeded_training_config.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/ppo_seeded_training_config.yaml b/tests/config/ppo_seeded_training_config.yaml index a176c793..0160ef53 100644 --- a/tests/config/ppo_seeded_training_config.yaml +++ b/tests/config/ppo_seeded_training_config.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/single_action_space_fixed_blue_actions_main_config.yaml b/tests/config/single_action_space_fixed_blue_actions_main_config.yaml index 0f378634..56c89e8d 100644 --- a/tests/config/single_action_space_fixed_blue_actions_main_config.yaml +++ b/tests/config/single_action_space_fixed_blue_actions_main_config.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/single_action_space_lay_down_config.yaml b/tests/config/single_action_space_lay_down_config.yaml index 9d05b84a..7d604034 100644 --- a/tests/config/single_action_space_lay_down_config.yaml +++ b/tests/config/single_action_space_lay_down_config.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. - item_type: PORTS ports_list: - port: '21' diff --git a/tests/config/single_action_space_main_config.yaml b/tests/config/single_action_space_main_config.yaml index c875757f..88616823 100644 --- a/tests/config/single_action_space_main_config.yaml +++ b/tests/config/single_action_space_main_config.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/test_random_red_main_config.yaml b/tests/config/test_random_red_main_config.yaml index 9e034355..5f17c0f0 100644 --- a/tests/config/test_random_red_main_config.yaml +++ b/tests/config/test_random_red_main_config.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/train_episode_step.yaml b/tests/config/train_episode_step.yaml index f112b741..59494c3e 100644 --- a/tests/config/train_episode_step.yaml +++ b/tests/config/train_episode_step.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/training_config_main_rllib.yaml b/tests/config/training_config_main_rllib.yaml index 88f82890..a616d302 100644 --- a/tests/config/training_config_main_rllib.yaml +++ b/tests/config/training_config_main_rllib.yaml @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/mock_and_patch/__init__.py b/tests/mock_and_patch/__init__.py index e69de29b..63f825c2 100644 --- a/tests/mock_and_patch/__init__.py +++ b/tests/mock_and_patch/__init__.py @@ -0,0 +1 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. diff --git a/tests/mock_and_patch/get_session_path_mock.py b/tests/mock_and_patch/get_session_path_mock.py index 90c0cb5d..f8e77ec9 100644 --- a/tests/mock_and_patch/get_session_path_mock.py +++ b/tests/mock_and_patch/get_session_path_mock.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import tempfile from datetime import datetime from pathlib import Path diff --git a/tests/test_active_node.py b/tests/test_active_node.py index addc595c..7f8673e2 100644 --- a/tests/test_active_node.py +++ b/tests/test_active_node.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. """Used to test Active Node functions.""" import pytest diff --git a/tests/test_observation_space.py b/tests/test_observation_space.py index d5844fd9..15009188 100644 --- a/tests/test_observation_space.py +++ b/tests/test_observation_space.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. """Test env creation and behaviour with different observation spaces.""" import numpy as np import pytest diff --git a/tests/test_primaite_session.py b/tests/test_primaite_session.py index 75ea5882..27497e51 100644 --- a/tests/test_primaite_session.py +++ b/tests/test_primaite_session.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import os import pytest diff --git a/tests/test_red_random_agent_behaviour.py b/tests/test_red_random_agent_behaviour.py index f8885f3e..a552168e 100644 --- a/tests/test_red_random_agent_behaviour.py +++ b/tests/test_red_random_agent_behaviour.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import pytest from primaite.config.lay_down_config import data_manipulation_config_path diff --git a/tests/test_resetting_node.py b/tests/test_resetting_node.py index fb7dc83d..04ec6103 100644 --- a/tests/test_resetting_node.py +++ b/tests/test_resetting_node.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. """Used to test Active Node functions.""" import pytest diff --git a/tests/test_reward.py b/tests/test_reward.py index 2edfd44a..e9695985 100644 --- a/tests/test_reward.py +++ b/tests/test_reward.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import pytest from primaite import getLogger diff --git a/tests/test_rllib_agent.py b/tests/test_rllib_agent.py index 645214e3..1ebf3b61 100644 --- a/tests/test_rllib_agent.py +++ b/tests/test_rllib_agent.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import pytest from primaite import getLogger diff --git a/tests/test_seeding_and_deterministic_session.py b/tests/test_seeding_and_deterministic_session.py index f52e9eee..4431e4d8 100644 --- a/tests/test_seeding_and_deterministic_session.py +++ b/tests/test_seeding_and_deterministic_session.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import pytest as pytest from primaite.config.lay_down_config import dos_very_basic_config_path diff --git a/tests/test_service_node.py b/tests/test_service_node.py index 4383fc1b..faf694fb 100644 --- a/tests/test_service_node.py +++ b/tests/test_service_node.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. """Used to test Service Node functions.""" import pytest diff --git a/tests/test_session_loading.py b/tests/test_session_loading.py index 54cac351..72e72f25 100644 --- a/tests/test_session_loading.py +++ b/tests/test_session_loading.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import os.path import shutil import tempfile diff --git a/tests/test_single_action_space.py b/tests/test_single_action_space.py index bfcffd42..785f9d65 100644 --- a/tests/test_single_action_space.py +++ b/tests/test_single_action_space.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import time import pytest diff --git a/tests/test_train_eval_episode_steps.py b/tests/test_train_eval_episode_steps.py index b839e630..eb73516f 100644 --- a/tests/test_train_eval_episode_steps.py +++ b/tests/test_train_eval_episode_steps.py @@ -1,3 +1,4 @@ +# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence. import pytest from primaite import getLogger