Code update

h-joo · The dataclass_array Authors · commit f575941e42b4 · 2026-07-02T03:53:30.000-07:00
PiperOrigin-RevId: 941349373
diff --git a/dataclass_array/__init__.py b/dataclass_array/__init__.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/array_dataclass.py b/dataclass_array/array_dataclass.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/array_dataclass_test.py b/dataclass_array/array_dataclass_test.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/conftest.py b/dataclass_array/conftest.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/field_utils.py b/dataclass_array/field_utils.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/import_test.py b/dataclass_array/import_test.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/ops.py b/dataclass_array/ops.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/shape_parsing.py b/dataclass_array/shape_parsing.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/shape_parsing_test.py b/dataclass_array/shape_parsing_test.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/testing.py b/dataclass_array/testing.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/type_parsing.py b/dataclass_array/type_parsing.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/type_parsing_test.py b/dataclass_array/type_parsing_test.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/typing.py b/dataclass_array/typing.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/utils/file_utils.py b/dataclass_array/utils/file_utils.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/utils/inspect_utils.py b/dataclass_array/utils/inspect_utils.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/utils/inspect_utils_test.py b/dataclass_array/utils/inspect_utils_test.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/utils/np_utils.py b/dataclass_array/utils/np_utils.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -78,7 +78,7 @@ def is_array(
 
 
 def asarray(
-    x: Union[DcOrArrayT, ArrayLike[Array['...']]],
+    x: Union[DcOrArrayT, ArrayLike[Array['...']]],  # pyrefly: ignore[not-a-type]
     *,
     xnp: enp.NpModule = None,
     dtype: Optional[DTypeArg] = None,
@@ -105,7 +105,7 @@ def asarray(
   # Potentially forward `None`, if optional is accepted
   if x is None:
     if optional:
-      return x
+      return x  # pyrefly: ignore[bad-return]
     else:
       raise ValueError('Expected array, got `None`')
 
@@ -120,10 +120,10 @@ def asarray(
         not isinstance(dtype, type) or not isinstance(x, dtype)
     ):
       raise TypeError(f'Expected {dtype}. Got: {type(x)}')
-    return x.as_xnp(xnp)
+    return x.as_xnp(xnp)  # pyrefly: ignore[bad-return]
 
   # Handle ndarray
-  dtype = enp.dtypes.DType.from_value(dtype)
+  dtype = enp.dtypes.DType.from_value(dtype)  # pyrefly: ignore[bad-assignment]
   return dtype.asarray(x, xnp=xnp, casting='all' if cast_dtype else 'none')
 
 
diff --git a/dataclass_array/utils/np_utils_test.py b/dataclass_array/utils/np_utils_test.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/utils/py_utils.py b/dataclass_array/utils/py_utils.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/utils/tree_utils.py b/dataclass_array/utils/tree_utils.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -38,8 +38,8 @@
 
 
 def tree_map(  # pylint: disable=redefined-builtin
-    fn: Callable[[Unpack[Tree[_InsT]]], _OutT],
-    *trees: Unpack[Tree[_InsT]],
+    fn: Callable[[Unpack[Tree[_InsT]]], _OutT],  # pyrefly: ignore[bad-specialization]
+    *trees: Unpack[Tree[_InsT]],  # pyrefly: ignore[bad-specialization]
 ) -> Tree[_OutT]:
   """Apply a function recursively to each element of a nested data struct.
 
diff --git a/dataclass_array/utils/tree_utils_test.py b/dataclass_array/utils/tree_utils_test.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -34,7 +34,7 @@ def map_fn(x):
 
   result = tree_utils.tree_map(
       map_fn,
-      {
+      {  # pyrefly: ignore[bad-argument-type]
           'a': 1,
           'b': {
               'c': 2,
@@ -50,12 +50,12 @@ def map_fn(x):
       },
   }
 
-  assert tree_utils.tree_map(map_fn, [1, 2, 3]) == [10, 20, 30]
-  assert tree_utils.tree_map(map_fn, (1, 2, 3)) == (10, 20, 30)
-  assert tree_utils.tree_map(map_fn, {}) == {}  # pylint: disable=g-explicit-bool-comparison
-  assert tree_utils.tree_map(map_fn, ()) == ()  # pylint: disable=g-explicit-bool-comparison
-  assert tree_utils.tree_map(map_fn, {'x': ([])}) == {'x': ([])}
-  assert tree_utils.tree_map(map_fn, 1) == 10
+  assert tree_utils.tree_map(map_fn, [1, 2, 3]) == [10, 20, 30]  # pyrefly: ignore[bad-argument-type]
+  assert tree_utils.tree_map(map_fn, (1, 2, 3)) == (10, 20, 30)  # pyrefly: ignore[bad-argument-type]
+  assert tree_utils.tree_map(map_fn, {}) == {}  # pylint: disable=g-explicit-bool-comparison  # pyrefly: ignore[bad-argument-type]
+  assert tree_utils.tree_map(map_fn, ()) == ()  # pylint: disable=g-explicit-bool-comparison  # pyrefly: ignore[bad-argument-type]
+  assert tree_utils.tree_map(map_fn, {'x': ([])}) == {'x': ([])}  # pyrefly: ignore[bad-argument-type]
+  assert tree_utils.tree_map(map_fn, 1) == 10  # pyrefly: ignore[bad-argument-type]
 
 
 def test_tree_map_multi_args():
@@ -72,7 +72,7 @@ def add_fn(x, y):
 def test_tree_map_chex():
   """Test tree_map with chex dataclasses."""
 
-  assert tree_utils.tree_map(lambda x: x * 10, A(x=1, y=2)) == A(x=10, y=20)
+  assert tree_utils.tree_map(lambda x: x * 10, A(x=1, y=2)) == A(x=10, y=20)  # pyrefly: ignore[bad-argument-type, unsupported-operation]
 
   def add_fn(x, y):
     return x + y
diff --git a/dataclass_array/vectorization.py b/dataclass_array/vectorization.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/dataclass_array/vectorization_test.py b/dataclass_array/vectorization_test.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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/docs/conf.py b/docs/conf.py
@@ -1,4 +1,4 @@
-# Copyright 2025 The dataclass_array Authors.
+# Copyright 2026 The dataclass_array 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 2025 The dataclass_array Authors.`
	`1`	`+# Copyright 2026 The dataclass_array 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.`