def find_class()

in src/neo_loader/xgboost_model_loader.py [0:0]

• 8 lines of code
• 6 McCabe index (conditional complexity)

    def find_class(self, module, name):
        if module in ['builtins', '__builtin__'] and name in SAFE_BUILTINS:
            return getattr(builtins, name)
        if module in ['copy_reg', 'copyreg'] and name in SAFE_COPYREG:
            return getattr(copyreg, name)
        if module == 'xgboost.core':
            return getattr(xgboost.core, name)
        raise pickle.UnpicklingError("Global '{}.{}' is forbidden".format(module, name))