class Api
Used for querying the W&B server.
Examples:
method Api.__init__
overrides
: You can setbase_url
if you areusing a W&B server other than
https: //api.wandb.ai
. You can also set defaults forentity
,project
, andrun
.timeout
: HTTP timeout in seconds for API requests. If not specified, the default timeout will be used.api_key
: API key to use for authentication. If not provided, the API key from the current environment or configuration will be used.
property Api.api_key
Returns W&B API key. Returns:str | None
: The api_key property value.
property Api.client
Returns the client object. Returns:RetryingClient
: The client property value.
property Api.default_entity
Returns the default W&B entity. Returns:str | None
: The default_entity property value.
property Api.user_agent
Returns W&B public user agent. Returns:str
: The user_agent property value.
property Api.viewer
Returns the viewer object. Raises:ValueError
: If viewer data is not able to be fetched from W&B.requests.RequestException
: If an error occurs while making the graphql request.
public.User
: The viewer property value.
method Api.artifact
name
: The artifact’s name. The name of an artifact resembles a filepath that consists, at a minimum, the name of the project the artifact was logged to, the name of the artifact, and the artifact’s version or alias. Optionally append the entity that logged the artifact as a prefix followed by a forward slash. If no entity is specified in the name, the Run or API setting’s entity is used.type
: The type of artifact to fetch.
Artifact
object.
Raises:
ValueError
: If the artifact name is not specified.ValueError
: If the artifact type is specified but does not match the type of the fetched artifact.
This method is intended for external use only. Do not call api.artifact()
within the wandb repository code.
method Api.artifact_collection
ArtifactCollection
object to retrieve information about specific artifacts in that collection, and more.
Args:
type_name
: The type of artifact collection to fetch.name
: An artifact collection name. Optionally append the entity that logged the artifact as a prefix followed by a forward slash.
ArtifactCollection
object.
Examples:
In the proceeding code snippet “type”, “entity”, “project”, and “artifact_name” are placeholders for the collection type, your W&B entity, name of the project the artifact is in, and the name of the artifact, respectively.
method Api.artifact_collection_exists
name
: An artifact collection name. Optionally append the entity that logged the artifact as a prefix followed by a forward slash. If entity or project is not specified, infer the collection from the override params if they exist. Otherwise, entity is pulled from the user settings and project will default to “uncategorized”.type
: The type of artifact collection.
method Api.artifact_collections
project_name
: The name of the project to filter on.type_name
: The name of the artifact type to filter on.per_page
: Sets the page size for query pagination. None will use the default size. Usually there is no reason to change this.
ArtifactCollections
object.
method Api.artifact_exists
name
: The name of artifact. Add the artifact’s entity and project as a prefix. Append the version or the alias of the artifact with a colon. If the entity or project is not specified, W&B uses override parameters if populated. Otherwise, the entity is pulled from the user settings and the project is set to “Uncategorized”.type
: The type of artifact.
method Api.artifact_type
ArtifactType
.
Args:
type_name
: The name of the artifact type to retrieve.project
: If given, a project name or path to filter on.
ArtifactType
object.
method Api.artifact_types
project
: The project name or path to filter on.
ArtifactTypes
object.
method Api.artifact_versions
Api.artifacts(type_name, name)
method instead.
method Api.artifacts
Artifacts
collection.
Args:
type_name: The type of artifacts to fetch. name: The artifact’s collection name. Optionally append the entity that logged the artifact as a prefix followed by a forward slash. per_page: Sets the page size for query pagination. If set to None
, use the default size. Usually there is no reason to change this. tags: Only return artifacts with all of these tags.
Returns:
An iterable Artifacts
object.
Examples:
In the proceeding code snippet, “type”, “entity”, “project”, and “artifact_name” are placeholders for the artifact type, W&B entity, name of the project the artifact was logged to, and the name of the artifact, respectively.
method Api.automation
name
: The name of the automation to fetch.entity
: The entity to fetch the automation for.
ValueError
: If zero or multiple Automations match the search criteria.
method Api.automations
entity
: The entity to fetch the automations for.name
: The name of the automation to fetch.per_page
: The number of automations to fetch per page. Defaults to 50. Usually there is no reason to change this.
method Api.create_automation
name
: The name of the automation.description
: The description of the automation.enabled
: Whether the automation is enabled.scope
: The scope of the automation.event
: The event that triggers the automation.action
: The action that is triggered by the automation.
method Api.create_custom_chart
entity
: The entity (user or team) that owns the chartname
: Unique identifier for the chart presetdisplay_name
: Human-readable name shown in the UIspec_type
: Type of specification. Must be “vega2” for Vega-Lite v2 specifications.access
: Access level for the chart:- “private”: Chart is only accessible to the entity that created it
- “public”: Chart is publicly accessible
spec
: The Vega/Vega-Lite specification as a dictionary or JSON string
wandb.Error
: If chart creation failsUnsupportedError
: If the server doesn’t support custom charts
method Api.create_project
name
: The name of the new project.entity
: The entity of the new project.
method Api.create_registry
name
: The name of the registry. Name must be unique within the organization.visibility
: The visibility of the registry.organization
: Anyone in the organization can view this registry. You can edit their roles later from the settings in the UI.restricted
: Only invited members via the UI can access this registry. Public sharing is disabled.organization
: The organization of the registry. If no organization is set in the settings, the organization will be fetched from the entity if the entity only belongs to one organization.description
: The description of the registry.artifact_types
: The accepted artifact types of the registry. A type is nomore than 128 characters and do not include characters
/or ``:
. If not specified, all types are accepted. Allowed types added to the registry cannot be removed later.
method Api.create_run
run_id
: The ID to assign to the run. If not specified, W&B creates a random ID.project
: The project where to log the run to. If no project is specified, log the run to a project called “Uncategorized”.entity
: The entity that owns the project. If no entity is specified, log the run to the default entity.
Run
.
method Api.create_run_queue
name
: Name of the queue to createtype
: Type of resource to be used for the queue. One of “local-container”, “local-process”, “kubernetes”,“sagemaker”, or “gcp-vertex”.entity
: Name of the entity to create the queue. IfNone
, use the configured or default entity.prioritization_mode
: Version of prioritization to use. Either “V0” orNone
.config
: Default resource configuration to be used for the queue. Use handlebars (eg.{{var}}
) to specify template variables.template_variables
: A dictionary of template variable schemas to use with the config.
RunQueue
.
Raises:
ValueError
if any of the parameters are invalid wandb.Error
on wandb API errors
method Api.create_team
team
: The name of the teamadmin_username
: Username of the admin user of the team. Defaults to the current user.
Team
object.
method Api.create_user
email
: The email address of the user.admin
: Set user as a global instance administrator.
User
object.
method Api.delete_automation
obj
: The automation to delete, or its ID.
method Api.flush
api.flush()
to get the latest values associated with the run.
method Api.from_path
path
: The path to the project, run, sweep or report
Project
, Run
, Sweep
, or BetaReport
instance.
Raises:
wandb.Error
if path is invalid or the object doesn’t exist.
Examples:
In the proceeding code snippets “project”, “team”, “run_id”, “sweep_id”, and “report_name” are placeholders for the project, team, run ID, sweep ID, and the name of a specific report, respectively.
method Api.integrations
entity
: The entity (e.g. team name) for which to fetch integrations. If not provided, the user’s default entity will be used.per_page
: Number of integrations to fetch per page. Defaults to 50. Usually there is no reason to change this.
Iterator[SlackIntegration | WebhookIntegration]
: An iterator of any supported integrations.
method Api.job
Job
object.
Args:
name
: The name of the job.path
: The root path to download the job artifact.
Job
object.
method Api.list_jobs
entity
: The entity for the listed jobs.project
: The project for the listed jobs.
method Api.project
Project
with the given name (and entity, if given).
Args:
name
: The project name.entity
: Name of the entity requested. If None, will fall back to the default entity passed toApi
. If no default entity, will raise aValueError
.
Project
object.
method Api.projects
entity
: Name of the entity requested. If None, will fall back to the default entity passed toApi
. If no default entity, will raise aValueError
.per_page
: Sets the page size for query pagination. If set toNone
, use the default size. Usually there is no reason to change this.
Projects
object which is an iterable collection of Project
objects.
method Api.queued_run
entity/project/queue_id/run_queue_item_id
.
method Api.registries
Registry
objects.
Use the iterator to search and filter registries, collections, or artifact versions across your organization’s registry.
Args:
organization
: (str, optional) The organization of the registry to fetch. If not specified, use the organization specified in the user’s settings.filter
: (dict, optional) MongoDB-style filter to apply to each object in the lazy registry iterator. Fields available to filter for registries arename
,description
,created_at
,updated_at
. Fields available to filter for collections arename
,tag
,description
,created_at
,updated_at
Fields available to filter for versions aretag
,alias
,created_at
,updated_at
,metadata
Registry
objects.
Examples:
Find all registries with the names that contain “model”
method Api.registry
name
: The name of the registry. This is without thewandb-registry-
prefix.organization
: The organization of the registry. If no organization is set in the settings, the organization will be fetched from the entity if the entity only belongs to one organization.
method Api.reports
wandb.Api.reports()
API is in beta and will likely change in future releases.
Args:
path
: The path to the project the report resides in. Specify the entity that created the project as a prefix followed by a forward slash.name
: Name of the report requested.per_page
: Sets the page size for query pagination. If set toNone
, use the default size. Usually there is no reason to change this.
Reports
object which is an iterable collection of BetaReport
objects.
Examples:
method Api.run
entity/project/run_id
.
Args:
path
: Path to run in the formentity/project/run_id
. Ifapi.entity
is set, this can be in the formproject/run_id
and ifapi.project
is set this can just be the run_id.
Run
object.
method Api.run_queue
RunQueue
for entity.
See Api.create_run_queue
for more information on how to create a run queue.
method Api.runs
Runs
object, which lazily iterates over Run
objects.
Fields you can filter by include:
createdAt
: The timestamp when the run was created. (in ISO 8601 format, e.g. “2023-01-01T12:00:00Z”)displayName
: The human-readable display name of the run. (e.g. “eager-fox-1”)duration
: The total runtime of the run in seconds.group
: The group name used to organize related runs together.host
: The hostname where the run was executed.jobType
: The type of job or purpose of the run.name
: The unique identifier of the run. (e.g. “a1b2cdef”)state
: The current state of the run.tags
: The tags associated with the run.username
: The username of the user who initiated the run
config.experiment_name
, summary_metrics.loss
, etc.
For more complex filtering, you can use MongoDB query operators. For details, see: https://docs.mongodb.com/manual/reference/operator/query The following operations are supported:
$and
$or
$nor
$eq
$ne
$gt
$gte
$lt
$lte
$in
$nin
$exists
$regex
path
: (str) path to project, should be in the form: “entity/project”filters
: (dict) queries for specific runs using the MongoDB query language. You can filter by run properties such as config.key, summary_metrics.key, state, entity, createdAt, etc.For example
:{"config.experiment_name": "foo"}
would find runs with a config entry of experiment name set to “foo”order
: (str) Order can becreated_at
,heartbeat_at
,config.*.value
, orsummary_metrics.*
. If you prepend order with a + order is ascending (default). If you prepend order with a - order is descending. The default order is run.created_at from oldest to newest.per_page
: (int) Sets the page size for query pagination.include_sweeps
: (bool) Whether to include the sweep runs in the results.
Runs
object, which is an iterable collection of Run
objects.
Examples:
method Api.slack_integrations
entity
: The entity (e.g. team name) for which to fetch integrations. If not provided, the user’s default entity will be used.per_page
: Number of integrations to fetch per page. Defaults to 50. Usually there is no reason to change this.
Iterator[SlackIntegration]
: An iterator of Slack integrations.
method Api.sweep
entity/project/sweep_id
.
Args:
path
: Path to sweep in the form entity/project/sweep_id. Ifapi.entity
is set, this can be in the form project/sweep_id and ifapi.project
is set this can just be the sweep_id.
Sweep
object.
method Api.sync_tensorboard
method Api.team
Team
with the given name.
Args:
team
: The name of the team.
Team
object.
method Api.update_automation
obj
: The automation to update. Must be an existing automation. create_missing (bool): If True, and the automation does not exist, create it. **kwargs: Any additional values to assign to the automation before updating it. If given, these will override any values that may already be set on the automation:name
: The name of the automation.description
: The description of the automation.enabled
: Whether the automation is enabled.scope
: The scope of the automation.event
: The event that triggers the automation.action
: The action that is triggered by the automation.
method Api.upsert_run_queue
name
: Name of the queue to createentity
: Optional name of the entity to create the queue. IfNone
, use the configured or default entity.resource_config
: Optional default resource configuration to be used for the queue. Use handlebars (eg.{{var}}
) to specify template variables.resource_type
: Type of resource to be used for the queue. One of “local-container”, “local-process”, “kubernetes”, “sagemaker”, or “gcp-vertex”.template_variables
: A dictionary of template variable schemas to be used with the config.external_links
: Optional dictionary of external links to be used with the queue.prioritization_mode
: Optional version of prioritization to use. Either “V0” or None
RunQueue
.
Raises:
ValueError if any of the parameters are invalid wandb.Error on wandb API errors
method Api.user
api.viewer
to get your own user object.
Args:
username_or_email
: The username or email address of the user.
User
object or None if a user is not found.
method Api.users
api.viewer
to get your own user object.
Args:
username_or_email
: The prefix or suffix of the user you want to find.
User
objects.
method Api.webhook_integrations
entity
: The entity (e.g. team name) for which to fetch integrations. If not provided, the user’s default entity will be used.per_page
: Number of integrations to fetch per page. Defaults to 50. Usually there is no reason to change this.
Iterator[WebhookIntegration]
: An iterator of webhook integrations.