wandb 0.18.4__py3-none-any.whl → 0.18.6__py3-none-any.whl

wandb/apis/public/teams.py

@@ -127,7 +127,7 @@ class Team(Attrs):
     def create(cls, api, team, admin_username=None):
         """Create a new team.
 
-        Arguments:
+        Args:
             api: (`Api`) The api instance to use
             team: (str) The name of the team
             admin_username: (str) optional username of the admin user of the team, defaults to the current user.
@@ -147,7 +147,7 @@ class Team(Attrs):
     def invite(self, username_or_email, admin=False):
         """Invite a user to a team.
 
-        Arguments:
+        Args:
             username_or_email: (str) The username or email address of the user you want to invite
             admin: (bool) Whether to make this user a team admin, defaults to False
 
@@ -168,7 +168,7 @@ class Team(Attrs):
     def create_service_account(self, description):
         """Create a service account for the team.
 
-        Arguments:
+        Args:
             description: (str) A description for this service account
 
         Returns:

wandb/apis/public/users.py

@@ -62,7 +62,7 @@ class User(Attrs):
     def create(cls, api, email, admin=False):
         """Create a new user.
 
-        Arguments:
+        Args:
             api: (`Api`) The api instance to use
             email: (str) The name of the team
             admin: (bool) Whether this user should be a global instance admin

wandb/apis/public/utils.py

@@ -0,0 +1,68 @@
+import re
+from enum import Enum
+from urllib.parse import urlparse
+
+from wandb.sdk.artifacts._validators import is_artifact_registry_project
+
+
+def parse_s3_url_to_s3_uri(url) -> str:
+    """Convert an S3 HTTP(S) URL to an S3 URI.
+
+    Arguments:
+        url (str): The S3 URL to convert, in the format
+                   'http(s)://<bucket>.s3.<region>.amazonaws.com/<key>'.
+                   or 'http(s)://<bucket>.s3.amazonaws.com/<key>'
+
+    Returns:
+        str: The corresponding S3 URI in the format 's3://<bucket>/<key>'.
+
+    Raises:
+        ValueError: If the provided URL is not a valid S3 URL.
+    """
+    # Regular expression to match S3 URL pattern
+    s3_pattern = r"^https?://.*s3.*amazonaws\.com.*"
+    parsed_url = urlparse(url)
+
+    # Check if it's an S3 URL
+    match = re.match(s3_pattern, parsed_url.geturl())
+    if not match:
+        raise ValueError("Invalid S3 URL")
+
+    # Extract bucket name and key
+    bucket_name, *_ = parsed_url.netloc.split(".")
+    key = parsed_url.path.lstrip("/")
+
+    # Construct the S3 URI
+    s3_uri = f"s3://{bucket_name}/{key}"
+
+    return s3_uri
+
+
+class PathType(Enum):
+    """We have lots of different paths users pass in to fetch artifacts, projects, etc.
+
+    This enum is used for specifying what format the path is in given a string path.
+    """
+
+    PROJECT = "PROJECT"
+    ARTIFACT = "ARTIFACT"
+
+
+def parse_org_from_registry_path(path: str, path_type: PathType) -> str:
+    """Parse the org from a registry path.
+
+    Essentially fetching the "entity" from the path but for Registries the entity is actually the org.
+
+    Args:
+        path (str): The path to parse. Can be a project path <entity>/<project> or <project> or an
+        artifact path like <entity>/<project>/<artifact> or <project>/<artifact> or <artifact>
+        path_type (PathType): The type of path to parse.
+    """
+    parts = path.split("/")
+    expected_parts = 3 if path_type == PathType.ARTIFACT else 2
+
+    if len(parts) >= expected_parts:
+        org, project = parts[:2]
+        if is_artifact_registry_project(project):
+            return org
+    return ""

wandb/cli/cli.py

@@ -30,7 +30,9 @@ import wandb.sdk.verify.verify as wandb_verify
 from wandb import Config, Error, env, util, wandb_agent, wandb_sdk
 from wandb.apis import InternalApi, PublicApi
 from wandb.apis.public import RunQueue
+from wandb.errors.links import url_registry
 from wandb.integration.magic import magic_install
+from wandb.sdk.artifacts._validators import is_artifact_registry_project
 from wandb.sdk.artifacts.artifact_file_cache import get_artifact_file_cache
 from wandb.sdk.launch import utils as launch_utils
 from wandb.sdk.launch._launch_add import _launch_add
@@ -38,7 +40,6 @@ from wandb.sdk.launch.errors import ExecutionError, LaunchError
 from wandb.sdk.launch.sweeps import utils as sweep_utils
 from wandb.sdk.launch.sweeps.scheduler import Scheduler
 from wandb.sdk.lib import filesystem
-from wandb.sdk.lib.wburls import wburls
 from wandb.sync import SyncManager, get_run_from_path, get_runs
 
 from .beta import beta
@@ -1172,7 +1173,7 @@ def launch_sweep(
     wandb.termlog(f"Scheduler added to launch queue ({queue})")
 
 
-@cli.command(help=f"Launch or queue a W&B Job. See {wburls.get('cli_launch')}")
+@cli.command(help=f"Launch or queue a W&B Job. See {url_registry.url('wandb-launch')}")
 @click.option(
     "--uri",
     "-u",
@@ -1566,7 +1567,6 @@ def launch(
     "queues",
     default=None,
     multiple=True,
-    metavar="<queue(s)>",
     help="The name of a queue for the agent to watch. Multiple -q flags supported.",
 )
 @click.option(
@@ -2391,6 +2391,15 @@ def get(path, root, type):
             artifact_name = artifact_parts[0]
         else:
             version = "latest"
+        if is_artifact_registry_project(project):
+            organization = path.split("/")[0] if path.count("/") == 2 else ""
+            # set entity to match the settings since in above code it was potentially set to an org
+            settings_entity = public_api.settings["entity"] or public_api.default_entity
+            # Registry artifacts are under the org entity. Because we offer a shorthand and alias for this path,
+            # we need to fetch the org entity to for the user behind the scenes.
+            entity = InternalApi()._resolve_org_entity_name(
+                entity=settings_entity, organization=organization
+            )
         full_path = f"{entity}/{project}/{artifact_name}:{version}"
         wandb.termlog(
             "Downloading {type} artifact {full_path}".format(

wandb/data_types.py

@@ -6,7 +6,7 @@ flexible containers for information, like tables and HTML, and more.
 For more on logging media, see [our guide](https://docs.wandb.com/guides/track/log/media)
 
 For more on logging structured data for interactive dataset and model analysis,
-see [our guide to W&B Tables](https://docs.wandb.com/guides/data-vis).
+see [our guide to W&B Tables](https://docs.wandb.com/guides/tables/).
 
 All of these special data types are subclasses of WBValue. All the data types
 serialize to JSON, since that is what wandb uses to save the objects locally

wandb/docker/__init__.py

@@ -87,7 +87,8 @@ def is_docker_installed() -> bool:
     try:
         # Run the docker --version command
         result = subprocess.run(
-            ["docker", "--version"], stdout=subprocess.PIPE, stderr=subprocess.PIPE
+            ["docker", "--version"],
+            capture_output=True,
         )
         if result.returncode == 0:
             return True

wandb/docker/auth.py

@@ -151,7 +151,7 @@ class AuthConfig(dict):
     ) -> Dict[str, Dict[str, Any]]:
         """Parse authentication entries.
 
-        Arguments:
+        Args:
           entries:        Dict of authentication entries.
           raise_on_error: If set to true, an invalid format will raise
                           InvalidConfigFileError
@@ -386,7 +386,7 @@ def parse_auth(
 ) -> Dict[str, Dict[str, Any]]:
     """Parse authentication entries.
 
-    Arguments:
+    Args:
       entries:        Dict of authentication entries.
       raise_on_error: If set to true, an invalid format will raise
                       InvalidConfigFileError
@@ -430,7 +430,6 @@ def _load_legacy_config(
         }
     except Exception as e:
         log.debug(e)
-        pass
 
     log.debug("All parsing attempts failed - returning empty config")
     return {}

wandb/errors/links.py

@@ -0,0 +1,73 @@
+"""Module containing the WBURLs class and WBURL dataclass.
+
+Used to store predefined URLs that can be associated with a name. The URLs are
+shortened using with the `wandb.me` domain, using dub.co as the shortening service.
+If the URLs need to be updates, use the dub.co service to point to the new URL.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+
+@dataclass
+class WBURL:
+    url: str
+    description: str
+
+
+class Registry:
+    """A collection of URLs that can be associated with a name."""
+
+    def __init__(self):
+        self.urls: dict[str, WBURL] = {
+            "wandb-launch": WBURL(
+                "https://wandb.me/launch",
+                "Link to the W&B launch marketing page",
+            ),
+            "wandb-init": WBURL(
+                "https://wandb.me/wandb-init",
+                "Link to the wandb.init reference documentation page",
+            ),
+            "define-metric": WBURL(
+                "https://wandb.me/define-metric",
+                "Link to the W&B developer guide documentation page on wandb.define_metric",
+            ),
+            "developer-guide": WBURL(
+                "https://wandb.me/developer-guide",
+                "Link to the W&B developer guide top level page",
+            ),
+            "wandb-core": WBURL(
+                "https://wandb.me/wandb-core",
+                "Link to the documentation for the wandb-core service",
+            ),
+            "wandb-server": WBURL(
+                "https://wandb.me/wandb-server",
+                "Link to the documentation for the self-hosted W&B server",
+            ),
+            "multiprocess": WBURL(
+                "https://wandb.me/multiprocess",
+                (
+                    "Link to the W&B developer guide documentation page on how to "
+                    "use wandb in a multiprocess environment"
+                ),
+            ),
+        }
+
+    def url(self, name: str) -> str:
+        """Get the URL associated with the given name."""
+        wb_url = self.urls.get(name)
+        if wb_url:
+            return wb_url.url
+        raise ValueError(f"URL not found for {name}")
+
+    def description(self, name: str) -> str:
+        """Get the description associated with the given name."""
+        wb_url = self.urls.get(name)
+        if wb_url:
+            return wb_url.description
+        raise ValueError(f"Description not found for {name}")
+
+
+# This is an instance of the Links class that can be used to access the URLs
+url_registry = Registry()

wandb/errors/term.py

@@ -136,12 +136,13 @@ def dynamic_text() -> Iterator[DynamicBlock | None]:
     with _dynamic_text_lock:
         _dynamic_blocks.append(block)
 
-    yield block
-
-    with _dynamic_text_lock:
-        block._lines_to_print = []
-        _l_rerender_dynamic_blocks()
-        _dynamic_blocks.remove(block)
+    try:
+        yield block
+    finally:
+        with _dynamic_text_lock:
+            block._lines_to_print = []
+            _l_rerender_dynamic_blocks()
+            _dynamic_blocks.remove(block)
 
 
 def _sys_stderr_isatty() -> bool:

wandb/filesync/step_prepare.py

@@ -149,7 +149,7 @@ class StepPrepare:
     ) -> Mapping[str, "CreateArtifactFilesResponseFile"]:
         """Execute the prepareFiles API call.
 
-        Arguments:
+        Args:
             batch: List of RequestPrepare objects
         Returns:
             dict of (save_name: ResponseFile) pairs where ResponseFile is a dict with

wandb/filesync/upload_job.py

@@ -30,7 +30,7 @@ class UploadJob:
     ) -> None:
         """A file uploader.
 
-        Arguments:
+        Args:
             push_function: function(save_name, actual_path) which actually uploads
                 the file.
             save_name: string logical location of the file relative to the run

wandb/integration/catboost/catboost.py

@@ -13,7 +13,7 @@ from wandb.sdk.lib import telemetry as wb_telemetry
 class WandbCallback:
     """`WandbCallback` automatically integrates CatBoost with wandb.
 
-    Arguments:
+    Args:
         - metric_period: (int) if you are passing `metric_period` to your CatBoost model please pass the same value here (default=1).
 
     Passing `WandbCallback` to CatBoost will:
@@ -115,7 +115,7 @@ def log_summary(
 ) -> None:
     """`log_summary` logs useful metrics about catboost model after training is done.
 
-    Arguments:
+    Args:
         model: it can be CatBoostClassifier or CatBoostRegressor.
         log_all_params: (boolean) if True (default) log the model hyperparameters as W&B config.
         save_model_checkpoint: (boolean) if True saves the model upload as W&B artifacts.

wandb/integration/diffusers/pipeline_resolver.py

@@ -28,7 +28,7 @@ class DiffusersPipelineResolver:
     ) -> Any:
         """Main call method for the `DiffusersPipelineResolver` class.
 
-        Arguments:
+        Args:
             args: (Sequence[Any]) List of arguments.
             kwargs: (Dict[str, Any]) Dictionary of keyword arguments.
             response: (wandb.sdk.integration_utils.auto_logging.Response) The response from

wandb/integration/diffusers/resolvers/multimodal.py

@@ -597,7 +597,7 @@ class DiffusersMultiModalPipelineResolver:
     `__call__` for `wandb.integration.diffusers.pipeline_resolver.DiffusersPipelineResolver`.
     This is based on `wandb.sdk.integration_utils.auto_logging.RequestResponseResolver`.
 
-    Arguments:
+    Args:
         pipeline_name: (str) The name of the Diffusion Pipeline.
     """
 
@@ -621,7 +621,7 @@ class DiffusersMultiModalPipelineResolver:
     ) -> Any:
         """Main call method for the `DiffusersPipelineResolver` class.
 
-        Arguments:
+        Args:
             args: (Sequence[Any]) List of arguments.
             kwargs: (Dict[str, Any]) Dictionary of keyword arguments.
             response: (wandb.sdk.integration_utils.auto_logging.Response) The response from
@@ -678,7 +678,7 @@ class DiffusersMultiModalPipelineResolver:
     def get_output_images(self, response: Response) -> List:
         """Unpack the generated images, audio, video, etc. from the Diffusion Pipeline's response.
 
-        Arguments:
+        Args:
             response: (wandb.sdk.integration_utils.auto_logging.Response) The response from
                 the request.
 
@@ -704,7 +704,7 @@ class DiffusersMultiModalPipelineResolver:
     def log_media(self, image: Any, loggable_kwarg_chunks: List, idx: int) -> None:
         """Log the generated images, audio, video, etc. from the Diffusion Pipeline's response along with an optional caption to a media panel in the run.
 
-        Arguments:
+        Args:
             image: (Any) The generated images, audio, video, etc. from the Diffusion
                 Pipeline's response.
             loggable_kwarg_chunks: (List) Loggable chunks of kwargs.
@@ -780,7 +780,7 @@ class DiffusersMultiModalPipelineResolver:
     ) -> None:
         """Populate the row of the `wandb.Table`.
 
-        Arguments:
+        Args:
             image: (Any) The generated images, audio, video, etc. from the Diffusion
                 Pipeline's response.
             loggable_kwarg_chunks: (List) Loggable chunks of kwargs.
@@ -819,7 +819,7 @@ class DiffusersMultiModalPipelineResolver:
     ) -> Dict[str, Any]:
         """Prepare the loggable dictionary, which is the packed data as a dictionary for logging to wandb, None if an exception occurred.
 
-        Arguments:
+        Args:
             pipeline: (Any) The Diffusion Pipeline.
             response: (wandb.sdk.integration_utils.auto_logging.Response) The response from
                 the request.

wandb/integration/diffusers/resolvers/utils.py

@@ -69,7 +69,7 @@ def postprocess_np_arrays_for_video(
 def decode_sdxl_t2i_latents(pipeline: Any, latents: "torch_float_tensor") -> List:
     """Decode latents generated by [`diffusers.StableDiffusionXLPipeline`](https://huggingface.co/docs/diffusers/main/en/api/pipelines/stable_diffusion/stable_diffusion_xl#stable-diffusion-xl).
 
-    Arguments:
+    Args:
         pipeline: (diffusers.DiffusionPipeline) The Diffusion Pipeline from
             [`diffusers`](https://huggingface.co/docs/diffusers).
         latents (torch.FloatTensor): The generated latents.

wandb/integration/fastai/__init__.py

@@ -50,11 +50,12 @@ import fastai
 from fastai.callbacks import TrackerCallback
 
 import wandb
+from wandb.sdk.lib import ipython
 
 try:
     import matplotlib
 
-    if wandb.wandb_lib.ipython._get_python_type() != "jupyter":  # type: ignore[attr-defined]
+    if not ipython.in_jupyter():
         matplotlib.use("Agg")  # non-interactive backend (avoid tkinter issues)
     import matplotlib.pyplot as plt
 except ImportError:
@@ -66,7 +67,7 @@ class WandbCallback(TrackerCallback):
 
     Optionally logs weights, gradients, sample predictions and best trained model.
 
-    Arguments:
+    Args:
         learn (fastai.basic_train.Learner): the fast.ai learner to hook.
         log (str): "gradients", "parameters", "all", or None. Losses & metrics are always logged.
         save_model (bool): save model at the end of each epoch. It will also load best model at the end of training.

wandb/integration/keras/callbacks/metrics_logger.py

@@ -38,7 +38,7 @@ class WandbMetricsLogger(callbacks.Callback):
     `step_size` is number of training steps per epoch. `step_size` can be calculated as
     the product of the cardinality of the training dataset and the batch size.
 
-    Arguments:
+    Args:
         log_freq: ("epoch", "batch", or int) if "epoch", logs metrics
             at the end of each epoch. If "batch", logs metrics at the end
             of each batch. If an integer, logs metrics at the end of that

wandb/integration/keras/callbacks/model_checkpoint.py

@@ -45,7 +45,7 @@ class WandbModelCheckpoint(callbacks.ModelCheckpoint):
         - Save only model weights, or save the whole model.
         - Save the model either in SavedModel format or in `.h5` format.
 
-    Arguments:
+    Args:
         filepath: (Union[str, os.PathLike]) path to save the model file. `filepath`
             can contain named formatting options, which will be filled by the value
             of `epoch` and keys in `logs` (passed in `on_epoch_end`). For example:

wandb/integration/keras/keras.py

@@ -313,7 +313,7 @@ class WandbCallback(tf.keras.callbacks.Callback):
 
     `WandbCallback` can optionally save training and validation data for wandb to visualize.
 
-    Arguments:
+    Args:
         monitor: (str) name of metric to monitor.  Defaults to `val_loss`.
         mode: (str) one of {`auto`, `min`, `max`}.
             `min` - save model when monitor is minimized

wandb/integration/kfp/kfp_patch.py

@@ -163,7 +163,7 @@ def _get_function_source_definition(func: Callable) -> str:
 
     # For wandb, allow decorators (so we can use the @wandb_log decorator)
     func_code_lines = itertools.dropwhile(
-        lambda x: not (x.startswith("def") or x.startswith("@wandb_log")),
+        lambda x: not (x.startswith(("def", "@wandb_log"))),
         func_code_lines,
     )
 

wandb/integration/lightgbm/__init__.py

@@ -155,7 +155,7 @@ class _WandbCallback:
 def wandb_callback(log_params: bool = True, define_metric: bool = True) -> Callable:
     """Automatically integrates LightGBM with wandb.
 
-    Arguments:
+    Args:
         log_params: (boolean) if True (default) logs params passed to lightgbm.train as W&B config
         define_metric: (boolean) if True (default) capture model performance at the best step, instead of the last step, of training in your `wandb.summary`
 
@@ -190,7 +190,7 @@ def log_summary(
 ) -> None:
     """Log useful metrics about lightgbm model after training is done.
 
-    Arguments:
+    Args:
         model: (Booster) is an instance of lightgbm.basic.Booster.
         feature_importance: (boolean) if True (default), logs the feature importance plot.
         save_model_checkpoint: (boolean) if True saves the best model and upload as W&B artifacts.

wandb/integration/magic.py

@@ -9,6 +9,7 @@ import yaml
 
 import wandb
 from wandb import trigger
+from wandb.sdk.lib import ipython
 from wandb.util import add_import_hook, get_optional_module
 
 _import_hook = None
@@ -505,8 +506,7 @@ def magic_install(init_args=None):
     # process system args
     _process_system_args()
     # install argparse wrapper
-    in_jupyter_or_ipython = wandb.wandb_sdk.lib.ipython._get_python_type() != "python"
-    if not in_jupyter_or_ipython:
+    if not ipython.in_notebook():
         _monkey_argparse()
 
     # track init calls

wandb/integration/metaflow/metaflow.py

@@ -300,7 +300,7 @@ def wandb_log(
     - Decorating the flow is equivalent to decorating all steps with a default
     - Decorating a step after decorating the flow will overwrite the flow decoration
 
-    Arguments:
+    Args:
         func: (`Callable`). The method or class being decorated (if decorating a step or flow respectively).
         datasets: (`bool`). If `True`, log datasets.  Datasets can be a `pd.DataFrame` or `pathlib.Path`.  The default value is `False`, so datasets are not logged.
         models: (`bool`). If `True`, log models.  Models can be a `nn.Module` or `sklearn.base.BaseEstimator`.  The default value is `False`, so models are not logged.

wandb/integration/sacred/__init__.py

@@ -10,7 +10,7 @@ import wandb
 class WandbObserver(RunObserver):
     """Log sacred experiment data to W&B.
 
-    Arguments:
+    Args:
         Accepts all the arguments accepted by wandb.init().
 
         name — A display name for this run, which shows up in the UI and is editable, doesn't have to be unique

wandb/integration/sagemaker/auth.py

@@ -7,7 +7,7 @@ from wandb import env
 def sagemaker_auth(overrides=None, path=".", api_key=None):
     """Write a secrets.env file with the W&B ApiKey and any additional secrets passed.
 
-    Arguments:
+    Args:
         overrides (dict, optional): Additional environment variables to write
                                     to secrets.env
         path (str, optional): The path to write the secrets file.

wandb/integration/sklearn/plot/classifier.py

@@ -37,7 +37,7 @@ def classifier(
 
     Should only be called with a fitted classifier (otherwise an error is thrown).
 
-    Arguments:
+    Args:
         model: (classifier) Takes in a fitted classifier.
         X_train: (arr) Training set features.
         y_train: (arr) Training set labels.
@@ -117,7 +117,7 @@ def roc(
 ):
     """Log the receiver-operating characteristic curve.
 
-    Arguments:
+    Args:
         y_true: (arr) Test set labels.
         y_probas: (arr) Test set predicted probabilities.
         labels: (list) Named labels for target variable (y). Makes plots easier to
@@ -150,7 +150,7 @@ def confusion_matrix(
 
     Confusion matrices depict the pattern of misclassifications by a model.
 
-    Arguments:
+    Args:
         y_true: (arr) Test set labels.
         y_probas: (arr) Test set predicted probabilities.
         labels: (list) Named labels for target variable (y). Makes plots easier to
@@ -194,7 +194,7 @@ def precision_recall(
     Precision-recall curves depict the tradeoff between positive predictive value (precision)
     and true positive rate (recall) as the threshold of a classifier is shifted.
 
-    Arguments:
+    Args:
         y_true: (arr) Test set labels.
         y_probas: (arr) Test set predicted probabilities.
         labels: (list) Named labels for target variable (y). Makes plots easier to
@@ -226,7 +226,7 @@ def feature_importances(
     Should only be called with a fitted classifier (otherwise an error is thrown).
     Only works with classifiers that have a feature_importances_ attribute, like trees.
 
-    Arguments:
+    Args:
         model: (clf) Takes in a fitted classifier.
         feature_names: (list) Names for features. Makes plots easier to read by
                               replacing feature indexes with corresponding names.
@@ -254,7 +254,7 @@ def class_proportions(y_train=None, y_test=None, labels=None):
 
     Useful for detecting imbalanced classes.
 
-    Arguments:
+    Args:
         y_train: (arr) Training set labels.
         y_test: (arr) Test set labels.
         labels: (list) Named labels for target variable (y). Makes plots easier to
@@ -298,7 +298,7 @@ def calibration_curve(clf=None, X=None, y=None, clf_name="Classifier"):  # noqa:
 
     Please note this function fits variations of the model on the training set when called.
 
-    Arguments:
+    Args:
         clf: (clf) Takes in a fitted classifier.
         X: (arr) Training set features.
         y: (arr) Training set labels.

wandb/integration/sklearn/plot/clusterer.py

@@ -20,7 +20,7 @@ def clusterer(model, X_train, cluster_labels, labels=None, model_name="Clusterer
 
     Should only be called with a fitted clusterer (otherwise an error is thrown).
 
-    Arguments:
+    Args:
         model: (clusterer) Takes in a fitted clusterer.
         X_train: (arr) Training set features.
         cluster_labels: (list) Names for cluster labels. Makes plots easier to read
@@ -68,7 +68,7 @@ def elbow_curve(
 
     Please note this function fits the model on the training set when called.
 
-    Arguments:
+    Args:
         model: (clusterer) Takes in a fitted clusterer.
         X: (arr) Training set features.
 
@@ -118,7 +118,7 @@ def silhouette(
 
     Please note this function fits the model on the training set when called.
 
-    Arguments:
+    Args:
         model: (clusterer) Takes in a fitted clusterer.
         X: (arr) Training set features.
         cluster_labels: (list) Names for cluster labels. Makes plots easier to read