Add copyright headers to torchvct tests

facebookresearch · Nov 8, 2023 · fbaf7e7 · fbaf7e7
1 parent 0cd49a9
commit fbaf7e7
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/projects/torch_vct/tests/neural/test_entropy_model_layers.py b/projects/torch_vct/tests/neural/test_entropy_model_layers.py
@@ -1,10 +1,14 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+
 import pytest
 import torch
 
 from projects.torch_vct.neural.entropy_model_layers import (
     LearnedPosition,
     StartSym,
-    Transformer,
     TransformerBlock,
 )
 

diff --git a/projects/torch_vct/tests/neural/test_layers_utils.py b/projects/torch_vct/tests/neural/test_layers_utils.py
@@ -1,3 +1,8 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+
 import pytest
 import torch
 

diff --git a/projects/torch_vct/tests/neural/test_patcher.py b/projects/torch_vct/tests/neural/test_patcher.py
@@ -1,3 +1,8 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+
 import pytest
 import torch