wandb 0.17.6__py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

package_readme.md

@@ -0,0 +1,95 @@
+<div align="center">
+  <img src="https://i.imgur.com/RUtiVzH.png" width="600" /><br><br>
+</div>
+
+# Weights and Biases [![PyPI](https://img.shields.io/pypi/v/wandb)](https://pypi.python.org/pypi/wandb) [![Conda (channel only)](https://img.shields.io/conda/vn/conda-forge/wandb)](https://anaconda.org/conda-forge/wandb) [![CircleCI](https://img.shields.io/circleci/build/github/wandb/wandb/main)](https://circleci.com/gh/wandb/wandb) [![Codecov](https://img.shields.io/codecov/c/gh/wandb/wandb)](https://codecov.io/gh/wandb/wandb)
+
+Use W&B to build better models faster. Track and visualize all the pieces of your machine learning pipeline, from datasets to production models.
+
+- Quickly identify model regressions. Use W&B to visualize results in real time, all in a central dashboard.
+- Focus on the interesting ML. Spend less time manually tracking results in spreadsheets and text files.
+- Capture dataset versions with W&B Artifacts to identify how changing data affects your resulting models.
+- Reproduce any model, with saved code, hyperparameters, launch commands, input data, and resulting model weights.
+
+[Sign up for a free account →](https://wandb.ai/login?signup=true)
+
+## Features
+
+-   Store hyper-parameters used in a training run
+-   Search, compare, and visualize training runs
+-   Analyze system usage metrics alongside runs
+-   Collaborate with team members
+-   Replicate historic results
+-   Run parameter sweeps
+-   Keep records of experiments available forever
+
+[Documentation →](https://docs.wandb.com)
+
+## Quickstart
+
+```shell
+pip install wandb
+```
+
+In your training script:
+
+```python
+import wandb
+
+# Your custom arguments defined here
+args = ...
+
+wandb.init(config=args, project="my-project")
+wandb.config["more"] = "custom"
+
+
+def training_loop():
+    while True:
+        # Do some machine learning
+        epoch, loss, val_loss = ...
+        # Framework agnostic / custom metrics
+        wandb.log({"epoch": epoch, "loss": loss, "val_loss": val_loss})
+```
+
+If you're already using Tensorboard or [TensorboardX](https://github.com/lanpa/tensorboardX), you can integrate with one line:
+
+```python
+wandb.init(sync_tensorboard=True)
+```
+
+## Running your script
+
+Run `wandb login` from your terminal to signup or authenticate your machine (we store your api key in ~/.netrc). You can also set the `WANDB_API_KEY` environment variable with a key from your [settings](https://app.wandb.ai/settings).
+
+Run your script with `python my_script.py` and all metadata will be synced to the cloud. You will see a url in your terminal logs when your script starts and finishes. Data is staged locally in a directory named _wandb_ relative to your script. If you want to test your script without syncing to the cloud you can set the environment variable `WANDB_MODE=dryrun`.
+
+If you are using [docker](https://docker.com) to run your code, we provide a wrapper command `wandb docker` that mounts your current directory, sets environment variables, and ensures the wandb library is installed. Training your models in docker gives you the ability to restore the exact code and environment with the `wandb restore` command.
+
+## Web Interface
+
+[Sign up for a free account →](https://wandb.com)
+[![Watch the video](https://i.imgur.com/PW0Ejlc.png)](https://youtu.be/EeqhOSvNX-A)
+[Introduction video →](https://youtu.be/EeqhOSvNX-A)
+
+## Detailed Usage
+
+Framework specific and detailed usage can be found in our [documentation](http://docs.wandb.com/).
+
+## Testing
+
+To run basic test use `make test`.  More detailed information can be found at CONTRIBUTING.md.
+
+We use [circleci](https://circleci.com) for CI.
+
+# Academic Researchers
+If you'd like a free academic account for your research group, [reach out to us →](https://www.wandb.com/academic)
+
+We make it easy to cite W&B in your published paper. [Learn more →](https://www.wandb.com/academic)
+[![](https://i.imgur.com/loKLiez.png)](https://www.wandb.com/academic)
+
+## Community
+Got questions, feedback or want to join a community of ML engineers working on exciting projects?
+
+<a href="https://bit.ly/wb-slack"><img src="https://svgshare.com/i/M93.svg" alt="slack" width="55"/></a> Join our [slack](https://bit.ly/wb-slack) community.
+
+[![Twitter](https://img.shields.io/twitter/follow/weights_biases?style=social)](https://twitter.com/weights_biases) Follow us on [Twitter](https://twitter.com/weights_biases).

wandb/__init__.py

@@ -0,0 +1,257 @@
+"""Use wandb to track machine learning work.
+
+The most commonly used functions/objects are:
+  - wandb.init — initialize a new run at the top of your training script
+  - wandb.config — track hyperparameters and metadata
+  - wandb.log — log metrics and media over time within your training loop
+
+For guides and examples, see https://docs.wandb.ai.
+
+For scripts and interactive notebooks, see https://github.com/wandb/examples.
+
+For reference documentation, see https://docs.wandb.com/ref/python.
+"""
+__version__ = "0.17.6"
+
+
+# Used with pypi checks and other messages related to pip
+_wandb_module = "wandb"
+
+from typing import Optional
+
+from wandb.errors import Error
+
+# This needs to be early as other modules call it.
+from wandb.errors.term import termsetup, termlog, termerror, termwarn
+
+from wandb import sdk as wandb_sdk
+
+import wandb
+
+wandb.wandb_lib = wandb_sdk.lib  # type: ignore
+
+init = wandb_sdk.init
+setup = wandb_sdk.setup
+_attach = wandb_sdk._attach
+_sync = wandb_sdk._sync
+_teardown = wandb_sdk.teardown
+watch = wandb_sdk.watch
+unwatch = wandb_sdk.unwatch
+finish = wandb_sdk.finish
+join = finish
+login = wandb_sdk.login
+helper = wandb_sdk.helper
+sweep = wandb_sdk.sweep
+controller = wandb_sdk.controller
+require = wandb_sdk.require
+Artifact = wandb_sdk.Artifact
+AlertLevel = wandb_sdk.AlertLevel
+Settings = wandb_sdk.Settings
+Config = wandb_sdk.Config
+
+from wandb.apis import InternalApi, PublicApi
+from wandb.errors import CommError, UsageError
+
+_preinit = wandb.wandb_lib.preinit  # type: ignore
+_lazyloader = wandb.wandb_lib.lazyloader  # type: ignore
+
+# Call import module hook to set up any needed require hooks
+wandb.sdk.wandb_require._import_module_hook()
+
+from wandb import wandb_torch
+
+# Move this (keras.__init__ expects it at top level)
+from wandb.sdk.data_types._private import _cleanup_media_tmp_dir
+
+_cleanup_media_tmp_dir()
+
+from wandb.data_types import Graph
+from wandb.data_types import Image
+from wandb.data_types import Plotly
+
+# from wandb.data_types import Bokeh # keeping out of top level for now since Bokeh plots have poor UI
+from wandb.data_types import Video
+from wandb.data_types import Audio
+from wandb.data_types import Table
+from wandb.data_types import Html
+from wandb.data_types import box3d
+from wandb.data_types import Object3D
+from wandb.data_types import Molecule
+from wandb.data_types import Histogram
+from wandb.data_types import Classes
+from wandb.data_types import JoinedTable
+
+from wandb.wandb_agent import agent
+
+# from wandb.core import *
+from wandb.viz import visualize
+from wandb import plot
+from wandb.integration.sagemaker import sagemaker_auth
+from wandb.sdk.internal import profiler
+
+# Artifact import types
+from wandb.sdk.artifacts.artifact_ttl import ArtifactTTL
+
+# Used to make sure we don't use some code in the incorrect process context
+_IS_INTERNAL_PROCESS = False
+
+
+def _set_internal_process(disable=False):
+    global _IS_INTERNAL_PROCESS
+    if _IS_INTERNAL_PROCESS is None:
+        return
+    if disable:
+        _IS_INTERNAL_PROCESS = None
+        return
+    _IS_INTERNAL_PROCESS = True
+
+
+def _assert_is_internal_process():
+    if _IS_INTERNAL_PROCESS is None:
+        return
+    assert _IS_INTERNAL_PROCESS
+
+
+def _assert_is_user_process():
+    if _IS_INTERNAL_PROCESS is None:
+        return
+    assert not _IS_INTERNAL_PROCESS
+
+
+# toplevel:
+# save()
+# restore()
+# login()
+# sweep()
+# agent()
+
+# globals
+Api = PublicApi
+api = InternalApi()
+run: Optional["wandb_sdk.wandb_run.Run"] = None
+config = _preinit.PreInitObject("wandb.config", wandb_sdk.wandb_config.Config)
+summary = _preinit.PreInitObject("wandb.summary", wandb_sdk.wandb_summary.Summary)
+log = _preinit.PreInitCallable("wandb.log", wandb_sdk.wandb_run.Run.log)  # type: ignore
+save = _preinit.PreInitCallable("wandb.save", wandb_sdk.wandb_run.Run.save)  # type: ignore
+restore = wandb_sdk.wandb_run.restore
+use_artifact = _preinit.PreInitCallable(
+    "wandb.use_artifact", wandb_sdk.wandb_run.Run.use_artifact  # type: ignore
+)
+log_artifact = _preinit.PreInitCallable(
+    "wandb.log_artifact", wandb_sdk.wandb_run.Run.log_artifact  # type: ignore
+)
+log_model = _preinit.PreInitCallable(
+    "wandb.log_model", wandb_sdk.wandb_run.Run.log_model  # type: ignore
+)
+use_model = _preinit.PreInitCallable(
+    "wandb.use_model", wandb_sdk.wandb_run.Run.use_model  # type: ignore
+)
+link_model = _preinit.PreInitCallable(
+    "wandb.link_model", wandb_sdk.wandb_run.Run.link_model  # type: ignore
+)
+define_metric = _preinit.PreInitCallable(
+    "wandb.define_metric", wandb_sdk.wandb_run.Run.define_metric  # type: ignore
+)
+
+mark_preempting = _preinit.PreInitCallable(
+    "wandb.mark_preempting", wandb_sdk.wandb_run.Run.mark_preempting  # type: ignore
+)
+
+plot_table = _preinit.PreInitCallable(
+    "wandb.plot_table", wandb_sdk.wandb_run.Run.plot_table
+)
+alert = _preinit.PreInitCallable("wandb.alert", wandb_sdk.wandb_run.Run.alert)  # type: ignore
+
+# record of patched libraries
+patched = {"tensorboard": [], "keras": [], "gym": []}  # type: ignore
+
+keras = _lazyloader.LazyLoader("wandb.keras", globals(), "wandb.integration.keras")
+sklearn = _lazyloader.LazyLoader("wandb.sklearn", globals(), "wandb.sklearn")
+tensorflow = _lazyloader.LazyLoader(
+    "wandb.tensorflow", globals(), "wandb.integration.tensorflow"
+)
+xgboost = _lazyloader.LazyLoader(
+    "wandb.xgboost", globals(), "wandb.integration.xgboost"
+)
+catboost = _lazyloader.LazyLoader(
+    "wandb.catboost", globals(), "wandb.integration.catboost"
+)
+tensorboard = _lazyloader.LazyLoader(
+    "wandb.tensorboard", globals(), "wandb.integration.tensorboard"
+)
+gym = _lazyloader.LazyLoader("wandb.gym", globals(), "wandb.integration.gym")
+lightgbm = _lazyloader.LazyLoader(
+    "wandb.lightgbm", globals(), "wandb.integration.lightgbm"
+)
+docker = _lazyloader.LazyLoader("wandb.docker", globals(), "wandb.docker")
+jupyter = _lazyloader.LazyLoader("wandb.jupyter", globals(), "wandb.jupyter")
+sacred = _lazyloader.LazyLoader("wandb.sacred", globals(), "wandb.integration.sacred")
+
+
+def ensure_configured():
+    global api
+    api = InternalApi()
+
+
+def set_trace():
+    import pdb  # TODO: support other debuggers
+
+    #  frame = sys._getframe().f_back
+    pdb.set_trace()  # TODO: pass the parent stack...
+
+
+def load_ipython_extension(ipython):
+    ipython.register_magics(wandb.jupyter.WandBMagics)
+
+
+if wandb_sdk.lib.ipython.in_notebook():
+    from IPython import get_ipython  # type: ignore[import-not-found]
+
+    load_ipython_extension(get_ipython())
+
+
+from .analytics import Sentry as _Sentry
+
+if "dev" in __version__:
+    import os
+
+    # disable error reporting in dev versions for the python client
+    os.environ["WANDB_ERROR_REPORTING"] = os.environ.get(
+        "WANDB_ERROR_REPORTING", "false"
+    )
+    # turn on wandb-core for dev versions
+    os.environ["WANDB__REQUIRE_CORE"] = os.environ.get("WANDB__REQUIRE_CORE", "true")
+
+_sentry = _Sentry()
+_sentry.setup()
+
+
+__all__ = (
+    "__version__",
+    "init",
+    "setup",
+    "save",
+    "sweep",
+    "controller",
+    "agent",
+    "config",
+    "log",
+    "summary",
+    "join",
+    "Api",
+    "Graph",
+    "Image",
+    "Plotly",
+    "Video",
+    "Audio",
+    "Table",
+    "Html",
+    "box3d",
+    "Object3D",
+    "Molecule",
+    "Histogram",
+    "ArtifactTTL",
+    "log_model",
+    "use_model",
+    "link_model",
+)

wandb/__main__.py

@@ -0,0 +1,3 @@
+from wandb.cli import cli
+
+cli.cli(prog_name="python -m wandb")

wandb/_globals.py

@@ -0,0 +1,19 @@
+# WARNING: This is an anti-pattern file and we should avoid
+# adding to it and remove entries whenever possible. This file
+# contains global objects which need to be referenced by multiple
+# submodules. If you need a global object, seriously reconsider. This
+# file is intended to be a stop gap to help during code migrations (eg.
+# when moving to typing a module) to avoid circular references. Anything
+# added here is pure tech debt. Use with care. - Tim
+
+_glob_datatypes_callback = None
+
+
+def _datatypes_set_callback(cb):
+    global _glob_datatypes_callback
+    _glob_datatypes_callback = cb
+
+
+def _datatypes_callback(fname):
+    if _glob_datatypes_callback:
+        _glob_datatypes_callback(fname)