wandb 0.13.10__py3-none-any.whl → 0.14.0__py3-none-any.whl

wandb/sklearn/plot/shared.py

@@ -1,4 +1,4 @@
-"""Defines plots used by multiple sklearn model classes."""
+"""Define plots used by multiple sklearn model classes."""
 from warnings import simplefilter
 
 import numpy as np
@@ -73,7 +73,7 @@ def learning_curve(
 
     Example:
     ```python
-        wandb.sklearn.plot_learning_curve(model, X, y)
+    wandb.sklearn.plot_learning_curve(model, X, y)
     ```
     """
     not_missing = utils.test_missing(model=model, X=X, y=y)

wandb/sync/__init__.py

@@ -1,5 +1,3 @@
-"""
-module sync
-"""
+"""module sync."""
 
 from .sync import TMPDIR, SyncManager, get_run_from_path, get_runs  # noqa: F401

wandb/sync/sync.py

@@ -1,6 +1,4 @@
-"""
-sync.
-"""
+"""sync."""
 
 import datetime
 import fnmatch
@@ -121,7 +119,7 @@ class SyncThread(threading.Thread):
         return tb_event_files, tb_logdirs, tb_root
 
     def _setup_tensorboard(self, tb_root, tb_logdirs, tb_event_files, sync_item):
-        """Returns true if this sync item can be synced as tensorboard"""
+        """Return true if this sync item can be synced as tensorboard."""
         if tb_root is not None:
             if tb_event_files > 0 and sync_item.endswith(WANDB_SUFFIX):
                 wandb.termwarn("Found .wandb file, not streaming tensorboard metrics.")
@@ -144,6 +142,7 @@ class SyncThread(threading.Thread):
         proto_run.run_id = self._run_id or wandb.util.generate_id()
         proto_run.project = self._project or wandb.util.auto_project_name(None)
         proto_run.entity = self._entity
+        proto_run.telemetry.feature.sync_tfevents = True
 
         url = "{}/{}/{}/runs/{}".format(
             self._app_url,
@@ -218,7 +217,7 @@ class SyncThread(threading.Thread):
         send_manager.finish()
 
     def _robust_scan(self, ds):
-        """Attempt to scan data, handling incomplete files"""
+        """Attempt to scan data, handling incomplete files."""
         try:
             return ds.scan_data()
         except AssertionError as e:

wandb/testing/relay.py

@@ -90,10 +90,10 @@ class Timer:
 
 
 class Context:
-    """
-    Implements a container used to store the snooped state/data of a test,
-    including raw requests and responses; parsed and processed data; and
-    a number of convenience methods and properties for accessing the data.
+    """A container used to store the snooped state/data of a test.
+
+    Includes raw requests and responses, parsed and processed data, and a number of
+    convenience methods and properties for accessing the data.
     """
 
     def __init__(self) -> None:
@@ -242,9 +242,9 @@ class Context:
 
 
 class QueryResolver:
-    """
-    Resolves request/response pairs against a set of known patterns
-    to extract and process useful data, to be later stored in a Context object.
+    """Resolve request/response pairs against a set of known patterns.
+
+    This extracts and processes useful data to be later stored in a Context object.
     """
 
     def __init__(self):
@@ -468,9 +468,10 @@ class InjectedResponse:
         self,
         other: Union["InjectedResponse", requests.Request, requests.PreparedRequest],
     ):
-        """
-        Equality check for InjectedResponse objects.
-        We use this to check if this response should be injected as a replacement of `other`.
+        """Check InjectedResponse object equality.
+
+        We use this to check if this response should be injected as a replacement of
+        `other`.
 
         :param other:
         :return:

wandb/trigger.py

@@ -1,4 +1,4 @@
-"""Module to facilitate adding hooks to wandb actions
+"""Module to facilitate adding hooks to wandb actions.
 
 Usage:
     import trigger

wandb/util.py

@@ -202,7 +202,7 @@ def sentry_exc(
 
 
 def sentry_reraise(exc: Any, delay: bool = False) -> None:
-    """Re-raise an exception after logging it to Sentry
+    """Re-raise an exception after logging it to Sentry.
 
     Use this for top-level exceptions when you want the user to see the traceback.
 
@@ -238,7 +238,8 @@ def sentry_set_scope(
         "sweep_url",
         "sweep_id",
         "deployment",
-        "_require_service",
+        "_disable_service",
+        "launch",
     ]
 
     s = settings_dict
@@ -293,12 +294,19 @@ def sentry_set_scope(
 
 
 def vendor_setup() -> Callable:
-    """This enables us to use the vendor directory for packages we don't depend on
-    Returns a function to call after imports are complete. Make sure to call this
-    function or you will modify the user's path which is never good. The pattern should be:
+    """Create a function that restores user paths after vendor imports.
+
+    This enables us to use the vendor directory for packages we don't depend on. Call
+    the returned function after imports are complete. If you don't you may modify the
+    user's path which is never good.
+
+    Usage:
+
+    ```python
     reset_path = vendor_setup()
     # do any vendor imports...
     reset_path()
+    ```
     """
     original_path = [directory for directory in sys.path]
 
@@ -329,8 +337,7 @@ def vendor_import(name: str) -> Any:
 
 
 def import_module_lazy(name: str) -> Any:
-    """
-    Import a module lazily, only when it is used.
+    """Import a module lazily, only when it is used.
 
     :param (str) name: Dot-separated module path. E.g., 'scipy.stats'.
     """
@@ -356,8 +363,7 @@ def get_module(
     required: Optional[Union[str, bool]] = None,
     lazy: bool = True,
 ) -> Any:
-    """
-    Return module or None. Absolute import is required.
+    """Return module or None. Absolute import is required.
 
     :param (str) name: Dot-separated module path. E.g., 'scipy.stats'.
     :param (str) required: A string to raise a ValueError if missing
@@ -390,7 +396,7 @@ VALUE_BYTES_LIMIT = 100000
 
 
 def app_url(api_url: str) -> str:
-    """Returns the frontend app url without a trailing slash."""
+    """Return the frontend app url without a trailing slash."""
     # TODO: move me to settings
     app_url = get_app_url()
     if app_url is not None:
@@ -409,8 +415,9 @@ def app_url(api_url: str) -> str:
 
 
 def get_full_typename(o: Any) -> Any:
-    """We determine types based on type names so we don't have to import
-    (and therefore depend on) PyTorch, TensorFlow, etc.
+    """Determine types based on type names.
+
+    Avoids needing to to import (and therefore depend on) PyTorch, TensorFlow, etc.
     """
     instance_name = o.__class__.__module__ + "." + o.__class__.__name__
     if instance_name in ["builtins.module", "__builtin__.module"]:
@@ -435,8 +442,8 @@ def is_uri(string: str) -> bool:
 
 
 def local_file_uri_to_path(uri: str) -> str:
-    """
-    Convert URI to local filesystem path.
+    """Convert URI to local filesystem path.
+
     No-op if the uri does not have the expected scheme.
     """
     path = urllib.parse.urlparse(uri).path if uri.startswith("file:") else uri
@@ -444,8 +451,10 @@ def local_file_uri_to_path(uri: str) -> str:
 
 
 def get_local_path_or_none(path_or_uri: str) -> Optional[str]:
-    """Check if the argument is a local path (no scheme or file:///) and return local path if true,
-    None otherwise.
+    """Return path if local, None otherwise.
+
+    Return None if the argument is a local path (not a scheme or file:///). Otherwise
+    return `path_or_uri`.
     """
     parsed_uri = urllib.parse.urlparse(path_or_uri)
     if (
@@ -581,8 +590,10 @@ def is_pandas_data_frame(obj: Any) -> bool:
 
 
 def ensure_matplotlib_figure(obj: Any) -> Any:
-    """Extract the current figure from a matplotlib object or return the object if it's a figure.
-    raises ValueError if the object can't be converted.
+    """Extract the current figure from a matplotlib object.
+
+    Return the object itself if it's a figure.
+    Raises ValueError if the object can't be converted.
     """
     import matplotlib  # type: ignore
     from matplotlib.figure import Figure  # type: ignore
@@ -593,6 +604,7 @@ def ensure_matplotlib_figure(obj: Any) -> Any:
 
     def is_frame_like(self: Any) -> bool:
         """Return True if directly on axes frame.
+
         This is useful for determining if a spine is the edge of an
         old style MPL plot. If so, this function will return True.
         """
@@ -671,10 +683,10 @@ def _find_all_matching_keys(
        match_fn: The function to determine if the key is a match.
        visited: Keep track of visited nodes so we dont recurse forever.
        key_path: Keep track of all the keys to get to the current node.
+
     Yields:
        (key_path, key): The location where the key was found, and the key
     """
-
     if visited is None:
         visited = set()
     me = id(d)
@@ -706,8 +718,8 @@ def _sanitize_numpy_keys(d: Dict) -> Tuple[Dict, bool]:
 
 def json_friendly(  # noqa: C901
     obj: Any,
-) -> Union[Tuple[Any, bool], Tuple[Union[None, str, float], bool]]:  # noqa: C901
-    """Convert an object into something that's more becoming of JSON"""
+) -> Union[Tuple[Any, bool], Tuple[Union[None, str, float], bool]]:
+    """Convert an object into something that's more becoming of JSON."""
     converted = True
     typename = get_full_typename(obj)
 
@@ -770,7 +782,7 @@ def json_friendly(  # noqa: C901
 
 
 def json_friendly_val(val: Any) -> Any:
-    """Make any value (including dict, slice, sequence, etc) JSON friendly"""
+    """Make any value (including dict, slice, sequence, etc) JSON friendly."""
     converted: Union[dict, list]
     if isinstance(val, dict):
         converted = {}
@@ -844,7 +856,7 @@ def maybe_compress_summary(obj: Any, h5_typename: str) -> Tuple[Any, bool]:
 
 
 def launch_browser(attempt_launch_browser: bool = True) -> bool:
-    """Decide if we should launch a browser"""
+    """Decide if we should launch a browser."""
     _display_variables = ["DISPLAY", "WAYLAND_DISPLAY", "MIR_SOCKET"]
     _webbrowser_names_blocklist = ["www-browser", "lynx", "links", "elinks", "w3m"]
 
@@ -873,7 +885,7 @@ def generate_id(length: int = 8) -> str:
 
 
 def parse_tfjob_config() -> Any:
-    """Attempts to parse TFJob config, returning False if it can't find it"""
+    """Attempt to parse TFJob config, returning False if it can't find it."""
     if os.getenv("TF_CONFIG"):
         try:
             return json.loads(os.environ["TF_CONFIG"])
@@ -910,7 +922,9 @@ class WandBJSONEncoderOld(json.JSONEncoder):
 
 class WandBHistoryJSONEncoder(json.JSONEncoder):
     """A JSON Encoder that handles some extra types.
-    This encoder turns numpy like objects with a size > 32 into histograms"""
+
+    This encoder turns numpy like objects with a size > 32 into histograms.
+    """
 
     def default(self, obj: Any) -> Any:
         obj, converted = json_friendly(obj)
@@ -922,7 +936,9 @@ class WandBHistoryJSONEncoder(json.JSONEncoder):
 
 class JSONEncoderUncompressed(json.JSONEncoder):
     """A JSON Encoder that handles some extra types.
-    This encoder turns numpy like objects with a size > 32 into histograms"""
+
+    This encoder turns numpy like objects with a size > 32 into histograms.
+    """
 
     def default(self, obj: Any) -> Any:
         if is_numpy_array(obj):
@@ -949,7 +965,7 @@ def json_dump_uncompressed(obj: Any, fp: IO[str], **kwargs: Any) -> None:
 
 
 def json_dumps_safer_history(obj: Any, **kwargs: Any) -> str:
-    """Convert obj to json, with some extra encodable types, including histograms"""
+    """Convert obj to json, with some extra encodable types, including histograms."""
     return json.dumps(obj, cls=WandBHistoryJSONEncoder, **kwargs)
 
 
@@ -1006,11 +1022,21 @@ def no_retry_auth(e: Any) -> bool:
         return True
     # Crash w/message on forbidden/unauthorized errors.
     if e.response.status_code == 401:
-        raise CommError("Invalid or missing api_key. Run `wandb login`")
+        raise CommError(
+            "The API key is either invalid or missing, or the host is incorrect. "
+            "To resolve this issue, you may try running the 'wandb login --host [hostname]' command. "
+            "The host defaults to 'https://api.wandb.ai' if not specified. "
+            f"(Error {e.response.status_code}: {e.response.reason})"
+        )
     elif wandb.run:
         raise CommError(f"Permission denied to access {wandb.run.path}")
     else:
-        raise CommError("Permission denied, ask the project owner to grant you access")
+        raise CommError(
+            "It appears that you do not have permission to access the requested resource. "
+            "Please reach out to the project owner to grant you access. "
+            "If you have the correct permissions, verify that there are no issues with your networking setup."
+            f"(Error {e.response.status_code}: {e.response.reason})"
+        )
 
 
 def check_retry_conflict(e: Any) -> Optional[bool]:
@@ -1078,6 +1104,7 @@ def find_runner(program: str) -> Union[None, list, List[str]]:
 
     Arguments:
         program: The string name of the program to try to run.
+
     Returns:
         commandline list of strings to run the program (eg. with subprocess.call()) or None
     """
@@ -1096,7 +1123,7 @@ def find_runner(program: str) -> Union[None, list, List[str]]:
 
 
 def downsample(values: Sequence, target_length: int) -> list:
-    """Downsamples 1d values to target_length, including start and end.
+    """Downsample 1d values to target_length, including start and end.
 
     Algorithm just rounds index down.
 
@@ -1131,7 +1158,7 @@ def get_log_file_path() -> str:
 
 
 def docker_image_regex(image: str) -> Any:
-    """regex for valid docker image names"""
+    """Regex match for valid docker image names."""
     if image:
         return re.match(
             r"^(?:(?=[^:\/]{1,253})(?!-)[a-zA-Z0-9-]{1,63}(?<!-)(?:\.(?!-)[a-zA-Z0-9-]{1,63}(?<!-))*(?::[0-9]{1,5})?/)?((?![._-])(?:[a-z0-9._-]*)(?<![._-])(?:/(?![._-])[a-z0-9._-]*(?<![._-]))*)(?::(?![.-])[a-zA-Z0-9_.-]{1,128})?$",
@@ -1141,9 +1168,11 @@ def docker_image_regex(image: str) -> Any:
 
 
 def image_from_docker_args(args: List[str]) -> Optional[str]:
-    """This scans docker run args and attempts to find the most likely docker image argument.
-    If excludes any argments that start with a dash, and the argument after it if it isn't a boolean
-    switch.  This can be improved, we currently fallback gracefully when this fails.
+    """Scan docker run args and attempt to find the most likely docker image argument.
+
+    It excludes any arguments that start with a dash, and the argument after it if it
+    isn't a boolean switch. This can be improved, we currently fallback gracefully when
+    this fails.
     """
     bool_args = [
         "-t",
@@ -1196,12 +1225,12 @@ def load_yaml(file: Any) -> Any:
 
 
 def image_id_from_k8s() -> Optional[str]:
-    """Pings the k8s metadata service for the image id.  Specify the
-    KUBERNETES_NAMESPACE environment variable if your pods are not in
-    the default namespace:
+    """Ping the k8s metadata service for the image id.
+
+    Specify the KUBERNETES_NAMESPACE environment variable if your pods are not in the
+    default namespace:
 
-    - name: KUBERNETES_NAMESPACE
-      valueFrom:
+    - name: KUBERNETES_NAMESPACE valueFrom:
         fieldRef:
           fieldPath: metadata.namespace
     """
@@ -1233,13 +1262,16 @@ def image_id_from_k8s() -> Optional[str]:
     return None
 
 
-def async_call(target: Callable, timeout: Optional[int] = None) -> Callable:
-    """Accepts a method and optional timeout.
-    Returns a new method that will call the original with any args, waiting for upto timeout seconds.
-    This new method blocks on the original and returns the result or None
-    if timeout was reached, along with the thread.
-    You can check thread.is_alive() to determine if a timeout was reached.
-    If an exception is thrown in the thread, we reraise it.
+def async_call(
+    target: Callable, timeout: Optional[Union[int, float]] = None
+) -> Callable:
+    """Wrap a method to run in the background with an optional timeout.
+
+    Returns a new method that will call the original with any args, waiting for upto
+    timeout seconds. This new method blocks on the original and returns the result or
+    None if timeout was reached, along with the thread. You can check thread.is_alive()
+    to determine if a timeout was reached. If an exception is thrown in the thread, we
+    reraise it.
     """
     q: "queue.Queue" = queue.Queue()
 
@@ -1286,7 +1318,7 @@ def read_many_from_queue(
 
 
 def stopwatch_now() -> float:
-    """Get a time value for interval comparisons
+    """Get a time value for interval comparisons.
 
     When possible it is a monotonic clock to prevent backwards time issues.
     """
@@ -1329,7 +1361,7 @@ def prompt_choices(
     input_timeout: Optional[int] = None,
     jupyter: bool = False,
 ) -> str:
-    """Allow a user to choose from a list of options"""
+    """Allow a user to choose from a list of options."""
     for i, choice in enumerate(choices):
         wandb.termlog(f"({i+1}) {choice}")
 
@@ -1351,7 +1383,7 @@ def prompt_choices(
 
 
 def guess_data_type(shape: Sequence[int], risky: bool = False) -> Optional[str]:
-    """Infer the type of data based on the shape of the tensors
+    """Infer the type of data based on the shape of the tensors.
 
     Arguments:
         shape (Sequence[int]): The shape of the data
@@ -1450,7 +1482,6 @@ def parse_sweep_id(parts_dict: dict) -> Optional[str]:
     Returns:
         None or str if there is an error
     """
-
     entity = None
     project = None
     sweep_id = parts_dict.get("name")
@@ -1546,26 +1577,26 @@ def add_import_hook(fullname: str, on_import: Callable) -> None:
 
 
 def host_from_path(path: Optional[str]) -> str:
-    """returns the host of the path"""
+    """Return the host of the path."""
     url = urllib.parse.urlparse(path)
     return str(url.netloc)
 
 
 def uri_from_path(path: Optional[str]) -> str:
-    """returns the URI of the path"""
+    """Return the URI of the path."""
     url = urllib.parse.urlparse(path)
     uri = url.path if url.path[0] != "/" else url.path[1:]
     return str(uri)
 
 
 def is_unicode_safe(stream: TextIO) -> bool:
-    """returns true if the stream supports UTF-8"""
+    """Return True if the stream supports UTF-8."""
     encoding = getattr(stream, "encoding", None)
     return encoding.lower() in {"utf-8", "utf_8"} if encoding else False
 
 
 def _has_internet() -> bool:
-    """Attempts to open a DNS connection to Googles root servers"""
+    """Attempt to open a DNS connection to Googles root servers."""
     try:
         s = socket.create_connection(("8.8.8.8", 53), 0.5)
         s.close()
@@ -1574,8 +1605,8 @@ def _has_internet() -> bool:
         return False
 
 
-def rand_alphanumeric(length: int = 8, rand: Optional[ModuleType] = None) -> str:
-    rand = rand or random
+def rand_alphanumeric(length: int = 8, rand: Optional[random.Random] = None) -> str:
+    rand = rand or random.Random()
     return "".join(rand.choice("0123456789ABCDEF") for _ in range(length))
 
 
@@ -1583,10 +1614,7 @@ def rand_alphanumeric(length: int = 8, rand: Optional[ModuleType] = None) -> str
 def fsync_open(
     path: Union[pathlib.Path, str], mode: str = "w", encoding: Optional[str] = None
 ) -> Generator[IO[Any], None, None]:
-    """
-    Opens a path for I/O, guaranteeing that the file is flushed and
-    fsynced when the file's context expires.
-    """
+    """Open a path for I/O and guarante that the file is flushed and synced."""
     with open(path, mode, encoding=encoding) as f:
         yield f
 
@@ -1597,7 +1625,7 @@ def fsync_open(
 def _is_kaggle() -> bool:
     return (
         os.getenv("KAGGLE_KERNEL_RUN_TYPE") is not None
-        or "kaggle_environments" in sys.modules  # noqa: W503
+        or "kaggle_environments" in sys.modules
     )
 
 
@@ -1633,21 +1661,19 @@ def _is_py_path(path: str) -> bool:
 
 
 def sweep_config_err_text_from_jsonschema_violations(violations: List[str]) -> str:
-    """Consolidate violation strings from wandb/sweeps describing the ways in which a
-    sweep config violates the allowed schema as a single string.
+    """Consolidate schema violation strings from wandb/sweeps into a single string.
 
     Parameters
     ----------
     violations: list of str
         The warnings to render.
 
-    Returns
+    Returns:
     -------
     violation: str
         The consolidated violation text.
 
     """
-
     violation_base = (
         "Malformed sweep config detected! This may cause your sweep to behave in unexpected ways.\n"
         "To avoid this, please fix the sweep config schema violations below:"
@@ -1661,15 +1687,13 @@ def sweep_config_err_text_from_jsonschema_violations(violations: List[str]) -> s
 
 
 def handle_sweep_config_violations(warnings: List[str]) -> None:
-    """Render warnings from gorilla describing the ways in which a
-    sweep config violates the allowed schema as terminal warnings.
+    """Echo sweep config schema violation warnings from Gorilla to the terminal.
 
     Parameters
     ----------
     warnings: list of str
         The warnings to render.
     """
-
     warning = sweep_config_err_text_from_jsonschema_violations(warnings)
     if len(warnings) > 0:
         term.termwarn(warning)
@@ -1677,7 +1701,6 @@ def handle_sweep_config_violations(warnings: List[str]) -> None:
 
 def _log_thread_stacks() -> None:
     """Log all threads, useful for debugging."""
-
     thread_map = {t.ident: t.name for t in threading.enumerate()}
 
     for thread_id, frame in sys._current_frames().items():
@@ -1745,7 +1768,7 @@ def load_json_yaml_dict(config: str) -> Any:
 
 
 def _parse_entity_project_item(path: str) -> tuple:
-    """Parses paths with the following formats: {item}, {project}/{item}, & {entity}/{project}/{item}.
+    """Parse paths with the following formats: {item}, {project}/{item}, & {entity}/{project}/{item}.
 
     Args:
         path: `str`, input path; must be between 0 and 3 in length.
@@ -1771,7 +1794,9 @@ def _parse_entity_project_item(path: str) -> tuple:
 
 
 def _resolve_aliases(aliases: Optional[Union[str, Iterable[str]]]) -> List[str]:
-    """Takes in `aliases` which can be None, str, or List[str] and returns List[str].
+    """Add the 'latest' alias and ensure that all aliases are unique.
+
+    Takes in `aliases` which can be None, str, or List[str] and returns List[str].
     Ensures that "latest" is always present in the returned list.
 
     Args:
@@ -1780,13 +1805,15 @@ def _resolve_aliases(aliases: Optional[Union[str, Iterable[str]]]) -> List[str]:
     Returns:
         List[str], with "latest" always present.
 
-    Example:
-        aliases = _resolve_aliases(["best", "dev"])
-        assert aliases == ["best", "dev", "latest"]
+    Usage:
 
-        aliases = _resolve_aliases("boom")
-        assert aliases == ["boom", "latest"]
+    ```python
+    aliases = _resolve_aliases(["best", "dev"])
+    assert aliases == ["best", "dev", "latest"]
 
+    aliases = _resolve_aliases("boom")
+    assert aliases == ["boom", "latest"]
+    ```
     """
     aliases = aliases or ["latest"]
 
@@ -1870,7 +1897,7 @@ def ensure_text(
 
 
 def make_artifact_name_safe(name: str) -> str:
-    """Make an artifact name safe for use in artifacts"""
+    """Make an artifact name safe for use in artifacts."""
     # artifact names may only contain alphanumeric characters, dashes, underscores, and dots.
     cleaned = re.sub(r"[^a-zA-Z0-9_\-.]", "_", name)
     if len(cleaned) <= 128:
@@ -1880,7 +1907,7 @@ def make_artifact_name_safe(name: str) -> str:
 
 
 def make_docker_image_name_safe(name: str) -> str:
-    """Make a docker image name safe for use in artifacts"""
+    """Make a docker image name safe for use in artifacts."""
     safe_chars = RE_DOCKER_IMAGE_NAME_CHARS.sub("__", name.lower())
     deduped = RE_DOCKER_IMAGE_NAME_SEPARATOR_REPEAT.sub("__", safe_chars)
     trimmed_start = RE_DOCKER_IMAGE_NAME_SEPARATOR_START.sub("", deduped)
@@ -1889,9 +1916,7 @@ def make_docker_image_name_safe(name: str) -> str:
 
 
 def merge_dicts(source: Dict[str, Any], destination: Dict[str, Any]) -> Dict[str, Any]:
-    """
-    Recursively merge two dictionaries.
-    """
+    """Recursively merge two dictionaries."""
     for key, value in source.items():
         if isinstance(value, dict):
             # get node or create one

wandb/viz.py

@@ -16,8 +16,8 @@ class Visualize:
         }
 
     @staticmethod
-    def get_config_key(key: str) -> Tuple[str]:
-        return ("_wandb", "viz", key)
+    def get_config_key(key: str) -> Tuple[str, str, str]:
+        return "_wandb", "viz", key
 
     @property
     def value(self) -> Table:
@@ -54,8 +54,8 @@ class CustomChart:
         }
 
     @staticmethod
-    def get_config_key(key: str) -> Tuple[str]:
-        return ("_wandb", "visualize", key)
+    def get_config_key(key: str) -> Tuple[str, str, str]:
+        return "_wandb", "visualize", key
 
     @staticmethod
     def user_query(table_key: str) -> Dict[str, Any]: