feat: references

automl · Jun 4, 2024 · 6c74a3f · 6c74a3f
1 parent 7ce21f6
commit 6c74a3f
Show file tree

Hide file tree

Showing 5 changed files with 235 additions and 252 deletions.
diff --git a/arlbench/core/algorithms/buffers.py b/arlbench/core/algorithms/buffers.py
@@ -1,4 +1,5 @@
-# From https://github.com/instadeepai/flashbax/blob/main/flashbax/buffers/prioritised_trajectory_buffer.py
+# Contains code from flashbax (https://github.com/instadeepai/flashbax/blob/main/flashbax/buffers/prioritised_trajectory_buffer.py)
+# Licensed under the Apache License 2.0 (http://www.apache.org/licenses/LICENSE-2.0).
 from __future__ import annotations
 
 from typing import TYPE_CHECKING

diff --git a/arlbench/core/algorithms/dqn/dqn.py b/arlbench/core/algorithms/dqn/dqn.py
@@ -1,4 +1,5 @@
-# The DQN Code is heavily based on PureJax: https://github.com/luchris429/purejaxrl
+# Parts of this code are based on PureJaxRL (https://github.com/luchris429/purejaxrl).
+# Licensed under the Apache License 2.0 (http://www.apache.org/licenses/LICENSE-2.0).
 from __future__ import annotations
 
 import functools

diff --git a/arlbench/core/algorithms/ppo/ppo.py b/arlbench/core/algorithms/ppo/ppo.py
@@ -1,4 +1,5 @@
-# The PPO Code is heavily based on PureJax: https://github.com/luchris429/purejaxrl
+# Parts of this code are based on PureJaxRL (https://github.com/luchris429/purejaxrl).
+# Licensed under the Apache License 2.0 (http://www.apache.org/licenses/LICENSE-2.0).
 from __future__ import annotations
 
 import functools

diff --git a/arlbench/core/algorithms/prioritised_item_buffer.py b/arlbench/core/algorithms/prioritised_item_buffer.py
@@ -1,16 +1,5 @@
-# Copyright 2023 InstaDeep Ltd. All rights reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# Contains code from flashbax (https://github.com/instadeepai/flashbax/blob/main/flashbax/buffers/prioritised_trajectory_buffer.py)
+# Licensed under the Apache License 2.0 (http://www.apache.org/licenses/LICENSE-2.0).
 from __future__ import annotations
 
 from typing import TYPE_CHECKING