Allow subclassing Collection.

CLU Authors · copybara-github · commit e8417a955304 · 2023-04-04T10:10:08.000-07:00
PiperOrigin-RevId: 521792975
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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.

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-# Copyright 2022 The CLU Authors.`
	`1`	`+# Copyright 2023 The CLU Authors.`
`2`	`2`	`#`
`3`	`3`	`# Licensed under the Apache License, Version 2.0 (the "License");`
`4`	`4`	`# you may not use this file except in compliance with the License.`