merge main

.github/workflows/build.yml

@@ -16,7 +16,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
 
     steps:
     - uses: actions/checkout@v3
@@ -27,7 +27,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install pytest numpy nbval tabulate
+        python -m pip install pytest numpy nbval tabulate scipy sympy
         python -m pip install -e .
     - name: Test with pytest
       run: |

checkpoint_schedules/basic_schedules.py

@@ -1,3 +1,10 @@
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 The University of Edinburgh and Imperial College
+# London
+# Developed originally by James R. Maddison ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).
+
 """This module contains basic checkpointing schedules.
 """
 

checkpoint_schedules/hrevolve.py

@@ -1,5 +1,10 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 The University of Edinburgh and Imperial College
+# London
+# Developed by Daiane I. Dolci ([email protected]),
+# James R. Maddison ([email protected])
+# and David A. Ham ([email protected]).
+
 """This module contains the checkpointing schedules for the H-Revolve,
 Disk Revolve, Periodic Disk Revolve and Revolve algorithms.
 """

checkpoint_schedules/hrevolve_sequences/basic_functions.py

@@ -1,4 +1,10 @@
-#!/usr/bin/python
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 Inria and Imperial College London
+# Developed originally by Guillaume Pallez ([email protected]),
+# Julien Herrmann ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).
+
 """This module contains the basic functions used in the H-ReVolve algorithm."""
 import math
 

checkpoint_schedules/hrevolve_sequences/disk_revolve.py

@@ -1,4 +1,10 @@
-#!/usr/bin/python
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 Inria and Imperial College London
+# Developed originally by Guillaume Pallez ([email protected]),
+# Julien Herrmann ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).
+
 """This module contains the functions to compute the Disk-Revolve schedules.
 """
 from functools import partial

checkpoint_schedules/hrevolve_sequences/hrevolve.py

@@ -1,3 +1,10 @@
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 Inria and Imperial College London
+# Developed originally by Guillaume Pallez ([email protected]),
+# Julien Herrmann ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).
+
 """This module contains the implementation of the H-Revolve schedule.
 """
 from functools import partial

checkpoint_schedules/hrevolve_sequences/periodic_disk_revolve.py

@@ -1,3 +1,9 @@
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 Inria and Imperial College London
+# Developed originally by Guillaume Pallez ([email protected]),
+# Julien Herrmann ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).
 
 """This module contains the periodic disk revolve checkpoint schedule."""
 from functools import partial

checkpoint_schedules/hrevolve_sequences/revolve.py

@@ -1,4 +1,10 @@
-#!/usr/bin/python
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 Inria and Imperial College London
+# Developed originally by Guillaume Pallez ([email protected]),
+# Julien Herrmann ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).
+
 """This module contains the functions used to compute the revolver sequences.
 """
 from functools import partial

checkpoint_schedules/hrevolve_sequences/revolve_1d.py

@@ -1,4 +1,10 @@
-#!/usr/bin/python
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 Inria and Imperial College London
+# Developed originally by Guillaume Pallez ([email protected]),
+# Julien Herrmann ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).
+
 """This module contains the functions used to compute the 1D revolver
 sequences.
 """

checkpoint_schedules/hrevolve_sequences/utils.py

@@ -1,4 +1,9 @@
-#!/usr/bin/python
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 Inria and Imperial College London
+# Developed originally by Guillaume Pallez ([email protected]),
+# Julien Herrmann ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).
 
 def revolver_parameters(wd, rd, uf, ub):
     """Parameter use to obtain the revolver sequences.

checkpoint_schedules/mixed.py

@@ -1,3 +1,9 @@
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 The University of Edinburgh and Imperial College
+# London
+# Developed originally by James R. Maddison ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).
 import warnings
 import functools
 import numpy as np

checkpoint_schedules/multistage.py

@@ -1,3 +1,9 @@
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 The University of Edinburgh and Imperial College
+# London
+# Developed originally by James R. Maddison ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).import functools
 import functools
 from operator import itemgetter
 from .schedule import CheckpointSchedule, Forward, Reverse, Copy, Move, \

checkpoint_schedules/schedule.py

@@ -1,3 +1,10 @@
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 The University of Edinburgh and Imperial College
+# London
+# Developed by Daiane I. Dolci ([email protected]),
+# James R. Maddison ([email protected])
+# and David A. Ham ([email protected]).
+
 """Classes used to define checkpointing schedules and actions in checkpointing
 schedules.
 """

checkpoint_schedules/twolevel_binomial.py

@@ -1,3 +1,9 @@
+# SPDX-License-Identifier: LGPL-3.0-only
+# Copyright (C) 2020 - 2024 The University of Edinburgh and Imperial College
+# London
+# Developed originally by James R. Maddison ([email protected]).
+# Modified by Daiane I. Dolci ([email protected])
+# and David A. Ham ([email protected]).
 from .schedule import CheckpointSchedule, Forward, Reverse, Copy, Move, \
     EndForward, EndReverse, StorageType
 from .multistage import n_advance