diff --git a/benchmark/config/benchmark_training_config.yaml b/benchmark/config/benchmark_training_config.yaml index 06186e26..71bb6ee1 100644 --- a/benchmark/config/benchmark_training_config.yaml +++ b/benchmark/config/benchmark_training_config.yaml @@ -62,7 +62,7 @@ observation_space: - name: LINK_TRAFFIC_LEVELS # Number of episodes for training to run per session -num_train_episodes: 500 +num_train_episodes: 10 # Number of time_steps for training per episode num_train_steps: 256 diff --git a/benchmark/primaite_benchmark.py b/benchmark/primaite_benchmark.py index 8c947435..2e8a7e51 100644 --- a/benchmark/primaite_benchmark.py +++ b/benchmark/primaite_benchmark.py @@ -393,11 +393,7 @@ def _plot_all_benchmarks_combined_session_av(): metadata_dict = json.load(file) df = _get_df_from_episode_av_reward_dict(metadata_dict["combined_av_reward_per_episode"]) - fig.add_trace( - go.Scatter( - x=df["episode"], y=df["rolling_av_reward"], mode="lines", name=dir.name, line={"color": "#FF0000"} - ) - ) + fig.add_trace(go.Scatter(x=df["episode"], y=df["rolling_av_reward"], mode="lines", name=dir.name)) # Set the layout of the graph fig.update_layout( @@ -417,7 +413,7 @@ def run(): """Run the PrimAITE benchmark.""" start_datetime = datetime.now() av_reward_per_episode_dicts = {} - for i in range(1, 11): + for i in range(1, 3): print(f"Starting Benchmark Session: {i}") with _get_benchmark_primaite_session() as session: session.learn() diff --git a/benchmark/results/PrimAITE Versions Learning Benchmark.png b/benchmark/results/PrimAITE Versions Learning Benchmark.png index b17c8e7c..fa01695a 100644 Binary files a/benchmark/results/PrimAITE Versions Learning Benchmark.png and b/benchmark/results/PrimAITE Versions Learning Benchmark.png differ diff --git a/docs/_templates/custom-class-template.rst b/docs/_templates/custom-class-template.rst index acffdc4c..66acd325 100644 --- a/docs/_templates/custom-class-template.rst +++ b/docs/_templates/custom-class-template.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK .. 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 8eebad3e..64ac520a 100644 --- a/docs/_templates/custom-module-template.rst +++ b/docs/_templates/custom-module-template.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK .. Credit to https://github.com/JamesALeedham/Sphinx-Autosummary-Recursion for the custom templates. diff --git a/docs/api.rst b/docs/api.rst index b24dafc3..aeaef4e2 100644 --- a/docs/api.rst +++ b/docs/api.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK .. DO NOT DELETE THIS FILE! It contains the all-important `.. autosummary::` directive with `:recursive:` option, without diff --git a/docs/conf.py b/docs/conf.py index b23bb57e..4a805916 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 de5bed46..208d5abc 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK Welcome to PrimAITE's documentation ==================================== diff --git a/docs/source/about.rst b/docs/source/about.rst index 2068472c..d12a59de 100644 --- a/docs/source/about.rst +++ b/docs/source/about.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK .. _about: diff --git a/docs/source/config.rst b/docs/source/config.rst index 23860d99..daf7f90b 100644 --- a/docs/source/config.rst +++ b/docs/source/config.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK .. _config: diff --git a/docs/source/custom_agent.rst b/docs/source/custom_agent.rst index aa4e2082..0f4f30ad 100644 --- a/docs/source/custom_agent.rst +++ b/docs/source/custom_agent.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK Custom Agents ============= diff --git a/docs/source/dependencies.rst b/docs/source/dependencies.rst index 0d3f21c3..942ccfd8 100644 --- a/docs/source/dependencies.rst +++ b/docs/source/dependencies.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK .. role:: raw-html(raw) :format: html diff --git a/docs/source/getting_started.rst b/docs/source/getting_started.rst index ef34a163..0ac2fdd4 100644 --- a/docs/source/getting_started.rst +++ b/docs/source/getting_started.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK .. _getting-started: diff --git a/docs/source/glossary.rst b/docs/source/glossary.rst index e8972d05..8340d559 100644 --- a/docs/source/glossary.rst +++ b/docs/source/glossary.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK Glossary ============= diff --git a/docs/source/migration_1.2_-_2.0.rst b/docs/source/migration_1.2_-_2.0.rst index 6369bb96..e1e24790 100644 --- a/docs/source/migration_1.2_-_2.0.rst +++ b/docs/source/migration_1.2_-_2.0.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK v1.2 to v2.0 Migration guide ============================ diff --git a/docs/source/primaite_session.rst b/docs/source/primaite_session.rst index 1b75699b..840e5717 100644 --- a/docs/source/primaite_session.rst +++ b/docs/source/primaite_session.rst @@ -1,6 +1,6 @@ .. only:: comment - Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. + © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK .. _run a primaite session: diff --git a/setup.py b/setup.py index efaf24bf..62bcbf16 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from setuptools import setup from wheel.bdist_wheel import bdist_wheel as _bdist_wheel # noqa diff --git a/src/primaite/__init__.py b/src/primaite/__init__.py index 6f9004b0..a0f5b7fe 100644 --- a/src/primaite/__init__.py +++ b/src/primaite/__init__.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import logging import logging.config import shutil diff --git a/src/primaite/acl/__init__.py b/src/primaite/acl/__init__.py index c6fd79f2..6dc02583 100644 --- a/src/primaite/acl/__init__.py +++ b/src/primaite/acl/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Access Control List. Models firewall functionality.""" diff --git a/src/primaite/acl/access_control_list.py b/src/primaite/acl/access_control_list.py index c61b0c10..88943f8f 100644 --- a/src/primaite/acl/access_control_list.py +++ b/src/primaite/acl/access_control_list.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """A class that implements the access control list implementation for the network.""" import logging from typing import Dict, Final, List, Union diff --git a/src/primaite/acl/acl_rule.py b/src/primaite/acl/acl_rule.py index 53c860cd..9c8deacd 100644 --- a/src/primaite/acl/acl_rule.py +++ b/src/primaite/acl/acl_rule.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """A class that implements an access control list rule.""" from primaite.common.enums import RulePermissionType diff --git a/src/primaite/agents/__init__.py b/src/primaite/agents/__init__.py index d987b43f..c742daf3 100644 --- a/src/primaite/agents/__init__.py +++ b/src/primaite/agents/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """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 3fd53869..48650139 100644 --- a/src/primaite/agents/agent_abc.py +++ b/src/primaite/agents/agent_abc.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from __future__ import annotations import json diff --git a/src/primaite/agents/hardcoded_abc.py b/src/primaite/agents/hardcoded_abc.py index 0336f00e..e75edbc5 100644 --- a/src/primaite/agents/hardcoded_abc.py +++ b/src/primaite/agents/hardcoded_abc.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK 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 b8c49c14..2440da06 100644 --- a/src/primaite/agents/hardcoded_acl.py +++ b/src/primaite/agents/hardcoded_acl.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from typing import Dict, List, Union import numpy as np diff --git a/src/primaite/agents/hardcoded_node.py b/src/primaite/agents/hardcoded_node.py index 10cc2b72..b08d8967 100644 --- a/src/primaite/agents/hardcoded_node.py +++ b/src/primaite/agents/hardcoded_node.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK 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 fb062f54..ab1b3af3 100644 --- a/src/primaite/agents/rllib.py +++ b/src/primaite/agents/rllib.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from __future__ import annotations import json diff --git a/src/primaite/agents/sb3.py b/src/primaite/agents/sb3.py index b347d44f..783f57eb 100644 --- a/src/primaite/agents/sb3.py +++ b/src/primaite/agents/sb3.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from __future__ import annotations import json diff --git a/src/primaite/agents/simple.py b/src/primaite/agents/simple.py index 18ffa72b..bfdff869 100644 --- a/src/primaite/agents/simple.py +++ b/src/primaite/agents/simple.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import numpy as np diff --git a/src/primaite/agents/utils.py b/src/primaite/agents/utils.py index ff0ca8d2..08d46294 100644 --- a/src/primaite/agents/utils.py +++ b/src/primaite/agents/utils.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from typing import Dict, List, Union import numpy as np diff --git a/src/primaite/cli.py b/src/primaite/cli.py index 2ab3ed2e..bd0c8446 100644 --- a/src/primaite/cli.py +++ b/src/primaite/cli.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Provides a CLI using Typer as an entry point.""" import logging import os diff --git a/src/primaite/common/__init__.py b/src/primaite/common/__init__.py index 738a30d1..5770bcbc 100644 --- a/src/primaite/common/__init__.py +++ b/src/primaite/common/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Objects which are shared between many PrimAITE modules.""" diff --git a/src/primaite/common/enums.py b/src/primaite/common/enums.py index d74ec795..006301f1 100644 --- a/src/primaite/common/enums.py +++ b/src/primaite/common/enums.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Enumerations for APE.""" from enum import Enum, IntEnum diff --git a/src/primaite/common/protocol.py b/src/primaite/common/protocol.py index 048ed0ab..6940ba3f 100644 --- a/src/primaite/common/protocol.py +++ b/src/primaite/common/protocol.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """The protocol class.""" diff --git a/src/primaite/common/service.py b/src/primaite/common/service.py index 7ee694db..956815e8 100644 --- a/src/primaite/common/service.py +++ b/src/primaite/common/service.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """The Service class.""" from primaite.common.enums import SoftwareState diff --git a/src/primaite/config/__init__.py b/src/primaite/config/__init__.py index 9bd899f7..92f5a7d2 100644 --- a/src/primaite/config/__init__.py +++ b/src/primaite/config/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Configuration parameters for running experiments.""" diff --git a/src/primaite/config/lay_down_config.py b/src/primaite/config/lay_down_config.py index aef6790f..65ca7e91 100644 --- a/src/primaite/config/lay_down_config.py +++ b/src/primaite/config/lay_down_config.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from logging import Logger from pathlib import Path from typing import Any, Dict, Final, Union diff --git a/src/primaite/config/training_config.py b/src/primaite/config/training_config.py index b68a4f7b..ebfee09a 100644 --- a/src/primaite/config/training_config.py +++ b/src/primaite/config/training_config.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from __future__ import annotations from dataclasses import dataclass, field diff --git a/src/primaite/data_viz/__init__.py b/src/primaite/data_viz/__init__.py index ad43c141..260579da 100644 --- a/src/primaite/data_viz/__init__.py +++ b/src/primaite/data_viz/__init__.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """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 f66fe99d..37750353 100644 --- a/src/primaite/data_viz/session_plots.py +++ b/src/primaite/data_viz/session_plots.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from pathlib import Path from typing import Dict, Optional, Union diff --git a/src/primaite/environment/__init__.py b/src/primaite/environment/__init__.py index e837fe1e..f0fd21b9 100644 --- a/src/primaite/environment/__init__.py +++ b/src/primaite/environment/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Gym/Gymnasium environment for RL agents consisting of a simulated computer network.""" diff --git a/src/primaite/environment/observations.py b/src/primaite/environment/observations.py index a0423b89..383a9b5a 100644 --- a/src/primaite/environment/observations.py +++ b/src/primaite/environment/observations.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Module for handling configurable observation spaces in PrimAITE.""" import logging from abc import ABC, abstractmethod diff --git a/src/primaite/environment/primaite_env.py b/src/primaite/environment/primaite_env.py index 6a145498..cde586ed 100644 --- a/src/primaite/environment/primaite_env.py +++ b/src/primaite/environment/primaite_env.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Main environment module containing the PRIMmary AI Training Evironment (Primaite) class.""" import copy import logging diff --git a/src/primaite/environment/reward.py b/src/primaite/environment/reward.py index 92ef89ec..aa9dc97d 100644 --- a/src/primaite/environment/reward.py +++ b/src/primaite/environment/reward.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Implements reward function.""" from logging import Logger from typing import Dict, TYPE_CHECKING, Union diff --git a/src/primaite/exceptions.py b/src/primaite/exceptions.py index 7ccfadd9..3b4058ac 100644 --- a/src/primaite/exceptions.py +++ b/src/primaite/exceptions.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK class PrimaiteError(Exception): """The root PrimAITe Error.""" diff --git a/src/primaite/links/__init__.py b/src/primaite/links/__init__.py index 21ce44ba..c91b6951 100644 --- a/src/primaite/links/__init__.py +++ b/src/primaite/links/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Network connections between nodes in the simulation.""" diff --git a/src/primaite/links/link.py b/src/primaite/links/link.py index aa3fa7fb..3830a15b 100644 --- a/src/primaite/links/link.py +++ b/src/primaite/links/link.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """The link class.""" from typing import List diff --git a/src/primaite/main.py b/src/primaite/main.py index aed39d73..03f4fb35 100644 --- a/src/primaite/main.py +++ b/src/primaite/main.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """The main PrimAITE session runner module.""" import argparse from pathlib import Path diff --git a/src/primaite/nodes/__init__.py b/src/primaite/nodes/__init__.py index 43b213d6..231b8d92 100644 --- a/src/primaite/nodes/__init__.py +++ b/src/primaite/nodes/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Nodes represent network hosts in the simulation.""" diff --git a/src/primaite/nodes/active_node.py b/src/primaite/nodes/active_node.py index b5df70b5..8f472e86 100644 --- a/src/primaite/nodes/active_node.py +++ b/src/primaite/nodes/active_node.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """An Active Node (i.e. not an actuator).""" import logging from typing import Final diff --git a/src/primaite/nodes/node.py b/src/primaite/nodes/node.py index 9118fa46..fc4d41d3 100644 --- a/src/primaite/nodes/node.py +++ b/src/primaite/nodes/node.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """The base Node class.""" from typing import Final diff --git a/src/primaite/nodes/node_state_instruction_green.py b/src/primaite/nodes/node_state_instruction_green.py index 8e03b40f..6e35d0ec 100644 --- a/src/primaite/nodes/node_state_instruction_green.py +++ b/src/primaite/nodes/node_state_instruction_green.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Defines node behaviour for Green PoL.""" from typing import TYPE_CHECKING, Union diff --git a/src/primaite/nodes/node_state_instruction_red.py b/src/primaite/nodes/node_state_instruction_red.py index fd9a93a5..eb87924b 100644 --- a/src/primaite/nodes/node_state_instruction_red.py +++ b/src/primaite/nodes/node_state_instruction_red.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Defines node behaviour for Green PoL.""" from typing import TYPE_CHECKING, Union diff --git a/src/primaite/nodes/passive_node.py b/src/primaite/nodes/passive_node.py index 88c8cc85..08dcbfa2 100644 --- a/src/primaite/nodes/passive_node.py +++ b/src/primaite/nodes/passive_node.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """The Passive Node class (i.e. an actuator).""" from primaite.common.enums import HardwareState, NodeType, Priority from primaite.config.training_config import TrainingConfig diff --git a/src/primaite/nodes/service_node.py b/src/primaite/nodes/service_node.py index ce1ffe92..b0d42785 100644 --- a/src/primaite/nodes/service_node.py +++ b/src/primaite/nodes/service_node.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """A Service Node (i.e. not an actuator).""" import logging from typing import Dict, Final diff --git a/src/primaite/notebooks/__init__.py b/src/primaite/notebooks/__init__.py index 1599d28d..bc1dcfcd 100644 --- a/src/primaite/notebooks/__init__.py +++ b/src/primaite/notebooks/__init__.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Contains default jupyter notebooks which demonstrate PrimAITE functionality.""" import importlib.util diff --git a/src/primaite/pol/__init__.py b/src/primaite/pol/__init__.py index 1adb1491..d0d9f616 100644 --- a/src/primaite/pol/__init__.py +++ b/src/primaite/pol/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Pattern of Life- Represents the actions of users on the network.""" diff --git a/src/primaite/pol/green_pol.py b/src/primaite/pol/green_pol.py index 0425a831..814aa314 100644 --- a/src/primaite/pol/green_pol.py +++ b/src/primaite/pol/green_pol.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Implements Pattern of Life on the network (nodes and links).""" from typing import Dict diff --git a/src/primaite/pol/ier.py b/src/primaite/pol/ier.py index 7fab340d..b8da28c0 100644 --- a/src/primaite/pol/ier.py +++ b/src/primaite/pol/ier.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """ Information Exchange Requirements for APE. diff --git a/src/primaite/pol/red_agent_pol.py b/src/primaite/pol/red_agent_pol.py index 89aa3303..ca1a58da 100644 --- a/src/primaite/pol/red_agent_pol.py +++ b/src/primaite/pol/red_agent_pol.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Implements POL on the network (nodes and links) resulting from the red agent attack.""" from typing import Dict diff --git a/src/primaite/primaite_session.py b/src/primaite/primaite_session.py index 9203122a..c64b51fb 100644 --- a/src/primaite/primaite_session.py +++ b/src/primaite/primaite_session.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """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 acfa48c4..12b77f1e 100644 --- a/src/primaite/setup/__init__.py +++ b/src/primaite/setup/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Utilities to prepare the user's data folders.""" diff --git a/src/primaite/setup/old_installation_clean_up.py b/src/primaite/setup/old_installation_clean_up.py index d23abf3c..412aed60 100644 --- a/src/primaite/setup/old_installation_clean_up.py +++ b/src/primaite/setup/old_installation_clean_up.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from primaite import getLogger diff --git a/src/primaite/setup/reset_demo_notebooks.py b/src/primaite/setup/reset_demo_notebooks.py index 1bc217f8..1f96c90f 100644 --- a/src/primaite/setup/reset_demo_notebooks.py +++ b/src/primaite/setup/reset_demo_notebooks.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import filecmp import os import shutil diff --git a/src/primaite/setup/reset_example_configs.py b/src/primaite/setup/reset_example_configs.py index e53d04e2..41345853 100644 --- a/src/primaite/setup/reset_example_configs.py +++ b/src/primaite/setup/reset_example_configs.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import filecmp import os import shutil diff --git a/src/primaite/transactions/__init__.py b/src/primaite/transactions/__init__.py index 9a881fd5..505c5080 100644 --- a/src/primaite/transactions/__init__.py +++ b/src/primaite/transactions/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Record data of the system's state and agent's observations and actions.""" diff --git a/src/primaite/transactions/transaction.py b/src/primaite/transactions/transaction.py index 1a702748..7d5f747c 100644 --- a/src/primaite/transactions/transaction.py +++ b/src/primaite/transactions/transaction.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """The Transaction class.""" from datetime import datetime from typing import List, Optional, Tuple, TYPE_CHECKING, Union diff --git a/src/primaite/utils/__init__.py b/src/primaite/utils/__init__.py index 5dbd1e5f..4f9deb13 100644 --- a/src/primaite/utils/__init__.py +++ b/src/primaite/utils/__init__.py @@ -1,2 +1,2 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Utilities for PrimAITE.""" diff --git a/src/primaite/utils/package_data.py b/src/primaite/utils/package_data.py index 96157b40..ac41e8bc 100644 --- a/src/primaite/utils/package_data.py +++ b/src/primaite/utils/package_data.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import os from logging import Logger from pathlib import Path diff --git a/src/primaite/utils/session_metadata_parser.py b/src/primaite/utils/session_metadata_parser.py index 0b0eaaec..2548a8b6 100644 --- a/src/primaite/utils/session_metadata_parser.py +++ b/src/primaite/utils/session_metadata_parser.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import json from pathlib import Path from typing import Any, Dict, Union diff --git a/src/primaite/utils/session_output_reader.py b/src/primaite/utils/session_output_reader.py index db761a34..30febff1 100644 --- a/src/primaite/utils/session_output_reader.py +++ b/src/primaite/utils/session_output_reader.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK 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 e7f1b248..0eb18038 100644 --- a/src/primaite/utils/session_output_writer.py +++ b/src/primaite/utils/session_output_writer.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import csv from logging import Logger from typing import Final, List, Tuple, TYPE_CHECKING, Union diff --git a/tests/__init__.py b/tests/__init__.py index f8e6fc55..5a06b646 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK from pathlib import Path from typing import Final diff --git a/tests/config/legacy_conversion/legacy_training_config.yaml b/tests/config/legacy_conversion/legacy_training_config.yaml index fb24e3d7..3477e6e0 100644 --- a/tests/config/legacy_conversion/legacy_training_config.yaml +++ b/tests/config/legacy_conversion/legacy_training_config.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 3df29d04..1ec36e97 100644 --- a/tests/config/legacy_conversion/new_training_config.yaml +++ b/tests/config/legacy_conversion/new_training_config.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # Main Config File # Generic config values diff --git a/tests/config/obs_tests/laydown.yaml b/tests/config/obs_tests/laydown.yaml index 4ab44755..e358d0d2 100644 --- a/tests/config/obs_tests/laydown.yaml +++ b/tests/config/obs_tests/laydown.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK - 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 689d6bb4..805ab31e 100644 --- a/tests/config/obs_tests/main_config_LINK_TRAFFIC_LEVELS.yaml +++ b/tests/config/obs_tests/main_config_LINK_TRAFFIC_LEVELS.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 885f7e79..535558aa 100644 --- a/tests/config/obs_tests/main_config_NODE_LINK_TABLE.yaml +++ b/tests/config/obs_tests/main_config_NODE_LINK_TABLE.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 c662e715..d1319c35 100644 --- a/tests/config/obs_tests/main_config_NODE_STATUSES.yaml +++ b/tests/config/obs_tests/main_config_NODE_STATUSES.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 a2af9763..26457c84 100644 --- a/tests/config/obs_tests/main_config_without_obs.yaml +++ b/tests/config/obs_tests/main_config_without_obs.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 65257d62..0f572d8d 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,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK - 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 dbe4256f..10af7a1f 100644 --- a/tests/config/one_node_states_on_off_main_config.yaml +++ b/tests/config/one_node_states_on_off_main_config.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 2160a3a3..fac2fe95 100644 --- a/tests/config/ppo_not_seeded_training_config.yaml +++ b/tests/config/ppo_not_seeded_training_config.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 7512dc85..e4d4fe5b 100644 --- a/tests/config/ppo_seeded_training_config.yaml +++ b/tests/config/ppo_seeded_training_config.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/session_test/training_config_main_rllib.yaml b/tests/config/session_test/training_config_main_rllib.yaml index 118b2d4e..374c6ac5 100644 --- a/tests/config/session_test/training_config_main_rllib.yaml +++ b/tests/config/session_test/training_config_main_rllib.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/config/session_test/training_config_main_sb3.yaml b/tests/config/session_test/training_config_main_sb3.yaml index 9065bf8a..733105ea 100644 --- a/tests/config/session_test/training_config_main_sb3.yaml +++ b/tests/config/session_test/training_config_main_sb3.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 644d5912..6210cf3e 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,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 866eebe8..9103e2b7 100644 --- a/tests/config/single_action_space_lay_down_config.yaml +++ b/tests/config/single_action_space_lay_down_config.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK - item_type: PORTS ports_list: - port: '80' diff --git a/tests/config/single_action_space_main_config.yaml b/tests/config/single_action_space_main_config.yaml index deaad090..67eaf49d 100644 --- a/tests/config/single_action_space_main_config.yaml +++ b/tests/config/single_action_space_main_config.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 3416029c..310c9dc6 100644 --- a/tests/config/test_random_red_main_config.yaml +++ b/tests/config/test_random_red_main_config.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # 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 31337b0c..a86e0f62 100644 --- a/tests/config/train_episode_step.yaml +++ b/tests/config/train_episode_step.yaml @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK # Training Config File # Sets which agent algorithm framework will be used. diff --git a/tests/conftest.py b/tests/conftest.py index 9e448574..f40b0b94 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import datetime import shutil import tempfile diff --git a/tests/mock_and_patch/__init__.py b/tests/mock_and_patch/__init__.py index 778748f7..e0315ce3 100644 --- a/tests/mock_and_patch/__init__.py +++ b/tests/mock_and_patch/__init__.py @@ -1 +1 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK diff --git a/tests/mock_and_patch/get_session_path_mock.py b/tests/mock_and_patch/get_session_path_mock.py index 190e1dba..16c4a274 100644 --- a/tests/mock_and_patch/get_session_path_mock.py +++ b/tests/mock_and_patch/get_session_path_mock.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import tempfile from datetime import datetime from pathlib import Path diff --git a/tests/test_acl.py b/tests/test_acl.py index 3491aab8..d8357cf6 100644 --- a/tests/test_acl.py +++ b/tests/test_acl.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Used to tes the ACL functions.""" from primaite.acl.access_control_list import AccessControlList diff --git a/tests/test_active_node.py b/tests/test_active_node.py index 880c0f02..44d38313 100644 --- a/tests/test_active_node.py +++ b/tests/test_active_node.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Used to test Active Node functions.""" import pytest diff --git a/tests/test_observation_space.py b/tests/test_observation_space.py index c4a9789c..ff3528e1 100644 --- a/tests/test_observation_space.py +++ b/tests/test_observation_space.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Test env creation and behaviour with different observation spaces.""" import numpy as np diff --git a/tests/test_primaite_session.py b/tests/test_primaite_session.py index 4b7b91ac..b76a2ecf 100644 --- a/tests/test_primaite_session.py +++ b/tests/test_primaite_session.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import os import pytest diff --git a/tests/test_red_random_agent_behaviour.py b/tests/test_red_random_agent_behaviour.py index 3496ed9d..e99f4adb 100644 --- a/tests/test_red_random_agent_behaviour.py +++ b/tests/test_red_random_agent_behaviour.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK 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 80e13c5b..d4e27c17 100644 --- a/tests/test_resetting_node.py +++ b/tests/test_resetting_node.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Used to test Active Node functions.""" import pytest diff --git a/tests/test_reward.py b/tests/test_reward.py index 741c6f13..2ac66af1 100644 --- a/tests/test_reward.py +++ b/tests/test_reward.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK 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 c4b47d5f..70ce229e 100644 --- a/tests/test_seeding_and_deterministic_session.py +++ b/tests/test_seeding_and_deterministic_session.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK 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 2f504cd6..906bcf55 100644 --- a/tests/test_service_node.py +++ b/tests/test_service_node.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK """Used to test Service Node functions.""" import pytest diff --git a/tests/test_session_loading.py b/tests/test_session_loading.py index 714b363f..f9990f76 100644 --- a/tests/test_session_loading.py +++ b/tests/test_session_loading.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import os.path import shutil import tempfile diff --git a/tests/test_single_action_space.py b/tests/test_single_action_space.py index b91bc2bf..5d300232 100644 --- a/tests/test_single_action_space.py +++ b/tests/test_single_action_space.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import time import pytest diff --git a/tests/test_train_eval_episode_steps.py b/tests/test_train_eval_episode_steps.py index 4f7bed16..1b53fe9d 100644 --- a/tests/test_train_eval_episode_steps.py +++ b/tests/test_train_eval_episode_steps.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import pytest from primaite import getLogger diff --git a/tests/test_training_config.py b/tests/test_training_config.py index 4123ee39..58f9c797 100644 --- a/tests/test_training_config.py +++ b/tests/test_training_config.py @@ -1,4 +1,4 @@ -# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. +# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK import yaml from primaite.config import training_config