From da12b65e2d102f01bc65d6fc442c0382f8fc89fd Mon Sep 17 00:00:00 2001
From: David Yastremsky <dyastremsky@nvidia.com>
Date: Tue, 28 Jan 2025 09:41:18 -0800
Subject: [PATCH] Add yml handler

---
 tools/add_copyright.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/add_copyright.py b/tools/add_copyright.py
index dc126c67..a95f0c50 100644
--- a/tools/add_copyright.py
+++ b/tools/add_copyright.py
@@ -216,7 +216,7 @@ def register_impl(func):
 
 @register(
     any_of(
-        has_ext([".py", ".pyi", ".sh", ".bash", ".yaml", ".pbtxt"]),
+        has_ext([".py", ".pyi", ".sh", ".bash", ".yaml", ".yml", ".pbtxt"]),
         basename_is("CMakeLists.txt"),
         path_contains("Dockerfile"),
     )