Source code for geowatch.monkey.monkey_torch

def fix_gelu_issue(method):
    # Torch 1.12 added an approximate parameter that our old models dont
    # have. Monkey patch it in.
    # https://github.com/pytorch/pytorch/pull/61439
    for name, mod in method.named_modules():
        if mod.__class__.__name__ == 'GELU':
            if not hasattr(mod, 'approximate'):
                mod.approximate = 'none'





def fix_package_modules():
    # Monkey Patch torch.package
    import sys
    if sys.version_info[0:2] >= (3, 10):
        try:
            from torch.package import _stdlib
            _stdlib._get_stdlib_modules = lambda: sys.stdlib_module_names
        except Exception:
            pass