Return the structure of a possibly nested dictionary object without all the values

scripts.py

import typing


def return_dict_structure(obj):
    """Return the structure of a possibly nested dictionary object."""
    new_obj = {}
    if isinstance(obj, typing.List) or isinstance(obj, typing.Tuple):
        if obj:
            return [return_dict_structure(obj[0]), '...']
        else:
            return []
    elif isinstance(obj, typing.Dict):
        for k, v in obj.items():
            new_obj[k] = return_dict_structure(v)
    else:
        return type(obj)
    return new_obj


# Train test validation stratified split
from sklearn.model_selection import train_test_split
def split_ds(X, y, val_test_size=0.3):
    X_train, X_test_val, y_train, y_test_val = train_test_split(X, y, test_size=val_test_size, random_state=32, stratify=y)
    X_test, X_val, y_test, y_val = train_test_split(X_test_val, y_test_val, test_size=0.5, random_state=32, stratify=y_test_val)
    return X_train, X_val, X_test, y_train, y_val, y_test