added missing license headers

Signed-off-by: Alp Dener <[email protected]>
NVIDIA · Nov 21, 2023 · 71c16ae · 71c16ae
1 parent f2e5fa9
commit 71c16ae
Show file tree

Hide file tree

Showing 10 changed files with 30 additions and 1 deletion.
diff --git a/examples/jax/__init__.py b/examples/jax/__init__.py
@@ -0,0 +1,3 @@
+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
diff --git a/examples/jax/encoder/__init__.py b/examples/jax/encoder/__init__.py
@@ -0,0 +1,3 @@
+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
diff --git a/examples/jax/mnist/__init__.py b/examples/jax/mnist/__init__.py
@@ -0,0 +1,3 @@
+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
diff --git a/examples/paddle/__init__.py b/examples/paddle/__init__.py
@@ -0,0 +1,3 @@
+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
diff --git a/examples/paddle/mnist/__init__.py b/examples/paddle/mnist/__init__.py
@@ -0,0 +1,3 @@
+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
diff --git a/examples/pytorch/__init__.py b/examples/pytorch/__init__.py
@@ -0,0 +1,3 @@
+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
diff --git a/examples/pytorch/mnist/__init__.py b/examples/pytorch/mnist/__init__.py
@@ -0,0 +1,3 @@
+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
diff --git a/setup.py b/setup.py
@@ -617,7 +617,7 @@ def main():
             }
         )
         test_paths = []
-        for path, dirs, files in os.walk(f'tests/{framework}'):
+        for path, _, _ in os.walk(f'tests/{framework}'):
             test_paths += [ f'{path}/*.py' ]
         package_data.update( { f'transformer_engine.{framework}.tests': test_paths } )
 

diff --git a/tests/jax/test_encoder.py b/tests/jax/test_encoder.py
@@ -1,3 +1,7 @@
+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
+
 import pytest
 import transformer_engine.jax as te
 from transformer_engine.jax.examples.encoder.single_gpu import encoder_parser, train_and_evaluate

diff --git a/tests/jax/test_mnist.py b/tests/jax/test_mnist.py
@@ -1,3 +1,7 @@
+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
+
 import pytest
 import transformer_engine.jax as te
 from transformer_engine.jax.examples.mnist.single_gpu import mnist_parser, train_and_evaluate