def unset_globals(): wandb.run = None wandb.config = preinit.PreInitObject("wandb.config") wandb.summary = preinit.PreInitObject("wandb.summary") wandb.log = preinit.PreInitCallable("wandb.log", wandb.wandb_sdk.wandb_run.Run.log) wandb.save = preinit.PreInitCallable("wandb.save", wandb.wandb_sdk.wandb_run.Run.save) wandb.restore = preinit.PreInitCallable( "wandb.restore", wandb.wandb_sdk.wandb_run.Run.restore) wandb.use_artifact = preinit.PreInitCallable( "wandb.use_artifact", wandb.wandb_sdk.wandb_run.Run.use_artifact) wandb.log_artifact = preinit.PreInitCallable( "wandb.log_artifact", wandb.wandb_sdk.wandb_run.Run.log_artifact)
from wandb.lib.ipython import _get_python_type # toplevel: # save() # restore() # login() # sweep() # agent() # globals Api = PublicApi api = InternalApi() run = None config = _preinit.PreInitObject("wandb.config") summary = _preinit.PreInitObject("wandb.summary") log = _preinit.PreInitCallable("wandb.log", wandb_sdk.wandb_run.Run.log) save = _preinit.PreInitCallable("wandb.save", wandb_sdk.wandb_run.Run.save) restore = _preinit.PreInitCallable("wandb.restore", wandb_sdk.wandb_run.Run.restore) use_artifact = _preinit.PreInitCallable("wandb.use_artifact", wandb_sdk.wandb_run.Run.use_artifact) log_artifact = _preinit.PreInitCallable("wandb.log_artifact", wandb_sdk.wandb_run.Run.log_artifact) # record of patched libraries patched = {"tensorboard": [], "keras": [], "gym": []} keras = _lazyloader.LazyLoader("wandb.keras", globals(), "wandb.integration.keras") sklearn = _lazyloader.LazyLoader("wandb.sklearn", globals(), "wandb.sklearn")