diff --git a/clu/__init__.py b/clu/__init__.py index 6390a47..d612d74 100644 --- a/clu/__init__.py +++ b/clu/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/asynclib.py b/clu/asynclib.py index 23c43c4..2d27809 100644 --- a/clu/asynclib.py +++ b/clu/asynclib.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/asynclib_test.py b/clu/asynclib_test.py index e8e9916..5b9cef5 100644 --- a/clu/asynclib_test.py +++ b/clu/asynclib_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/checkpoint.py b/clu/checkpoint.py index 00a04ce..4a34158 100644 --- a/clu/checkpoint.py +++ b/clu/checkpoint.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/checkpoint_test.py b/clu/checkpoint_test.py index 7bb9aeb..364f43e 100644 --- a/clu/checkpoint_test.py +++ b/clu/checkpoint_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/data/__init__.py b/clu/data/__init__.py index 041ba2c..30539c5 100644 --- a/clu/data/__init__.py +++ b/clu/data/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/data/dataset_iterator.py b/clu/data/dataset_iterator.py index 0e92aad..21d90dd 100644 --- a/clu/data/dataset_iterator.py +++ b/clu/data/dataset_iterator.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/data/dataset_iterator_test.py b/clu/data/dataset_iterator_test.py index 801c07e..92c5fde 100644 --- a/clu/data/dataset_iterator_test.py +++ b/clu/data/dataset_iterator_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/deterministic_data.py b/clu/deterministic_data.py index dee672c..15ae2d8 100644 --- a/clu/deterministic_data.py +++ b/clu/deterministic_data.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/deterministic_data_test.py b/clu/deterministic_data_test.py index 977e7f3..e506d9b 100644 --- a/clu/deterministic_data_test.py +++ b/clu/deterministic_data_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/internal/__init__.py b/clu/internal/__init__.py index 6390a47..d612d74 100644 --- a/clu/internal/__init__.py +++ b/clu/internal/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/internal/utils.py b/clu/internal/utils.py index b45cccf..8255937 100644 --- a/clu/internal/utils.py +++ b/clu/internal/utils.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/internal/utils_test.py b/clu/internal/utils_test.py index 9d6d1c4..2977114 100644 --- a/clu/internal/utils_test.py +++ b/clu/internal/utils_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/__init__.py b/clu/metric_writers/__init__.py index affa825..356e639 100644 --- a/clu/metric_writers/__init__.py +++ b/clu/metric_writers/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/async_writer.py b/clu/metric_writers/async_writer.py index 184aced..8710660 100644 --- a/clu/metric_writers/async_writer.py +++ b/clu/metric_writers/async_writer.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/async_writer_test.py b/clu/metric_writers/async_writer_test.py index d141a1e..eadd737 100644 --- a/clu/metric_writers/async_writer_test.py +++ b/clu/metric_writers/async_writer_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/interface.py b/clu/metric_writers/interface.py index b806b9a..82276d5 100644 --- a/clu/metric_writers/interface.py +++ b/clu/metric_writers/interface.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/logging_writer.py b/clu/metric_writers/logging_writer.py index 4daf53f..f64f745 100644 --- a/clu/metric_writers/logging_writer.py +++ b/clu/metric_writers/logging_writer.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/logging_writer_test.py b/clu/metric_writers/logging_writer_test.py index f5bc5f3..d58e9f0 100644 --- a/clu/metric_writers/logging_writer_test.py +++ b/clu/metric_writers/logging_writer_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/multi_writer.py b/clu/metric_writers/multi_writer.py index 96e9ad4..caf3e04 100644 --- a/clu/metric_writers/multi_writer.py +++ b/clu/metric_writers/multi_writer.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/multi_writer_test.py b/clu/metric_writers/multi_writer_test.py index 22f18a2..ca5d871 100644 --- a/clu/metric_writers/multi_writer_test.py +++ b/clu/metric_writers/multi_writer_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/summary_writer.py b/clu/metric_writers/summary_writer.py index f0ab778..72e5ea8 100644 --- a/clu/metric_writers/summary_writer.py +++ b/clu/metric_writers/summary_writer.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/summary_writer_test.py b/clu/metric_writers/summary_writer_test.py index f6cdf88..11b88b1 100644 --- a/clu/metric_writers/summary_writer_test.py +++ b/clu/metric_writers/summary_writer_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/torch_tensorboard_writer.py b/clu/metric_writers/torch_tensorboard_writer.py index 74c02c5..39e5126 100644 --- a/clu/metric_writers/torch_tensorboard_writer.py +++ b/clu/metric_writers/torch_tensorboard_writer.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/torch_tensorboard_writer_test.py b/clu/metric_writers/torch_tensorboard_writer_test.py index b433ae4..d1f7b2d 100644 --- a/clu/metric_writers/torch_tensorboard_writer_test.py +++ b/clu/metric_writers/torch_tensorboard_writer_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/utils.py b/clu/metric_writers/utils.py index 24686f6..ac469ce 100644 --- a/clu/metric_writers/utils.py +++ b/clu/metric_writers/utils.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metric_writers/utils_test.py b/clu/metric_writers/utils_test.py index fb773a6..53e1831 100644 --- a/clu/metric_writers/utils_test.py +++ b/clu/metric_writers/utils_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/metrics.py b/clu/metrics.py index 5e7deda..e024f23 100644 --- a/clu/metrics.py +++ b/clu/metrics.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -471,7 +471,7 @@ class MyMetrics(metrics.Collection): A subclass of Collection with fields defined by provided `metrics`. """ return flax.struct.dataclass( - type("_InlineCollection", (Collection,), {"__annotations__": metrics})) + type("_InlineCollection", (cls,), {"__annotations__": metrics})) @classmethod def create_collection(cls, **metrics: Metric) -> "Collection": diff --git a/clu/metrics_test.py b/clu/metrics_test.py index 4afcb0f..11ae597 100644 --- a/clu/metrics_test.py +++ b/clu/metrics_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/parameter_overview.py b/clu/parameter_overview.py index 3a4aa3f..8dd971f 100644 --- a/clu/parameter_overview.py +++ b/clu/parameter_overview.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/parameter_overview_test.py b/clu/parameter_overview_test.py index 9352f4d..5bbc7e5 100644 --- a/clu/parameter_overview_test.py +++ b/clu/parameter_overview_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/periodic_actions.py b/clu/periodic_actions.py index f6d7964..c41b526 100644 --- a/clu/periodic_actions.py +++ b/clu/periodic_actions.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/periodic_actions_test.py b/clu/periodic_actions_test.py index aa54611..3c79301 100644 --- a/clu/periodic_actions_test.py +++ b/clu/periodic_actions_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/platform/__init__.py b/clu/platform/__init__.py index 6cd3ed5..a372d6a 100644 --- a/clu/platform/__init__.py +++ b/clu/platform/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/platform/interface.py b/clu/platform/interface.py index f5e6355..a87893c 100644 --- a/clu/platform/interface.py +++ b/clu/platform/interface.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/platform/local.py b/clu/platform/local.py index b86131f..769c205 100644 --- a/clu/platform/local.py +++ b/clu/platform/local.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/preprocess_spec.py b/clu/preprocess_spec.py index 5f4968a..3322224 100644 --- a/clu/preprocess_spec.py +++ b/clu/preprocess_spec.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/preprocess_spec_test.py b/clu/preprocess_spec_test.py index 8b6c09f..9a0cb04 100644 --- a/clu/preprocess_spec_test.py +++ b/clu/preprocess_spec_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/profiler.py b/clu/profiler.py index 4153622..5718dae 100644 --- a/clu/profiler.py +++ b/clu/profiler.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/clu/values.py b/clu/values.py index 2705b3f..ca9f4a1 100644 --- a/clu/values.py +++ b/clu/values.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/setup.py b/setup.py index 7983bd2..3882b7d 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,4 @@ -# Copyright 2022 The CLU Authors. +# Copyright 2023 The CLU Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License.