Update and fix copyright headers (#1589)

Updates the header according to
https://docs.opensource.microsoft.com/releasing/general-guidance/copyright-headers/
and enforce it with ruff. Ref:
https://docs.opensource.microsoft.com/news/2019-07-02-cela-all-rights-reserved/

Fix additional ruff lint errors due to enabling `--preview` rules for
the copyright header lint check.

docs/conf.py

@@ -1,5 +1,9 @@
-# Configuration file for the Sphinx documentation builder.
-# To run the documentation: python -m sphinx docs dist/html
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+"""Configuration file for the Sphinx documentation builder.
+
+To run the documentation: python -m sphinx docs dist/html
+"""
 
 import os
 import re

docs/examples/01_plot_selu.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """
 Generating a FunctionProto
 ==========================

docs/examples/02_plot_square_loss.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """
 Generating a ModelProto
 =======================

docs/examples/03_export_lib.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """
 Generating a LibProto
 =====================

docs/examples/04_plot_eager_mode_evaluation.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """
 Eager mode evaluation
 =====================

docs/examples/05_plot_model_props.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """
 ModelProto Properties
 =====================

docs/examples/06_plot_model_local_funs.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """
 Model Local Functions
 =====================

docs/test/test_documentation_examples.py

@@ -1,7 +1,5 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------
 
 import os
 import subprocess

docs/tutorial/examples/dropout.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnxscript import opset15 as op
 from onnxscript import script
 

docs/tutorial/examples/firstdim.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnxscript import opset15 as op
 from onnxscript import script
 

docs/tutorial/examples/forloop.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnxscript import opset15 as op
 from onnxscript import script
 

docs/tutorial/examples/forwhileloop.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnxscript import opset15 as op
 from onnxscript import script
 

docs/tutorial/examples/hardmax_end_to_end.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 import onnx
 
 # We use ONNX opset 15 to define the function below.

docs/tutorial/examples/leaky_relu.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnxscript import opset15 as op
 from onnxscript import script
 

docs/tutorial/examples/leaky_relu_attr_promoted.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnxscript import opset15 as op
 from onnxscript import script
 

docs/tutorial/examples/omitted_input.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnxscript import opset15 as op
 from onnxscript import script
 

docs/tutorial/examples/outerscope_redef_error.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnxscript import graph, script
 from onnxscript import opset15 as op
 
@@ -13,7 +15,7 @@ def Sum(sum_in, next):
             return sum_out, sum_out
 
         g = op.Constant(value=1)
-        all_sum, cumulative_sum = op.Scan(0, X, body=Sum, num_scan_inputs=1)
+        _all_sum, cumulative_sum = op.Scan(0, X, body=Sum, num_scan_inputs=1)
         return cumulative_sum
 
 except Exception as e:

docs/tutorial/examples/scanloop.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnxscript import graph, script
 from onnxscript import opset15 as op
 
@@ -9,5 +11,5 @@ def Sum(sum_in, next):
         sum_out = sum_in + next
         return sum_out, sum_out
 
-    all_sum, cumulative_sum = op.Scan(0, X, body=Sum, num_scan_inputs=1)
+    _all_sum, cumulative_sum = op.Scan(0, X, body=Sum, num_scan_inputs=1)
     return cumulative_sum

docs/tutorial/examples/softplus.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 # We use ONNX opset 15 to define the function below.
 from onnxscript import opset15 as op
 from onnxscript import script

docs/tutorial/examples/tensor_attr.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnx import TensorProto, helper
 
 from onnxscript import opset15 as op

docs/tutorial/examples/tensor_attr2.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnx import TensorProto, helper
 
 from onnxscript import opset15 as op

docs/tutorial/examples/tensor_attr_short.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnxscript import opset15 as op
 from onnxscript import script
 

docs/tutorial/examples/whileloop.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from onnx import TensorProto
 from onnx.helper import make_tensor
 

docs/tutorial/rewriter/examples/broadcast_matmul.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """Onnx Pattern Rewriting with match condition parameter.
 
 This script shows how to define a rewriting rule based on patterns while

docs/tutorial/rewriter/examples/erfgelu.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """Onnx Pattern Rewriting.
 
 This script shows how to define a rewriting rule based on patterns.

docs/update_readme.py

@@ -1,4 +1,6 @@
-# Script to update end-to-end example in README.md.
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+"""Script to update end-to-end example in README.md."""
 
 updated_readme = []
 with open("README.md", encoding="utf-8") as f:
@@ -12,7 +14,7 @@
             with open(
                 "docs/tutorial/examples/hardmax_end_to_end.py", encoding="utf-8"
             ) as example_f:
-                example_code = example_f.readlines()
+                example_code = example_f.readlines()[2:]  # Skip the copyright header
                 updated_readme += example_code
         if line == "```\n" and in_stub:
             updated_readme.append(line)

examples/pattern_rewriting.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """Onnx Pattern Rewriting.
 
 This script shows how to define a rewriting rule based on patterns.
@@ -74,7 +76,9 @@ def rotary_match_pattern(op, x, pos_ids, axis):
 
     matmul = op.MatMul(pos_ids, cast)
     transpose = op.Transpose(matmul)
-    output, length = op.ConcatTraining(transpose, transpose, domain="com.microsoft", outputs=2)
+    output, _length = op.ConcatTraining(
+        transpose, transpose, domain="com.microsoft", outputs=2
+    )
 
     sin = op.Sin(output)
     cast1 = op.Cast(sin, to=onnx.TensorProto.FLOAT)

noxfile.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """Test with different environment configuration with nox.
 
 Documentation:

onnxscript/__init__.py

@@ -1,7 +1,5 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------
 
 __all__ = [
     "script",

onnxscript/_internal/analysis.py

@@ -1,7 +1,5 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------
 from __future__ import annotations
 
 import ast

onnxscript/_internal/analysis_test.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from __future__ import annotations
 
 import ast

onnxscript/_internal/ast_utils.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """Utilities for working with Python ASTs."""
 
 from __future__ import annotations

onnxscript/_internal/autocast.py

@@ -1,7 +1,5 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------
 
 from __future__ import annotations
 

onnxscript/_internal/deprecation.py

@@ -1,7 +1,5 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------
 """Utility for deprecating APIs."""
 
 # Reference: https://github.com/pytorch/pytorch/blob/aed9bee0413dac190452fbfa9ab2a44b6e6843f5/torch/onnx/_deprecation.py

onnxscript/_internal/param_manipulation.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """Function for manipulating input parameters of an Op or a OnnxFunction."""
 
 from __future__ import annotations

onnxscript/_internal/param_manipulation_test.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 # mypy: disable-error-code=misc
 
 import collections

onnxscript/_internal/runtime_typing.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """An internal wrapper for the beartype library.
 
 Decorate a function with `@runtime_typing.checked` to enable runtime

onnxscript/_internal/utils.py

@@ -1,7 +1,5 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------
 from __future__ import annotations
 
 import numbers

onnxscript/_internal/version_utils.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """Version utils for testing."""
 
 import packaging.version

onnxscript/_legacy_ir/__init__.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from __future__ import annotations
 
 import dataclasses

onnxscript/_legacy_ir/visitor.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 from __future__ import annotations
 
 import dataclasses

onnxscript/_legacy_ir/visitor_test.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 import unittest
 
 import onnx

onnxscript/_thirdparty/asciichartpy.py

@@ -1,5 +1,5 @@
-# SPDX-License-Identifier: MIT
-# Modifications Copyright (c) Microsoft.
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 #
 # Copyright © 2016 Igor Kroitor
 #

onnxscript/backend/__init__.py

@@ -1,4 +1,2 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------

onnxscript/backend/onnx_backend.py

@@ -1,7 +1,5 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------
 
 
 import os
@@ -291,7 +289,7 @@ def enumerate_onnx_tests(series, fct_filter=None) -> Iterator[OnnxBackendTest]:
     sub = os.path.join(root, "data", series)
     if not os.path.exists(sub):
         raise FileNotFoundError(
-            "Unable to find series of tests in {root!r}, subfolders:\n"
+            f"Unable to find series of tests in {root!r}, subfolders:\n"
             + "\n".join(os.listdir(root))
         )
     tests = os.listdir(sub)

onnxscript/backend/onnx_backend_test.py

@@ -1,7 +1,5 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------
 
 import os
 import unittest

onnxscript/backend/onnx_export.py

@@ -1,7 +1,5 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------
 from __future__ import annotations
 
 from typing import Any, Optional, Sequence
@@ -311,7 +309,7 @@ def _translate_onnx_var_ref(self, var):
 
     def _rename_domain(self, domain: str) -> str:
         if domain in {"", "ai.onnx"}:
-            return "opset"  #  TODO: Need checks to avoid name conflicts.
+            return "opset"  # TODO: Need checks to avoid name conflicts.
         return _cleanup_variable_name(domain)  # type: ignore[return-value]
 
     def _make_opset_name(self, domain, version):

onnxscript/backend/onnx_export_test.py

@@ -1,7 +1,5 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
+# Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-# --------------------------------------------------------------------------
 from __future__ import annotations
 
 import dataclasses