From fbaf7e727c3636711c948279cd27c180d9404802 Mon Sep 17 00:00:00 2001 From: Matthew Muckley Date: Wed, 8 Nov 2023 07:09:40 -0800 Subject: [PATCH] Add copyright headers to torchvct tests --- .../torch_vct/tests/neural/test_entropy_model_layers.py | 6 +++++- projects/torch_vct/tests/neural/test_layers_utils.py | 5 +++++ projects/torch_vct/tests/neural/test_patcher.py | 5 +++++ 3 files changed, 15 insertions(+), 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 index d272fcd2..9a4d4b7a 100644 --- 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 index 625d7ea3..5f3cf607 100644 --- 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 index be360534..2e3c730c 100644 --- 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