Function get_data_dir

axlearn/common/utils.py:1559–1562

()

        # Use type() so that if in_tree is a VDict, out_tree is also a VDict.
        out_tree = type(in_tree)()
        for k, v in in_tree.items():
            path = _concat(prefix=prefix, suffix=k, separator=separator)
            v = prune_tree(v, should_prune, prefix=path, separator=separator)
            if not should_prune(path, v):
                out_tree[k] = v
        in_tree = out_tree
    return in_tree