You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@hamilton.apache.org by sk...@apache.org on 2025/12/28 09:32:37 UTC
(hamilton) 03/04: Adds headers to ui/sdk/tests
This is an automated email from the ASF dual-hosted git repository.
skrawcz pushed a commit to branch add_more_licenses
in repository https://gitbox.apache.org/repos/asf/hamilton.git
commit 0d767e5280a83f5192a830cc7760542d400c9fe8
Author: Stefan Krawczyk <st...@cs.stanford.edu>
AuthorDate: Sun Dec 28 20:29:34 2025 +1100
Adds headers to ui/sdk/tests
---
ui/sdk/tests/conftest.py | 17 +++++++++++++++++
ui/sdk/tests/resources/basic_dag_with_config.py | 17 +++++++++++++++++
ui/sdk/tests/resources/parallel_dag.py | 17 +++++++++++++++++
ui/sdk/tests/resources/parallel_dag_error.py | 17 +++++++++++++++++
ui/sdk/tests/test_adapters.py | 17 +++++++++++++++++
ui/sdk/tests/test_driver.py | 17 +++++++++++++++++
ui/sdk/tests/test_package_to_hash/submodule1.py | 17 +++++++++++++++++
ui/sdk/tests/test_package_to_hash/submodule2.py | 17 +++++++++++++++++
ui/sdk/tests/test_tracking.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_constants.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_dataframe_stats.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_ibis_stats.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_langchain_stats.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_pandas_col_stats.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_pandas_stats.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_polars_col_stats.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_polars_stats.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_pydantic_stats.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_pyspark_stats.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_runs.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_stats.py | 17 +++++++++++++++++
ui/sdk/tests/tracking/test_utils.py | 17 +++++++++++++++++
22 files changed, 374 insertions(+)
diff --git a/ui/sdk/tests/conftest.py b/ui/sdk/tests/conftest.py
index 04d11ddc..ce510679 100644
--- a/ui/sdk/tests/conftest.py
+++ b/ui/sdk/tests/conftest.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
from hamilton import telemetry
telemetry.disable_telemetry()
diff --git a/ui/sdk/tests/resources/basic_dag_with_config.py b/ui/sdk/tests/resources/basic_dag_with_config.py
index 5c10b41a..098fd34e 100644
--- a/ui/sdk/tests/resources/basic_dag_with_config.py
+++ b/ui/sdk/tests/resources/basic_dag_with_config.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
from hamilton.function_modifiers import config
diff --git a/ui/sdk/tests/resources/parallel_dag.py b/ui/sdk/tests/resources/parallel_dag.py
index d6ff78e7..f17fa527 100644
--- a/ui/sdk/tests/resources/parallel_dag.py
+++ b/ui/sdk/tests/resources/parallel_dag.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import dataclasses
import os
from typing import List
diff --git a/ui/sdk/tests/resources/parallel_dag_error.py b/ui/sdk/tests/resources/parallel_dag_error.py
index 9b171540..bea7f444 100644
--- a/ui/sdk/tests/resources/parallel_dag_error.py
+++ b/ui/sdk/tests/resources/parallel_dag_error.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import dataclasses
import os
from typing import List
diff --git a/ui/sdk/tests/test_adapters.py b/ui/sdk/tests/test_adapters.py
index cac37949..65d439fb 100644
--- a/ui/sdk/tests/test_adapters.py
+++ b/ui/sdk/tests/test_adapters.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import os.path
import pytest
diff --git a/ui/sdk/tests/test_driver.py b/ui/sdk/tests/test_driver.py
index 7a295099..f32180aa 100644
--- a/ui/sdk/tests/test_driver.py
+++ b/ui/sdk/tests/test_driver.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import hashlib
import logging
from types import ModuleType
diff --git a/ui/sdk/tests/test_package_to_hash/submodule1.py b/ui/sdk/tests/test_package_to_hash/submodule1.py
index 5ddf3491..11bdf850 100644
--- a/ui/sdk/tests/test_package_to_hash/submodule1.py
+++ b/ui/sdk/tests/test_package_to_hash/submodule1.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import hashlib
from . import subpackage
diff --git a/ui/sdk/tests/test_package_to_hash/submodule2.py b/ui/sdk/tests/test_package_to_hash/submodule2.py
index ecef9952..5f712724 100644
--- a/ui/sdk/tests/test_package_to_hash/submodule2.py
+++ b/ui/sdk/tests/test_package_to_hash/submodule2.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import pprint
from . import subpackage
diff --git a/ui/sdk/tests/test_tracking.py b/ui/sdk/tests/test_tracking.py
index 983f7875..7fb2cc02 100644
--- a/ui/sdk/tests/test_tracking.py
+++ b/ui/sdk/tests/test_tracking.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import functools
from datetime import datetime, timezone
from typing import Any, Callable, Dict, List
diff --git a/ui/sdk/tests/tracking/test_constants.py b/ui/sdk/tests/tracking/test_constants.py
index 9ccab2ec..3ba5db5a 100644
--- a/ui/sdk/tests/tracking/test_constants.py
+++ b/ui/sdk/tests/tracking/test_constants.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import configparser
from hamilton_sdk.tracking import constants
diff --git a/ui/sdk/tests/tracking/test_dataframe_stats.py b/ui/sdk/tests/tracking/test_dataframe_stats.py
index 8e53f214..8a996fa0 100644
--- a/ui/sdk/tests/tracking/test_dataframe_stats.py
+++ b/ui/sdk/tests/tracking/test_dataframe_stats.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import math
import numpy as np
diff --git a/ui/sdk/tests/tracking/test_ibis_stats.py b/ui/sdk/tests/tracking/test_ibis_stats.py
index b1f9808a..dd4dc6ea 100644
--- a/ui/sdk/tests/tracking/test_ibis_stats.py
+++ b/ui/sdk/tests/tracking/test_ibis_stats.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import ibis
import pandas as pd
from hamilton_sdk.tracking import ibis_stats
diff --git a/ui/sdk/tests/tracking/test_langchain_stats.py b/ui/sdk/tests/tracking/test_langchain_stats.py
index 34eee23a..dfcd3f08 100644
--- a/ui/sdk/tests/tracking/test_langchain_stats.py
+++ b/ui/sdk/tests/tracking/test_langchain_stats.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
from hamilton_sdk.tracking import langchain_stats
from langchain_core import documents as lc_documents
from langchain_core import messages as lc_messages
diff --git a/ui/sdk/tests/tracking/test_pandas_col_stats.py b/ui/sdk/tests/tracking/test_pandas_col_stats.py
index 550c736e..684db46a 100644
--- a/ui/sdk/tests/tracking/test_pandas_col_stats.py
+++ b/ui/sdk/tests/tracking/test_pandas_col_stats.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
"""Module for testing pandas column stats."""
import pandas as pd
diff --git a/ui/sdk/tests/tracking/test_pandas_stats.py b/ui/sdk/tests/tracking/test_pandas_stats.py
index da490301..2eb2721b 100644
--- a/ui/sdk/tests/tracking/test_pandas_stats.py
+++ b/ui/sdk/tests/tracking/test_pandas_stats.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import pandas as pd
from hamilton_sdk.tracking import pandas_stats as ps
diff --git a/ui/sdk/tests/tracking/test_polars_col_stats.py b/ui/sdk/tests/tracking/test_polars_col_stats.py
index 3f3a976a..f80c12b8 100644
--- a/ui/sdk/tests/tracking/test_polars_col_stats.py
+++ b/ui/sdk/tests/tracking/test_polars_col_stats.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
"""Module for testing pandas column stats."""
import polars as pl
diff --git a/ui/sdk/tests/tracking/test_polars_stats.py b/ui/sdk/tests/tracking/test_polars_stats.py
index a8a9d43a..141e4d90 100644
--- a/ui/sdk/tests/tracking/test_polars_stats.py
+++ b/ui/sdk/tests/tracking/test_polars_stats.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
from datetime import date
import polars as pl
from hamilton_sdk.tracking import polars_stats as ps
diff --git a/ui/sdk/tests/tracking/test_pydantic_stats.py b/ui/sdk/tests/tracking/test_pydantic_stats.py
index f6703a39..5059a840 100644
--- a/ui/sdk/tests/tracking/test_pydantic_stats.py
+++ b/ui/sdk/tests/tracking/test_pydantic_stats.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
from hamilton_sdk.tracking import pydantic_stats
from pydantic import BaseModel
diff --git a/ui/sdk/tests/tracking/test_pyspark_stats.py b/ui/sdk/tests/tracking/test_pyspark_stats.py
index 694afc34..2e22c66d 100644
--- a/ui/sdk/tests/tracking/test_pyspark_stats.py
+++ b/ui/sdk/tests/tracking/test_pyspark_stats.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import pandas as pd
import pyspark.sql as ps
from hamilton_sdk.tracking import pyspark_stats
diff --git a/ui/sdk/tests/tracking/test_runs.py b/ui/sdk/tests/tracking/test_runs.py
index 7feee0b4..ee615b78 100644
--- a/ui/sdk/tests/tracking/test_runs.py
+++ b/ui/sdk/tests/tracking/test_runs.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import json
from datetime import date
from typing import Any
diff --git a/ui/sdk/tests/tracking/test_stats.py b/ui/sdk/tests/tracking/test_stats.py
index 7d030090..fbe634d0 100644
--- a/ui/sdk/tests/tracking/test_stats.py
+++ b/ui/sdk/tests/tracking/test_stats.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import sys
from collections import namedtuple
from typing import NamedTuple
diff --git a/ui/sdk/tests/tracking/test_utils.py b/ui/sdk/tests/tracking/test_utils.py
index 62cafb87..9a8951ab 100644
--- a/ui/sdk/tests/tracking/test_utils.py
+++ b/ui/sdk/tests/tracking/test_utils.py
@@ -1,3 +1,20 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
import dataclasses
import datetime
from enum import Enum