wandb 0.17.8rc1__py3-none-any.whl → 0.18.0__py3-none-any.whl

package_readme.md

@@ -4,92 +4,86 @@
 
 # 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.
+Use W&B to build better models faster. Track and visualize all the pieces of your machine learning pipeline, from datasets to production machine learning models. Get started with W&B today, [sign up for an account!](https://wandb.com?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=readme)
 
-- 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
+See the [W&B Developer Guide](https://docs.wandb.ai/?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=documentation) and [API Reference Guide](https://docs.wandb.ai/ref?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=documentation) for a full technical description of the W&B platform.
 
--   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
 
-## Quickstart
+Get started with W&B in four steps:
+
+1. First, sign up for a [W&B account](https://wandb.ai/login?utm_source=github&utm_medium=code&utm_campaign=wandb&utm_content=quickstart).
+
+2. Second, install the W&B SDK with [pip](https://pip.pypa.io/en/stable/). Navigate to your terminal and type the following command:
 
 ```shell
 pip install wandb
 ```
 
-In your training script:
+3. Third, log into W&B:
 
 ```python
-import wandb
+wandb.login()
+```
 
-# Your custom arguments defined here
-args = ...
+4. Use the example code snippet below as a template to integrate W&B to your Python script:
 
-wandb.init(config=args, project="my-project")
-wandb.config["more"] = "custom"
+```python
+import wandb
 
+# Start a W&B Run with wandb.init
+run = wandb.init(project="my_first_project")
 
-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})
-```
+# Save model inputs and hyperparameters in a wandb.config object
+config = run.config
+config.learning_rate = 0.01
 
-If you're already using Tensorboard or [TensorboardX](https://github.com/lanpa/tensorboardX), you can integrate with one line:
+# Model training code here ...
 
-```python
-wandb.init(sync_tensorboard=True)
+# Log metrics over time to visualize performance with wandb.log
+for i in range(10):
+    run.log({"loss": ...})
+
+# Mark the run as finished, and finish uploading all data
+run.finish()
 ```
 
-## Running your script
+For example, if the preceding code was stored in a script called train.py:
 
-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).
+```shell
+python train.py
+```
 
-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`.
+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. Navigate to the W&B App to view a dashboard of your first W&B Experiment. Use the W&B App to compare multiple experiments in a unified place, dive into the results of a single run, and much more!
 
-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
+# Integrations
 
-[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)
+Use your favorite framework with W&B. W&B integrations make it fast and easy to set up experiment tracking and data versioning inside existing projects. For more information on how to integrate W&B with the framework of your choice, see [W&B Integrations](https://docs.wandb.ai/guides/integrations) in the W&B Developer Guide.
 
-## Detailed Usage
+ 
 
-Framework specific and detailed usage can be found in our [documentation](http://docs.wandb.com/).
+# Contribution guidelines
+Weights & Biases ❤️ open source, and we welcome contributions from the community! See the [Contribution guide](https://github.com/wandb/wandb/blob/main/CONTRIBUTING.md) for more information on the development workflow and the internals of the wandb library. For wandb bugs and feature requests, visit [GitHub Issues](https://github.com/wandb/wandb/issues) or contact support@wandb.com.
 
-## Testing
+ 
 
-To run basic test use `make test`.  More detailed information can be found at CONTRIBUTING.md.
+# Academic Researchers
+Reach out to W&B Support at support@wandb.com to get a [free academic license](https://www.wandb.com/academic) for you and your research group.
 
-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)
+# W&B Community
 
-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)
+Be a part of the growing W&B Community and interact with the W&B team in our [Discord](https://wandb.me/discord). Stay connected with the latest ML updates and tutorials with [W&B Fully Connected](https://wandb.ai/fully-connected).
 
-## 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.
+# License
 
-[![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).
+[MIT License](https://github.com/wandb/wandb/blob/main/LICENSE)

wandb/__init__.py

@@ -8,7 +8,7 @@ For scripts and interactive notebooks, see https://github.com/wandb/examples.
 
 For reference documentation, see https://docs.wandb.com/ref/python.
 """
-__version__ = "0.17.8rc1"
+__version__ = "0.18.0"
 
 from typing import Optional
 
@@ -76,8 +76,7 @@ from wandb.data_types import JoinedTable
 
 from wandb.wandb_agent import agent
 
-# from wandb.core import *
-from wandb.viz import visualize
+from wandb.sdk.lib.viz import visualize
 from wandb import plot
 from wandb.integration.sagemaker import sagemaker_auth
 from wandb.sdk.internal import profiler
@@ -199,14 +198,13 @@ if wandb_sdk.lib.ipython.in_notebook():
 from .analytics import Sentry as _Sentry
 
 if "dev" in __version__:
+    import wandb.env
     import os
 
     # disable error reporting in dev versions for the python client
-    os.environ["WANDB_ERROR_REPORTING"] = os.environ.get(
-        "WANDB_ERROR_REPORTING", "false"
+    os.environ[wandb.env.ERROR_REPORTING] = os.environ.get(
+        wandb.env.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()
@@ -215,6 +213,7 @@ _sentry.setup()
 __all__ = (
     "__version__",
     "init",
+    "finish",
     "setup",
     "save",
     "sweep",
@@ -237,6 +236,8 @@ __all__ = (
     "Molecule",
     "Histogram",
     "ArtifactTTL",
+    "log_artifact",
+    "use_artifact",
     "log_model",
     "use_model",
     "link_model",

wandb/__init__.pyi

@@ -12,7 +12,9 @@ For reference documentation, see https://docs.wandb.com/ref/python.
 __all__ = (
     "__version__",
     "init",
+    "finish",
     "setup",
+    "login",
     "save",
     "sweep",
     "controller",
@@ -33,6 +35,8 @@ __all__ = (
     "Molecule",
     "Histogram",
     "ArtifactTTL",
+    "log_artifact",
+    "use_artifact",
     "log_model",
     "use_model",
     "link_model",
@@ -48,7 +52,7 @@ __all__ = (
 )
 
 import os
-from typing import Any, Callable, Dict, List, Optional, Sequence, Union
+from typing import Any, Callable, Dict, List, Literal, Optional, Sequence, Union
 
 from wandb.analytics import Sentry as _Sentry
 from wandb.apis import InternalApi, PublicApi
@@ -75,7 +79,7 @@ from wandb.sdk.wandb_run import Run
 from wandb.sdk.wandb_setup import _WandbSetup
 from wandb.wandb_controller import _WandbController
 
-__version__: str = "0.17.8rc1"
+__version__: str = "0.18.0"
 
 run: Optional[Run] = None
 config = wandb_config.Config
@@ -143,7 +147,7 @@ def setup(
 
             # Optional: Explicitly shut down the backend
             wandb.teardown()
-    ```
+        ```
     """
     ...
 
@@ -339,6 +343,7 @@ def init(
             mode if a user isn't logged in to W&B. (default: `False`)
         sync_tensorboard: (bool, optional) Synchronize wandb logs from tensorboard or
             tensorboardX and save the relevant events file. (default: `False`)
+        tensorboard: (bool, optional) Alias for `sync_tensorboard`, deprecated.
         monitor_gym: (bool, optional) Automatically log videos of environment when
             using OpenAI Gym. (default: `False`)
             See [our guide to this integration](https://docs.wandb.com/guides/integrations/openai-gym).
@@ -352,6 +357,12 @@ def init(
             a moment in a previous run to fork a new run from. Creates a new run that picks up
             logging history from the specified run at the specified moment. The target run must
             be in the current project. Example: `fork_from="my-run-id?_step=1234"`.
+        resume_from: (str, optional) A string with the format {run_id}?_step={step} describing
+            a moment in a previous run to resume a run from. This allows users to truncate
+            the history logged to a run at an intermediate step and resume logging from that step.
+            It uses run forking under the hood. The target run must be in the
+            current project. Example: `resume_from="my-run-id?_step=1234"`.
+        settings: (dict, wandb.Settings, optional) Settings to use for this run. (default: None)
 
     Examples:
     ### Set where the run is logged
@@ -389,7 +400,56 @@ def init(
         KeyboardInterrupt: if user interrupts the run.
 
     Returns:
-    A `Run` object.
+        A `Run` object.
+    """
+    ...
+
+def finish(exit_code: Optional[int] = None, quiet: Optional[bool] = None) -> None:
+    """Mark a run as finished, and finish uploading all data.
+
+    This is used when creating multiple runs in the same process.
+    We automatically call this method when your script exits.
+
+    Arguments:
+        exit_code: Set to something other than 0 to mark a run as failed
+        quiet: Set to true to minimize log output
+    """
+    ...
+
+def login(
+    anonymous: Optional[Literal["must", "allow", "never"]] = None,
+    key: Optional[str] = None,
+    relogin: Optional[bool] = None,
+    host: Optional[str] = None,
+    force: Optional[bool] = None,
+    timeout: Optional[int] = None,
+    verify: bool = False,
+) -> bool:
+    """Set up W&B login credentials.
+
+    By default, this will only store credentials locally without
+    verifying them with the W&B server. To verify credentials, pass
+    `verify=True`.
+
+    Arguments:
+        anonymous: (string, optional) Can be "must", "allow", or "never".
+            If set to "must", always log a user in anonymously. If set to
+            "allow", only create an anonymous user if the user
+            isn't already logged in. If set to "never", never log a
+            user anonymously. Default set to "never".
+        key: (string, optional) The API key to use.
+        relogin: (bool, optional) If true, will re-prompt for API key.
+        host: (string, optional) The host to connect to.
+        force: (bool, optional) If true, will force a relogin.
+        timeout: (int, optional) Number of seconds to wait for user input.
+        verify: (bool) Verify the credentials with the W&B server.
+
+    Returns:
+        bool: if key is configured
+
+    Raises:
+        AuthenticationError - if api_key fails verification with the server
+        UsageError - if api_key cannot be configured and no tty
     """
     ...
 
@@ -623,7 +683,7 @@ def log(
 
     Raises:
         wandb.Error: if called before `wandb.init`
-    ValueError: if invalid data is passed
+        ValueError: if invalid data is passed
     """
     ...
 
@@ -677,7 +737,7 @@ def save(
     Returns:
         Paths to the symlinks created for the matched files.
 
-    For historical reasons, this may return a boolean in legacy code.
+        For historical reasons, this may return a boolean in legacy code.
     """
     ...
 
@@ -714,7 +774,7 @@ def sweep(
       prior_runs: The run IDs of existing runs to add to this sweep.
 
     Returns:
-    sweep_id: str. A unique identifier for the sweep.
+      sweep_id: str. A unique identifier for the sweep.
     """
     ...
 
@@ -734,7 +794,7 @@ def controller(
         print(tuner.sweep_id)
         tuner.configure_search(...)
         tuner.configure_stopping(...)
-    ```
+        ```
     """
     ...
 
@@ -764,7 +824,7 @@ def agent(
         project: The name of the project where W&B runs created from
             the sweep are sent to. If the project is not specified, the
             run is sent to a project labeled "Uncategorized".
-    count: The number of sweep config trials to try.
+        count: The number of sweep config trials to try.
     """
     ...
 
@@ -801,7 +861,67 @@ def define_metric(
             previous calls.
 
     Returns:
-    An object that represents this call but can otherwise be discarded.
+        An object that represents this call but can otherwise be discarded.
+    """
+    ...
+
+def log_artifact(
+    artifact_or_path: Union[Artifact, StrPath],
+    name: Optional[str] = None,
+    type: Optional[str] = None,
+    aliases: Optional[List[str]] = None,
+    tags: Optional[List[str]] = None,
+) -> Artifact:
+    """Declare an artifact as an output of a run.
+
+    Arguments:
+        artifact_or_path: (str or Artifact) A path to the contents of this artifact,
+            can be in the following forms:
+                - `/local/directory`
+                - `/local/directory/file.txt`
+                - `s3://bucket/path`
+            You can also pass an Artifact object created by calling
+            `wandb.Artifact`.
+        name: (str, optional) An artifact name. Valid names can be in the following forms:
+                - name:version
+                - name:alias
+                - digest
+            This will default to the basename of the path prepended with the current
+            run id  if not specified.
+        type: (str) The type of artifact to log, examples include `dataset`, `model`
+        aliases: (list, optional) Aliases to apply to this artifact,
+            defaults to `["latest"]`
+        tags: (list, optional) Tags to apply to this artifact, if any.
+
+    Returns:
+        An `Artifact` object.
+    """
+    ...
+
+def use_artifact(
+    artifact_or_name: Union[str, Artifact],
+    type: Optional[str] = None,
+    aliases: Optional[List[str]] = None,
+    use_as: Optional[str] = None,
+) -> Artifact:
+    """Declare an artifact as an input to a run.
+
+    Call `download` or `file` on the returned object to get the contents locally.
+
+    Arguments:
+        artifact_or_name: (str or Artifact) An artifact name.
+            May be prefixed with entity/project/. Valid names
+            can be in the following forms:
+                - name:version
+                - name:alias
+            You can also pass an Artifact object created by calling `wandb.Artifact`
+        type: (str, optional) The type of artifact to use.
+        aliases: (list, optional) Aliases to apply to this artifact
+        use_as: (string, optional) Optional string indicating what purpose the artifact was used with.
+                                   Will be shown in UI.
+
+    Returns:
+        An `Artifact` object.
     """
     ...
 
@@ -847,7 +967,7 @@ def log_model(
         ValueError: if name has invalid special characters
 
     Returns:
-    None
+        None
     """
     ...
 
@@ -888,7 +1008,7 @@ def use_model(name: str) -> FilePathStr:
         AssertionError: if model artifact 'name' is of a type that does not contain the substring 'model'.
 
     Returns:
-    path: (str) path to downloaded model artifact file(s).
+        path: (str) path to downloaded model artifact file(s).
     """
     ...
 
@@ -920,7 +1040,7 @@ def link_model(
         registered_model_name: (str) - the name of the registered model that the model is to be linked to.
             A registered model is a collection of model versions linked to the model registry, typically representing a
             team's specific ML Task. The entity that this registered model belongs to will be derived from the run
-            name: (str, optional) - the name of the model artifact that files in 'path' will be logged to. This will
+        name: (str, optional) - the name of the model artifact that files in 'path' will be logged to. This will
             default to the basename of the path prepended with the current run id  if not specified.
         aliases: (List[str], optional) - alias(es) that will only be applied on this linked artifact
             inside the registered model.
@@ -959,6 +1079,6 @@ def link_model(
         ValueError: if name has invalid special characters
 
     Returns:
-    None
+        None
     """
     ...

wandb/analytics/sentry.py

@@ -211,7 +211,7 @@ class Sentry:
             "sweep_id",
             "deployment",
             "_disable_service",
-            "_require_core",
+            "_require_legacy_service",
             "launch",
         )
 

wandb/cli/cli.py

@@ -35,7 +35,7 @@ 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 import WandbCoreNotAvailableError
+from wandb.errors import UsageError, WandbCoreNotAvailableError
 from wandb.integration.magic import magic_install
 from wandb.sdk.artifacts.artifact_file_cache import get_artifact_file_cache
 from wandb.sdk.launch import utils as launch_utils
@@ -123,6 +123,9 @@ _api = None  # caching api instance allows patching from unit tests
 
 def _get_cling_api(reset=None):
     """Get a reference to the internal api with cling settings."""
+    # TODO: move CLI to wandb-core backend
+    wandb.require("legacy-service")
+
     global _api
     if reset:
         _api = None
@@ -226,6 +229,9 @@ def projects(entity, display=True):
 @click.option("--verify", default=False, is_flag=True, help="Verify login credentials")
 @display_error
 def login(key, host, cloud, relogin, anonymously, verify, no_offline=False):
+    # TODO: move CLI to wandb-core backend
+    wandb.require("legacy-service")
+
     # TODO: handle no_offline
     anon_mode = "must" if anonymously else "never"
 
@@ -433,8 +439,15 @@ def init(ctx, project, entity, reset, mode):
 def beta():
     """Beta versions of wandb CLI commands. Requires wandb-core."""
     # this is the future that requires wandb-core!
+    import wandb.env
+
     wandb._sentry.configure_scope(process_context="wandb_beta")
-    wandb.require("core")
+
+    if wandb.env.is_require_legacy_service():
+        raise UsageError(
+            "wandb beta commands can only be used with wandb-core. "
+            f"Please make sure that `{wandb.env._REQUIRE_LEGACY_SERVICE}` is not set."
+        )
 
     try:
         get_core_path()
@@ -2942,14 +2955,6 @@ def enabled(service):
         )
 
 
-@cli.command("gc", hidden=True, context_settings={"ignore_unknown_options": True})
-@click.argument("args", nargs=-1)
-def gc(args):
-    click.echo(
-        "`wandb gc` command has been removed. Use `wandb sync --clean` to clean up synced runs."
-    )
-
-
 @cli.command(context_settings=CONTEXT, help="Verify your local instance")
 @click.option("--host", default=None, help="Test a specific instance of W&B")
 def verify(host):

wandb/data_types.py

@@ -1420,6 +1420,7 @@ class Graph(Media):
 
     @classmethod
     def from_keras(cls, model):
+        # TODO: his method requires a refactor to work with the keras 3.
         graph = cls()
         # Shamelessly copied (then modified) from keras/keras/utils/layer_utils.py
         sequential_like = cls._is_sequential(model)

wandb/env.py

@@ -90,7 +90,7 @@ _EXECUTABLE = "WANDB_EXECUTABLE"
 LAUNCH_QUEUE_NAME = "WANDB_LAUNCH_QUEUE_NAME"
 LAUNCH_QUEUE_ENTITY = "WANDB_LAUNCH_QUEUE_ENTITY"
 LAUNCH_TRACE_ID = "WANDB_LAUNCH_TRACE_ID"
-_REQUIRE_CORE = "WANDB__REQUIRE_CORE"
+_REQUIRE_LEGACY_SERVICE = "WANDB__REQUIRE_LEGACY_SERVICE"
 
 # For testing, to be removed in future version
 USE_V1_ARTIFACTS = "_WANDB_USE_V1_ARTIFACTS"
@@ -153,8 +153,9 @@ def _env_as_bool(
         return False
 
 
-def is_require_core(env: Optional[Env] = None) -> bool:
-    return _env_as_bool(_REQUIRE_CORE, default="False", env=env)
+def is_require_legacy_service(env: Optional[Env] = None) -> bool:
+    """Return whether wandb.require("legacy-service") was used."""
+    return _env_as_bool(_REQUIRE_LEGACY_SERVICE, default="False", env=env)
 
 
 def is_debug(default: Optional[str] = None, env: Optional[Env] = None) -> bool:

wandb/integration/keras/__init__.py

@@ -1,7 +1,4 @@
-"""Tools for integrating `wandb` with [`Keras`](https://keras.io/).
-
-Keras is a deep learning API for [`TensorFlow`](https://www.tensorflow.org/).
-"""
+"""Tools for integrating `wandb` with [`Keras`](https://keras.io/)."""
 
 __all__ = (
     "WandbCallback",
@@ -11,4 +8,4 @@ __all__ = (
 )
 
 from .callbacks import WandbEvalCallback, WandbMetricsLogger, WandbModelCheckpoint
-from .keras import WandbCallback  # todo: legacy callback to be deprecated
+from .keras import WandbCallback  # TODO: legacy callback to be deprecated

wandb/integration/keras/callbacks/metrics_logger.py

@@ -2,7 +2,7 @@ import sys
 from typing import Any, Dict, Optional, Union
 
 import tensorflow as tf  # type: ignore
-from tensorflow.keras import callbacks  # type: ignore
+from tensorflow.keras import callbacks
 
 import wandb
 from wandb.integration.keras.keras import patch_tf_keras
@@ -86,14 +86,20 @@ class WandbMetricsLogger(callbacks.Callback):
             wandb.define_metric("epoch/*", step_metric="epoch/epoch")
 
     def _get_lr(self) -> Union[float, None]:
-        if isinstance(self.model.optimizer.learning_rate, tf.Variable):
+        if isinstance(
+            self.model.optimizer.learning_rate,
+            (tf.Variable, tf.Tensor),
+        ) or (
+            hasattr(self.model.optimizer.learning_rate, "shape")
+            and self.model.optimizer.learning_rate.shape == ()
+        ):
             return float(self.model.optimizer.learning_rate.numpy().item())
         try:
             return float(
                 self.model.optimizer.learning_rate(step=self.global_step).numpy().item()
             )
-        except Exception:
-            wandb.termerror("Unable to log learning rate.", repeat=False)
+        except Exception as e:
+            wandb.termerror(f"Unable to log learning rate: {e}", repeat=False)
             return None
 
     def on_epoch_end(self, epoch: int, logs: Optional[Dict[str, Any]] = None) -> None:

wandb/integration/keras/callbacks/model_checkpoint.py

@@ -71,9 +71,6 @@ class WandbModelCheckpoint(callbacks.ModelCheckpoint):
             Note that when monitoring validation metrics such as `val_acc` or `val_loss`,
             save_freq must be set to "epoch" as those metrics are only available at the
             end of an epoch.
-        options: (Optional[str]) Optional `tf.train.CheckpointOptions` object if
-            `save_weights_only` is true or optional `tf.saved_model.SaveOptions`
-            object if `save_weights_only` is false.
         initial_value_threshold: (Optional[float]) Floating point initial "best" value of the metric
             to be monitored.
     """
@@ -87,7 +84,6 @@ class WandbModelCheckpoint(callbacks.ModelCheckpoint):
         save_weights_only: bool = False,
         mode: Mode = "auto",
         save_freq: Union[SaveStrategy, int] = "epoch",
-        options: Optional[str] = None,
         initial_value_threshold: Optional[float] = None,
         **kwargs: Any,
     ) -> None:
@@ -99,7 +95,6 @@ class WandbModelCheckpoint(callbacks.ModelCheckpoint):
             save_weights_only=save_weights_only,
             mode=mode,
             save_freq=save_freq,
-            options=options,
             initial_value_threshold=initial_value_threshold,
             **kwargs,
         )

wandb/integration/keras/keras.py

@@ -417,6 +417,17 @@ class WandbCallback(tf.keras.callbacks.Callback):
     ):
         if wandb.run is None:
             raise wandb.Error("You must call wandb.init() before WandbCallback()")
+
+        deprecate(
+            field_name=Deprecated.keras_callback,
+            warning_message=(
+                "WandbCallback is deprecated and will be removed in a future release. "
+                "Please use the WandbMetricsLogger, WandbModelCheckpoint, and WandbEvalCallback "
+                "callbacks instead. "
+                "See https://docs.wandb.ai/guides/integrations/keras for more information."
+            ),
+        )
+
         with wandb.wandb_lib.telemetry.context(run=wandb.run) as tel:
             tel.feature.keras = True
         self.validation_data = None
@@ -552,7 +563,7 @@ class WandbCallback(tf.keras.callbacks.Callback):
         self.params = params
 
     def set_model(self, model):
-        self.model = model
+        super().set_model(model)
         if self.input_type == "auto" and len(model.inputs) == 1:
             self.input_type = wandb.util.guess_data_type(
                 model.inputs[0].shape, risky=True