a186e2aa2857d92d42ced706f09a59f73629d391,wandb/sdk/wandb_init.py,_WandbInit,setup,#_WandbInit#Any#,58

Before Change

                " or config=wandb.helper.parse_config(config_object, exclude=("key",))"
            )

        if config_exclude_keys and config_include_keys:
            raise UsageError(
                "Expected at most only one of config_exclude_keys or "
                "config_include_keys"
            )
        init_config = parse_config(
            init_config, include=config_include_keys, exclude=config_exclude_keys
        )
        if config_include_keys:
            init_config = {
                key: value
                for key, value in iteritems(init_config)
                if key in config_include_keys
            }
        if config_exclude_keys:
            init_config = {
                key: value
                for key, value in iteritems(init_config)
                if key not in config_exclude_keys
            }

        // merge config with sweep (or config file)
        self.config = self._wl._config or dict()
        for k, v in init_config.items():

After Change

        )

        sm_config = sagemaker.parse_sm_config()
        if sm_config:
            sm_api_key = sm_config.get("wandb_api_key", None)
            sm_run, sm_env = sagemaker.parse_sm_resources()
            if sm_env:
                if sm_api_key:
                    sm_env["WANDB_API_KEY"] = sm_api_key
                settings._apply_environ(sm_env)
            for k, v in six.iteritems(sm_run):
                kwargs.setdefault(k, v)

        // Remove parameters that are not part of settings
        init_config = kwargs.pop("config", None) or dict()
        config_include_keys = kwargs.pop("config_include_keys", None)
        config_exclude_keys = kwargs.pop("config_exclude_keys", None)

Instances