Skip to content

omnipy.components.prefect.lazy_import

FUNCTION DESCRIPTION
set_prefect_config_path
use_local_api_for_tests

set_prefect_config_path 

set_prefect_config_path()
Source code in src/omnipy/components/prefect/lazy_import.py 
 7
 8
 9
10
11
def set_prefect_config_path():
    prefect_module_dir = os.path.dirname(inspect.getabsfile(inspect.currentframe()))

    os.environ['PREFECT_LOGGING_SETTINGS_PATH'] = \
        f"{Path(prefect_module_dir).joinpath('settings', 'logging.yml')}"

use_local_api_for_tests 

use_local_api_for_tests()
Source code in src/omnipy/components/prefect/lazy_import.py 
14
15
16
17
def use_local_api_for_tests():
    if 'pytest' in sys.modules:
        os.environ['PREFECT_API_KEY'] = ''
        os.environ['PREFECT_API_URL'] = ''