Beispiel #1
0
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)
Beispiel #2
0

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")