Skip to main content

class Api

Used for querying the W&B server. Examples:

method Api.__init__

Initialize the API. Args:
  • overrides: You can set base_url if you are
  • using a W&B server other than https: //api.wandb.ai. You can also set defaults for entity, project, and run.
  • 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. Prompts for an API key if none is provided or configured in the environment.

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.
  • WandbApiFailedError: If an error occurs while making the GraphQL request.
Returns:
  • User: The viewer property value.

method Api.artifact

Returns a single artifact. Args:
  • 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.
Returns: An 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.
Examples: In the following code snippets “entity”, “project”, “artifact”, “version”, and “alias” are placeholders for your W&B entity, name of the project the artifact is in, the name of the artifact, and artifact’s version, respectively.
Note:
This method is intended for external use only. Do not call api.artifact() within the wandb repository code.

method Api.artifact_collection

Returns a single artifact collection by type. You can use the returned 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.
Returns: An 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

Whether an artifact collection exists within a specified project and entity. Args:
  • 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.
Returns: True if the artifact collection exists, False otherwise. Examples: In the proceeding code snippet “type”, and “collection_name” refer to the type of the artifact collection and the name of the collection, respectively.

method Api.artifact_collections

Returns a collection of matching artifact collections. Args:
  • project_name: The name of the project to filter on.
  • type_name: The name of the artifact type to filter on.
  • order: Optional string to specify the order of the results. If prefixed with ’+’, sorts ascending (default). If prefixed with ’-’, sorts descending.
  • per_page: Sets the page size for query pagination. Usually there is no reason to change this.
  • start: Pagination cursor for resuming a past query, captured from a previous paginator’s .cursor attribute.
Returns: An iterable ArtifactCollections object.

method Api.artifact_exists

Whether an artifact version exists within the specified project and entity. Args:
  • 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.
Returns: True if the artifact version exists, False otherwise. Examples: In the proceeding code snippets “entity”, “project”, “artifact”, “version”, and “alias” are placeholders for your W&B entity, name of the project the artifact is in, the name of the artifact, and artifact’s version, respectively.

method Api.artifact_type

Returns the matching ArtifactType. Args:
  • type_name: The name of the artifact type to retrieve.
  • project: If given, a project name or path to filter on.
Returns: An ArtifactType object.

method Api.artifact_types

Returns a collection of matching artifact types. Args:
  • project: The project name or path to filter on.
  • start: Pagination cursor for resuming a past query, captured from a previous paginator’s .cursor attribute.
Returns: An iterable ArtifactTypes object.

method Api.artifact_versions

Deprecated. Use Api.artifacts(type_name, name) method instead.

method Api.artifacts

Return an 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.
  • order: Optional string to specify the order of the results. If prefixed with ’+’, sorts ascending (default). If prefixed with ’-’, sorts descending.
  • per_page: Sets the page size for query pagination. Usually there is no reason to change this.
  • tags: Only return artifacts with all of these tags.
  • start: Pagination cursor for resuming a past query, captured from a previous paginator’s .cursor attribute.
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.
Pause iteration and resume later from the same position by saving the paginator’s .cursor and passing it as start=:

method Api.automation

Returns the only Automation matching the parameters. Args:
  • name: The name of the automation to fetch.
  • entity: The entity to fetch the automation for.
Raises:
  • ValueError: If zero or multiple Automations match the search criteria.
Examples: Get an existing automation named “my-automation”:
Get an existing automation named “other-automation”, from the entity “my-team”:

method Api.automations

Returns an iterator over all Automations that match the given parameters. If no parameters are provided, the returned iterator will contain all Automations that the user has access to. Args:
  • 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.
  • start: Pagination cursor for resuming a past query, captured from a previous paginator’s .cursor attribute.
Returns: A list of automations. Examples: Fetch all existing automations for the entity “my-team”:

method Api.create_automation

Create a new Automation. Args: obj: The automation to create. fetch_existing: If True, and a conflicting automation already exists, attempt to fetch the existing automation instead of raising an error. **kwargs: Any additional values to assign to the automation before creating 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.
Returns: The saved Automation. Examples: Create a new automation named “my-automation” that sends a Slack notification when a run within a specific project logs a metric exceeding a custom threshold:

method Api.create_custom_chart

Create a custom chart preset and return its id. Args:
  • entity: The entity (user or team) that owns the chart
  • name: Unique identifier for the chart preset
  • display_name: Human-readable name shown in the UI
  • spec_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
Returns: The ID of the created chart preset in the format “entity/name” Raises:
  • wandb.Error: If chart creation fails
  • UnsupportedError: If the server doesn’t support custom charts
Example:

method Api.create_project

Create a new project. Args:
  • name: The name of the new project.
  • entity: The entity of the new project.

method Api.create_registry

Create a new registry. Args:
  • 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 no
  • more 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.
Returns: A registry object. Examples:

method Api.create_run

Create a new run. Args:
  • 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.
Returns: The newly created Run.

method Api.create_run_queue

Create a new run queue in W&B Launch. Args:
  • name: Name of the queue to create
  • type: 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. If None, use the configured or default entity.
  • prioritization_mode: Version of prioritization to use. Either “V0” or None.
  • 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.
Returns: The newly created RunQueue. Raises: ValueError if any of the parameters are invalid wandb.Error on wandb API errors

method Api.create_team

Create a new team. Args:
  • team: The name of the team
  • admin_username: Username of the admin user of the team. Defaults to the current user.
Returns: A Team object.

method Api.create_user

Create a new user. Args:
  • email: The email address of the user.
  • admin: Set user as a global instance administrator.
Returns: A User object.

method Api.delete_automation

Delete an automation. Args:
  • obj: The automation to delete, or its ID.
Returns: True if the automation was deleted successfully.

method Api.flush

Flush the local cache. The api object keeps a local cache of runs, so if the state of the run may change while executing your script you must clear the local cache with api.flush() to get the latest values associated with the run.

method Api.from_path

Return a run, sweep, project or report from a path. Args:
  • path: The path to the project, run, sweep or report
Returns: A 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

Return an iterator of all integrations for an entity. Args:
  • 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.
Yields:
  • Iterator[SlackIntegration | WebhookIntegration]: An iterator of any supported integrations.

method Api.job

Return a Job object. Args:
  • name: The name of the job.
  • path: The root path to download the job artifact.
Returns: A Job object.

method Api.list_jobs

Return a list of jobs, if any, for the given entity and project. Args:
  • entity: The entity for the listed jobs.
  • project: The project for the listed jobs.
Returns: A list of matching jobs.

method Api.project

Return the 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 to Api. If no default entity, will raise a ValueError.
Returns: A Project object.

method Api.projects

Get projects for a given entity. Args:
  • entity: Name of the entity requested. If None, will fall back to the default entity passed to Api. If no default entity, will raise a ValueError.
  • per_page: Sets the page size for query pagination. Usually there is no reason to change this.
Returns: A Projects object which is an iterable collection of Projectobjects.

method Api.queued_run

Return a single queued run based on the path. Parses paths of the form entity/project/queue_id/run_queue_item_id.

method Api.registries

Returns a lazy iterator of Registry objects. Use the iterator to search and filter registries, collections, or artifact versions across your organization’s registry. Args:
  • organization: The organization of the registry to fetch. If not specified, use the organization specified in the user’s settings.
  • filter: Optional MongoDB-style filter to apply to each object in the lazy registry iterator. Fields available to filter for registries are name, description, created_at, updated_at. Fields available to filter for collections are name, tag, description, created_at, updated_at Fields available to filter for versions are tag, alias, created_at, updated_at, metadata
  • order: Optional string to specify the order of the results. If prefixed with ’+’, sorts ascending (default). If prefixed with ’-’, sorts descending.
  • per_page: Sets the page size for query pagination.
  • start: Pagination cursor for resuming a past query, captured from a previous paginator’s .cursor attribute.
Returns: A lazy iterator of Registry objects. Examples: Find all registries with the names that contain “model”
Find all collections in the registries with the name “my_collection” and the tag “my_tag”
Find all artifact versions in the registries with a collection name that contains “my_collection” and a version that has the alias “best”
Find all artifact versions in the registries that contain “model” and have the tag “prod” or alias “best”
Pause iteration and resume later from the same position by saving the paginator’s .cursor and passing it as start=:

method Api.registry

Return a registry given a registry name. Args:
  • name: The name of the registry. This is without the wandb-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.
Returns: A registry object. Examples: Fetch and update a registry

method Api.reports

Get reports for a given project path. Note: 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. Usually there is no reason to change this.
Returns: A Reports object which is an iterable collection of BetaReport objects. Examples:

method Api.run

Return a single run by parsing path in the form entity/project/run_id. Args:
  • path: Path to run in the form entity/project/run_id. If api.entity is set, this can be in the form project/run_id and if api.project is set this can just be the run_id.
Returns: A Run object.

method Api.run_queue

Return the named RunQueue for entity. See Api.create_run_queue for more information on how to create a run queue.

method Api.runs

Returns a 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
Additionally, you can filter by items in the run config or summary metrics. Such as 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
Args:
  • 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 be created_at, heartbeat_at, config.*.value, or summary_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 eagerly fetch the sweep object in each run result.
  • lazy: (bool) Whether to use lazy loading for faster performance. When True (default), only essential run metadata is loaded initially. Heavy fields like config, summaryMetrics, and systemMetrics are loaded on-demand when accessed. Set to False for full data upfront.
Returns: A Runs object, which is an iterable collection of Run objects. Examples:

method Api.slack_integrations

Returns an iterator of Slack integrations for an entity. Args:
  • 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.
Yields:
  • Iterator[SlackIntegration]: An iterator of Slack integrations.
Examples: Get all registered Slack integrations for the team “my-team”:
Find only Slack integrations that post to channel names starting with “team-alerts-”:

method Api.sweep

Return a sweep by parsing path in the form entity/project/sweep_id. Args:
  • path: Path to sweep in the form entity/project/sweep_id. If api.entity is set, this can be in the form project/sweep_id and if api.project is set this can just be the sweep_id.
Returns: A Sweep object.

method Api.sync_tensorboard

Sync a local directory containing tfevent files to wandb.

method Api.team

Return the matching Team with the given name. Args:
  • team: The name of the team.
Returns: A Team object.

method Api.update_automation

Update an existing automation. Args:
  • 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.
Returns: The updated automation. Examples: Disable and edit the description of an existing automation (“my-automation”):
OR

method Api.upsert_run_queue

Upsert a run queue in W&B Launch. Args:
  • name: Name of the queue to create
  • entity: Optional name of the entity to create the queue. If None, 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
Returns: The upserted RunQueue. Raises: ValueError if any of the parameters are invalid wandb.Error on wandb API errors

method Api.user

Return a user from a username or email address. This function only works for local administrators. Use api.viewer to get your own user object. Args:
  • username_or_email: The username or email address of the user.
Returns: A User object or None if a user is not found.

method Api.users

Return all users from a partial username or email address query. This function only works for local administrators. Use api.viewer to get your own user object. Args:
  • username_or_email: The prefix or suffix of the user you want to find.
Returns: An array of User objects.

method Api.webhook_integrations

Returns an iterator of webhook integrations for an entity. Args:
  • 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.
Yields:
  • Iterator[WebhookIntegration]: An iterator of webhook integrations.
Examples: Get all registered webhook integrations for the team “my-team”:
Find only webhook integrations that post requests to “https://my-fake-url.com”: