From a14b6c09de814b0ab9b1525780e853b609ddea0b Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Thu, 8 Jun 2023 02:41:21 -0700 Subject: [PATCH] Remove references to deprecated submodule jax.abstract_arrays Use jax.core instead (see https://github.com/google/jax/pull/16271) PiperOrigin-RevId: 538729607 --- diffren/__init__.py | 2 +- diffren/common/__init__.py | 2 +- diffren/common/compare_images.py | 2 +- diffren/common/kernels/rasterize_test_utils.cc | 2 +- diffren/common/kernels/rasterize_test_utils.h | 2 +- diffren/common/kernels/rasterize_triangles_impl.cc | 2 +- diffren/common/kernels/rasterize_triangles_impl.h | 2 +- .../kernels/rasterize_triangles_impl_cuda.cu.cc | 2 +- .../kernels/rasterize_triangles_impl_cuda.cu.h | 2 +- .../kernels/rasterize_triangles_impl_cuda_test.cc | 2 +- .../common/kernels/rasterize_triangles_impl_test.cc | 2 +- diffren/common/kernels/rasterize_triangles_types.h | 2 +- diffren/common/kernels/rasterize_utils.h | 2 +- diffren/common/obj_loader.py | 2 +- diffren/common/obj_loader_test.py | 2 +- diffren/common/test_utils.py | 2 +- diffren/jax/__init__.py | 2 +- diffren/jax/camera.py | 2 +- diffren/jax/camera_test.py | 2 +- diffren/jax/composite.py | 2 +- diffren/jax/constants.py | 2 +- diffren/jax/internal/__init__.py | 2 +- diffren/jax/internal/framebuffer.py | 2 +- diffren/jax/internal/framebuffer_test.py | 2 +- diffren/jax/internal/interpolate.py | 2 +- diffren/jax/internal/interpolate_test.py | 2 +- diffren/jax/internal/kernels/descriptors.proto | 2 +- .../jax/internal/kernels/rasterize_triangles_cpu.cc | 2 +- .../jax/internal/kernels/rasterize_triangles_gpu.cc | 2 +- .../jax/internal/kernels/rasterize_triangles_xla.py | 13 ++++++------- .../kernels/rasterize_triangles_xla_test.py | 2 +- diffren/jax/internal/rasterize.py | 2 +- diffren/jax/internal/rasterize_test.py | 2 +- diffren/jax/internal/splat.py | 2 +- diffren/jax/internal/splat_test.py | 2 +- diffren/jax/render.py | 2 +- diffren/jax/render_test.py | 2 +- diffren/jax/utils/__init__.py | 2 +- diffren/jax/utils/depthmap.py | 2 +- diffren/jax/utils/depthmap_test.py | 2 +- diffren/jax/utils/image.py | 2 +- diffren/jax/utils/image_test.py | 2 +- diffren/jax/utils/mesh.py | 2 +- diffren/jax/utils/mesh_test.py | 2 +- diffren/jax/utils/shaders.py | 2 +- diffren/jax/utils/shaders_test.py | 2 +- diffren/jax/utils/transforms.py | 2 +- diffren/jax/utils/transforms_test.py | 2 +- 48 files changed, 53 insertions(+), 54 deletions(-) diff --git a/diffren/__init__.py b/diffren/__init__.py index b490bb7..bd21148 100644 --- a/diffren/__init__.py +++ b/diffren/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/common/__init__.py b/diffren/common/__init__.py index 9a34e74..fea290b 100644 --- a/diffren/common/__init__.py +++ b/diffren/common/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/common/compare_images.py b/diffren/common/compare_images.py index ccb4e0c..5c3c4bd 100644 --- a/diffren/common/compare_images.py +++ b/diffren/common/compare_images.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/common/kernels/rasterize_test_utils.cc b/diffren/common/kernels/rasterize_test_utils.cc index 1a96d9a..646905c 100644 --- a/diffren/common/kernels/rasterize_test_utils.cc +++ b/diffren/common/kernels/rasterize_test_utils.cc @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/common/kernels/rasterize_test_utils.h b/diffren/common/kernels/rasterize_test_utils.h index e2f1864..9f04cd3 100644 --- a/diffren/common/kernels/rasterize_test_utils.h +++ b/diffren/common/kernels/rasterize_test_utils.h @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/common/kernels/rasterize_triangles_impl.cc b/diffren/common/kernels/rasterize_triangles_impl.cc index 36313d2..b791844 100644 --- a/diffren/common/kernels/rasterize_triangles_impl.cc +++ b/diffren/common/kernels/rasterize_triangles_impl.cc @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/common/kernels/rasterize_triangles_impl.h b/diffren/common/kernels/rasterize_triangles_impl.h index 06e2d49..441df36 100644 --- a/diffren/common/kernels/rasterize_triangles_impl.h +++ b/diffren/common/kernels/rasterize_triangles_impl.h @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/common/kernels/rasterize_triangles_impl_cuda.cu.cc b/diffren/common/kernels/rasterize_triangles_impl_cuda.cu.cc index f1f4fc2..f3e5e60 100644 --- a/diffren/common/kernels/rasterize_triangles_impl_cuda.cu.cc +++ b/diffren/common/kernels/rasterize_triangles_impl_cuda.cu.cc @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/common/kernels/rasterize_triangles_impl_cuda.cu.h b/diffren/common/kernels/rasterize_triangles_impl_cuda.cu.h index 51faf8a..9cefbf5 100644 --- a/diffren/common/kernels/rasterize_triangles_impl_cuda.cu.h +++ b/diffren/common/kernels/rasterize_triangles_impl_cuda.cu.h @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/common/kernels/rasterize_triangles_impl_cuda_test.cc b/diffren/common/kernels/rasterize_triangles_impl_cuda_test.cc index 85d7d1c..a1ea5a3 100644 --- a/diffren/common/kernels/rasterize_triangles_impl_cuda_test.cc +++ b/diffren/common/kernels/rasterize_triangles_impl_cuda_test.cc @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/common/kernels/rasterize_triangles_impl_test.cc b/diffren/common/kernels/rasterize_triangles_impl_test.cc index c1ce960..9025bf5 100644 --- a/diffren/common/kernels/rasterize_triangles_impl_test.cc +++ b/diffren/common/kernels/rasterize_triangles_impl_test.cc @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/common/kernels/rasterize_triangles_types.h b/diffren/common/kernels/rasterize_triangles_types.h index ca1dabe..319da01 100644 --- a/diffren/common/kernels/rasterize_triangles_types.h +++ b/diffren/common/kernels/rasterize_triangles_types.h @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/common/kernels/rasterize_utils.h b/diffren/common/kernels/rasterize_utils.h index 39a448c..87c13e9 100644 --- a/diffren/common/kernels/rasterize_utils.h +++ b/diffren/common/kernels/rasterize_utils.h @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/common/obj_loader.py b/diffren/common/obj_loader.py index 7f6c151..f377f33 100644 --- a/diffren/common/obj_loader.py +++ b/diffren/common/obj_loader.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/common/obj_loader_test.py b/diffren/common/obj_loader_test.py index 496ba76..b9c1103 100644 --- a/diffren/common/obj_loader_test.py +++ b/diffren/common/obj_loader_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/common/test_utils.py b/diffren/common/test_utils.py index ea2d261..108911d 100644 --- a/diffren/common/test_utils.py +++ b/diffren/common/test_utils.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/__init__.py b/diffren/jax/__init__.py index 9a34e74..fea290b 100644 --- a/diffren/jax/__init__.py +++ b/diffren/jax/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/camera.py b/diffren/jax/camera.py index bfdca29..10b1325 100644 --- a/diffren/jax/camera.py +++ b/diffren/jax/camera.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/camera_test.py b/diffren/jax/camera_test.py index 1d84899..db672e9 100644 --- a/diffren/jax/camera_test.py +++ b/diffren/jax/camera_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/composite.py b/diffren/jax/composite.py index 3d66752..a7e0cb3 100644 --- a/diffren/jax/composite.py +++ b/diffren/jax/composite.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/constants.py b/diffren/jax/constants.py index 140bd9f..5671125 100644 --- a/diffren/jax/constants.py +++ b/diffren/jax/constants.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/internal/__init__.py b/diffren/jax/internal/__init__.py index 9a34e74..fea290b 100644 --- a/diffren/jax/internal/__init__.py +++ b/diffren/jax/internal/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/internal/framebuffer.py b/diffren/jax/internal/framebuffer.py index 6f34bac..7d0ae65 100644 --- a/diffren/jax/internal/framebuffer.py +++ b/diffren/jax/internal/framebuffer.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/internal/framebuffer_test.py b/diffren/jax/internal/framebuffer_test.py index 062a319..da29631 100644 --- a/diffren/jax/internal/framebuffer_test.py +++ b/diffren/jax/internal/framebuffer_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/internal/interpolate.py b/diffren/jax/internal/interpolate.py index 6124526..1ed12ed 100644 --- a/diffren/jax/internal/interpolate.py +++ b/diffren/jax/internal/interpolate.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/internal/interpolate_test.py b/diffren/jax/internal/interpolate_test.py index 67f3bce..01271f3 100644 --- a/diffren/jax/internal/interpolate_test.py +++ b/diffren/jax/internal/interpolate_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/internal/kernels/descriptors.proto b/diffren/jax/internal/kernels/descriptors.proto index 254234a..1f14238 100644 --- a/diffren/jax/internal/kernels/descriptors.proto +++ b/diffren/jax/internal/kernels/descriptors.proto @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/jax/internal/kernels/rasterize_triangles_cpu.cc b/diffren/jax/internal/kernels/rasterize_triangles_cpu.cc index c8bbf5a..b22df8f 100644 --- a/diffren/jax/internal/kernels/rasterize_triangles_cpu.cc +++ b/diffren/jax/internal/kernels/rasterize_triangles_cpu.cc @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/jax/internal/kernels/rasterize_triangles_gpu.cc b/diffren/jax/internal/kernels/rasterize_triangles_gpu.cc index e3ad66a..c7f6932 100644 --- a/diffren/jax/internal/kernels/rasterize_triangles_gpu.cc +++ b/diffren/jax/internal/kernels/rasterize_triangles_gpu.cc @@ -1,4 +1,4 @@ -/* Copyright 2022 The diffren Authors. +/* Copyright 2023 The diffren 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/diffren/jax/internal/kernels/rasterize_triangles_xla.py b/diffren/jax/internal/kernels/rasterize_triangles_xla.py index 3d422c0..a06b061 100644 --- a/diffren/jax/internal/kernels/rasterize_triangles_xla.py +++ b/diffren/jax/internal/kernels/rasterize_triangles_xla.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -20,7 +20,6 @@ from . import rasterize_triangles_gpu from diffren.jax.internal.kernels import descriptors_pb2 import jax -from jax import abstract_arrays from jax.interpreters import batching import jax.numpy as jnp import numpy as np @@ -150,11 +149,11 @@ def rasterize_triangles_abstract_eval(vertices, triangles, *, image_width, assert triangles.shape[1] == 3 del face_culling_mode - return (abstract_arrays.ShapedArray((num_layers, image_height, image_width), - dtype=np.int32), - abstract_arrays.ShapedArray((num_layers, image_height, image_width), - dtype=np.float32), - abstract_arrays.ShapedArray( + return (jax.core.ShapedArray((num_layers, image_height, image_width), + dtype=np.int32), + jax.core.ShapedArray((num_layers, image_height, image_width), + dtype=np.float32), + jax.core.ShapedArray( (num_layers, image_height, image_width, 3), dtype=np.float32)) diff --git a/diffren/jax/internal/kernels/rasterize_triangles_xla_test.py b/diffren/jax/internal/kernels/rasterize_triangles_xla_test.py index b86ef23..1ceb584 100644 --- a/diffren/jax/internal/kernels/rasterize_triangles_xla_test.py +++ b/diffren/jax/internal/kernels/rasterize_triangles_xla_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/internal/rasterize.py b/diffren/jax/internal/rasterize.py index 4993986..df801dc 100644 --- a/diffren/jax/internal/rasterize.py +++ b/diffren/jax/internal/rasterize.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/internal/rasterize_test.py b/diffren/jax/internal/rasterize_test.py index d6ed364..0919823 100644 --- a/diffren/jax/internal/rasterize_test.py +++ b/diffren/jax/internal/rasterize_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/internal/splat.py b/diffren/jax/internal/splat.py index 5811200..50180c3 100644 --- a/diffren/jax/internal/splat.py +++ b/diffren/jax/internal/splat.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/internal/splat_test.py b/diffren/jax/internal/splat_test.py index bdc5f0a..8eb867a 100644 --- a/diffren/jax/internal/splat_test.py +++ b/diffren/jax/internal/splat_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/render.py b/diffren/jax/render.py index 91b1a9f..0e1ae0a 100644 --- a/diffren/jax/render.py +++ b/diffren/jax/render.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/render_test.py b/diffren/jax/render_test.py index 2e06e89..a96e131 100644 --- a/diffren/jax/render_test.py +++ b/diffren/jax/render_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/__init__.py b/diffren/jax/utils/__init__.py index 9a34e74..fea290b 100644 --- a/diffren/jax/utils/__init__.py +++ b/diffren/jax/utils/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/depthmap.py b/diffren/jax/utils/depthmap.py index 2c82b65..59d0460 100644 --- a/diffren/jax/utils/depthmap.py +++ b/diffren/jax/utils/depthmap.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/depthmap_test.py b/diffren/jax/utils/depthmap_test.py index 73f30f4..81a3792 100644 --- a/diffren/jax/utils/depthmap_test.py +++ b/diffren/jax/utils/depthmap_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/image.py b/diffren/jax/utils/image.py index e71df9a..7838faf 100644 --- a/diffren/jax/utils/image.py +++ b/diffren/jax/utils/image.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/image_test.py b/diffren/jax/utils/image_test.py index 750f523..c0ac9e1 100644 --- a/diffren/jax/utils/image_test.py +++ b/diffren/jax/utils/image_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/mesh.py b/diffren/jax/utils/mesh.py index cb41015..ebb06a7 100644 --- a/diffren/jax/utils/mesh.py +++ b/diffren/jax/utils/mesh.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/mesh_test.py b/diffren/jax/utils/mesh_test.py index 050aa62..87123e3 100644 --- a/diffren/jax/utils/mesh_test.py +++ b/diffren/jax/utils/mesh_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/shaders.py b/diffren/jax/utils/shaders.py index 9bd2e6e..e9c0e5e 100644 --- a/diffren/jax/utils/shaders.py +++ b/diffren/jax/utils/shaders.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/shaders_test.py b/diffren/jax/utils/shaders_test.py index 77a1ea1..a7e4958 100644 --- a/diffren/jax/utils/shaders_test.py +++ b/diffren/jax/utils/shaders_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/transforms.py b/diffren/jax/utils/transforms.py index 28fd1b9..5b44713 100644 --- a/diffren/jax/utils/transforms.py +++ b/diffren/jax/utils/transforms.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren 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/diffren/jax/utils/transforms_test.py b/diffren/jax/utils/transforms_test.py index 5f5e8f6..f0d4b80 100644 --- a/diffren/jax/utils/transforms_test.py +++ b/diffren/jax/utils/transforms_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 The diffren Authors. +# Copyright 2023 The diffren Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License.