diff --git a/pybuda/test/mlir/llama/utils/utils.py b/pybuda/test/mlir/llama/utils/utils.py
index 03751ccb0..a3b7f1d78 100644
--- a/pybuda/test/mlir/llama/utils/utils.py
+++ b/pybuda/test/mlir/llama/utils/utils.py
@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+
+# SPDX-License-Identifier: Apache-2.0
+
 from transformers import LlamaConfig, LlamaForCausalLM, LlamaTokenizer
 
 import pybuda
diff --git a/pybuda/test/mlir/mnist/training/mnist_linear_pybuda.py b/pybuda/test/mlir/mnist/training/mnist_linear_pybuda.py
index eb0364aae..dba714365 100644
--- a/pybuda/test/mlir/mnist/training/mnist_linear_pybuda.py
+++ b/pybuda/test/mlir/mnist/training/mnist_linear_pybuda.py
@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+
+# SPDX-License-Identifier: Apache-2.0
+
 import torch
 from torchvision import datasets, transforms
 from torch.utils.tensorboard import SummaryWriter
diff --git a/pybuda/test/mlir/mnist/training/mnist_linear_pytorch.py b/pybuda/test/mlir/mnist/training/mnist_linear_pytorch.py
index 7958745e4..a9a4b8f92 100644
--- a/pybuda/test/mlir/mnist/training/mnist_linear_pytorch.py
+++ b/pybuda/test/mlir/mnist/training/mnist_linear_pytorch.py
@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+
+# SPDX-License-Identifier: Apache-2.0
+
 import torch
 from torchvision import datasets, transforms
 from torch.utils.tensorboard import SummaryWriter