setup_utils.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# 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.
#

import inspect
import os
import sys


def add_utils_to_sys_path():
    """Add the sibling utils/ directory to sys.path so shared modules can be imported.

    Uses inspect.stack() to resolve the calling script's location from the bytecode,
    which works both in standard execution and in Databricks exec(compile(...)) contexts
    where __file__ is not defined.
    """
    caller_file = inspect.stack()[1].filename
    parent_dir = os.path.abspath(os.path.join(os.path.dirname(caller_file), '..'))
    utils_dir = os.path.join(parent_dir, 'utils')
    if utils_dir not in sys.path:
        sys.path.insert(0, utils_dir)