57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
# Copyright 2024 Bytedance Ltd. and/or its affiliates
|
|
#
|
|
# Licensed 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.
|
|
"""
|
|
Contain small python utility functions
|
|
"""
|
|
|
|
from typing import Dict
|
|
from types import SimpleNamespace
|
|
|
|
|
|
def union_two_dict(dict1: Dict, dict2: Dict):
|
|
"""Union two dict. Will throw an error if there is an item not the same object with the same key.
|
|
|
|
Args:
|
|
dict1:
|
|
dict2:
|
|
|
|
Returns:
|
|
|
|
"""
|
|
for key, val in dict2.items():
|
|
if key in dict1:
|
|
assert dict2[key] == dict1[key], \
|
|
f'{key} in meta_dict1 and meta_dict2 are not the same object'
|
|
dict1[key] = val
|
|
|
|
return dict1
|
|
|
|
|
|
def append_to_dict(data: Dict, new_data: Dict):
|
|
for key, val in new_data.items():
|
|
if key not in data:
|
|
data[key] = []
|
|
data[key].append(val)
|
|
|
|
|
|
class NestedNamespace(SimpleNamespace):
|
|
|
|
def __init__(self, dictionary, **kwargs):
|
|
super().__init__(**kwargs)
|
|
for key, value in dictionary.items():
|
|
if isinstance(value, dict):
|
|
self.__setattr__(key, NestedNamespace(value))
|
|
else:
|
|
self.__setattr__(key, value)
|