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

wandb/sdk/wandb_run.py

@@ -39,7 +39,6 @@ from wandb._globals import _datatypes_set_callback
 from wandb.apis import internal, public
 from wandb.apis.internal import Api
 from wandb.apis.public import Api as PublicApi
-from wandb.errors import MailboxError
 from wandb.proto.wandb_internal_pb2 import (
     MetricRecord,
     PollExitResponse,
@@ -67,6 +66,7 @@ from wandb.viz import CustomChart, Visualize, custom_chart
 from . import wandb_artifacts, wandb_config, wandb_metric, wandb_summary
 from .data_types._dtypes import TypeRegistry
 from .interface.artifacts import Artifact as ArtifactInterface
+from .interface.artifacts import ArtifactNotLoggedError
 from .interface.interface import GlobStr, InterfaceBase
 from .interface.summary_record import SummaryRecord
 from .lib import (
@@ -82,7 +82,7 @@ from .lib import (
 )
 from .lib.exit_hooks import ExitHooks
 from .lib.git import GitRepo
-from .lib.mailbox import MailboxHandle, MailboxProbe, MailboxProgress
+from .lib.mailbox import MailboxError, MailboxHandle, MailboxProbe, MailboxProgress
 from .lib.printer import get_printer
 from .lib.proto_util import message_to_dict
 from .lib.reporting import Reporter
@@ -233,6 +233,8 @@ class RunStatusChecker:
 
             if result:
                 process(result)
+                # if request finished, clear the handle to send on the next interval
+                local_handle = None
 
             time_elapsed = time.monotonic() - time_probe
             wait_time = max(self._stop_polling_interval - time_elapsed, 0)
@@ -295,7 +297,6 @@ class RunStatusChecker:
 
 
 class _run_decorator:  # noqa: N801
-
     _is_attaching: str = ""
 
     class Dummy:
@@ -305,7 +306,6 @@ class _run_decorator:  # noqa: N801
     def _attach(cls, func: Callable) -> Callable:
         @functools.wraps(func)
         def wrapper(self: Type["Run"], *args: Any, **kwargs: Any) -> Any:
-
             # * `_attach_id` is only assigned in service hence for all non-service cases
             # it will be a passthrough.
             # * `_attach_pid` is only assigned in _init (using _attach_pid guarantees single attach):
@@ -315,7 +315,6 @@ class _run_decorator:  # noqa: N801
                 getattr(self, "_attach_id", None)
                 and getattr(self, "_attach_pid", None) != os.getpid()
             ):
-
                 if cls._is_attaching:
                     message = (
                         f"Trying to attach `{func.__name__}` "
@@ -362,7 +361,7 @@ class _run_decorator:  # noqa: N801
                     settings = getattr(self, "_settings", None)
                     if settings and settings["strict"]:
                         wandb.termerror(message, repeat=False)
-                        raise errors.MultiprocessError(
+                        raise errors.UnsupportedError(
                             f"`{func.__name__}` does not support multiprocessing"
                         )
                     wandb.termwarn(message, repeat=False)
@@ -430,7 +429,7 @@ class Run:
     and then log information only from that process, or you can create a run in each process,
     logging from each separately, and group the results together with the `group` argument
     to `wandb.init`. For more details on distributed training with W&B, check out
-    [our guide](https://docs.wandb.ai/guides/track/advanced/distributed-training).
+    [our guide](https://docs.wandb.ai/guides/track/log/distributed-training).
 
     Currently, there is a parallel `Run` object in the `wandb.Api`. Eventually these
     two objects will be merged.
@@ -528,7 +527,6 @@ class Run:
         sweep_config: Optional[Dict[str, Any]] = None,
         launch_config: Optional[Dict[str, Any]] = None,
     ) -> None:
-
         self._settings = settings
         self._config = wandb_config.Config()
         self._config._set_callback(self._config_callback)
@@ -643,7 +641,7 @@ class Run:
         self._attach_pid = os.getpid()
 
         # for now, use runid as attach id, this could/should be versioned in the future
-        if self._settings._require_service:
+        if not self._settings._disable_service:
             self._attach_id = self._settings.run_id
 
     def _set_iface_pid(self, iface_pid: int) -> None:
@@ -774,10 +772,13 @@ class Run:
         except Exception:
             wandb.termwarn("Cannot find valid git repo associated with this directory.")
 
+    def __deepcopy__(self, memo: Dict[int, Any]) -> "Run":
+        return self
+
     def __getstate__(self) -> Any:
-        """Custom pickler."""
+        """Return run state as a custom pickle."""
         # We only pickle in service mode
-        if not self._settings or not self._settings._require_service:
+        if not self._settings or self._settings._disable_service:
             return
 
         _attach_id = self._attach_id
@@ -787,7 +788,7 @@ class Run:
         return dict(_attach_id=self._attach_id, _init_pid=self._init_pid)
 
     def __setstate__(self, state: Any) -> None:
-        """Custom unpickler."""
+        """Set run state from a custom pickle."""
         if not state:
             return
 
@@ -809,7 +810,7 @@ class Run:
     @property
     @_run_decorator._attach
     def settings(self) -> Settings:
-        """Returns a frozen copy of run's Settings object."""
+        """A frozen copy of run's Settings object."""
         cp = self._settings.copy()
         cp.freeze()
         return cp
@@ -817,13 +818,13 @@ class Run:
     @property
     @_run_decorator._attach
     def dir(self) -> str:
-        """Returns the directory where files associated with the run are saved."""
+        """The directory where files associated with the run are saved."""
         return self._settings.files_dir
 
     @property
     @_run_decorator._attach
     def config(self) -> wandb_config.Config:
-        """Returns the config object associated with this run."""
+        """Config object associated with this run."""
         return self._config
 
     @property
@@ -834,7 +835,7 @@ class Run:
     @property
     @_run_decorator._attach
     def name(self) -> Optional[str]:
-        """Returns the display name of the run.
+        """Display name of the run.
 
         Display names are not guaranteed to be unique and may be descriptive.
         By default, they are randomly generated.
@@ -857,7 +858,7 @@ class Run:
     @property
     @_run_decorator._attach
     def notes(self) -> Optional[str]:
-        """Returns the notes associated with the run, if there are any.
+        """Notes associated with the run, if there are any.
 
         Notes can be a multiline string and can also use markdown and latex equations
         inside `$$`, like `$x + 3$`.
@@ -878,7 +879,7 @@ class Run:
     @property
     @_run_decorator._attach
     def tags(self) -> Optional[Tuple]:
-        """Returns the tags associated with the run, if there are any."""
+        """Tags associated with the run, if there are any."""
         if self._tags:
             return self._tags
         run_obj = self._run_obj or self._run_obj_offline
@@ -898,7 +899,7 @@ class Run:
     @property
     @_run_decorator._attach
     def id(self) -> str:
-        """Returns the identifier for this run."""
+        """Identifier for this run."""
         if TYPE_CHECKING:
             assert self._run_id is not None
         return self._run_id
@@ -906,7 +907,7 @@ class Run:
     @property
     @_run_decorator._attach
     def sweep_id(self) -> Optional[str]:
-        """Returns the ID of the sweep associated with the run, if there is one."""
+        """ID of the sweep associated with the run, if there is one."""
         if not self._run_obj:
             return None
         return self._run_obj.sweep_id or None
@@ -920,7 +921,7 @@ class Run:
     @property
     @_run_decorator._attach
     def path(self) -> str:
-        """Returns the path to the run.
+        """Path to the run.
 
         Run paths include entity, project, and run ID, in the format
         `entity/project/run_id`.
@@ -937,7 +938,7 @@ class Run:
     @property
     @_run_decorator._attach
     def start_time(self) -> float:
-        """Returns the unix time stamp, in seconds, when the run started."""
+        """Unix timestamp (in seconds) of when the run started."""
         return self._get_start_time()
 
     def _get_starting_step(self) -> int:
@@ -946,19 +947,19 @@ class Run:
     @property
     @_run_decorator._attach
     def starting_step(self) -> int:
-        """Returns the first step of the run."""
+        """The first step of the run."""
         return self._get_starting_step()
 
     @property
     @_run_decorator._attach
     def resumed(self) -> bool:
-        """Returns True if the run was resumed, False otherwise."""
+        """True if the run was resumed, False otherwise."""
         return self._run_obj.resumed if self._run_obj else False
 
     @property
     @_run_decorator._attach
     def step(self) -> int:
-        """Returns the current value of the step.
+        """Current value of the step.
 
         This counter is incremented by `wandb.log`.
         """
@@ -998,7 +999,7 @@ class Run:
     @property
     @_run_decorator._attach
     def group(self) -> str:
-        """Returns the name of the group associated with the run.
+        """Name of the group associated with the run.
 
         Setting a group helps the W&B UI organize runs in a sensible way.
 
@@ -1018,7 +1019,7 @@ class Run:
     @property
     @_run_decorator._attach
     def project(self) -> str:
-        """Returns the name of the W&B project associated with the run."""
+        """Name of the W&B project associated with the run."""
         return self.project_name()
 
     @_run_decorator._attach
@@ -1029,7 +1030,7 @@ class Run:
         include_fn: Callable[[str], bool] = _is_py_path,
         exclude_fn: Callable[[str], bool] = filenames.exclude_wandb_fn,
     ) -> Optional[Artifact]:
-        """Saves the current state of your code to a W&B Artifact.
+        """Save the current state of your code to a W&B Artifact.
 
         By default, it walks the current directory and logs all files that end with `.py`.
 
@@ -1086,7 +1087,7 @@ class Run:
         return self._log_artifact(art)
 
     def get_url(self) -> Optional[str]:
-        """Returns the url for the W&B run, if there is one.
+        """Return the url for the W&B run, if there is one.
 
         Offline runs will not have a url.
         """
@@ -1096,7 +1097,7 @@ class Run:
         return self._settings.run_url
 
     def get_project_url(self) -> Optional[str]:
-        """Returns the url for the W&B project associated with the run, if there is one.
+        """Return the url for the W&B project associated with the run, if there is one.
 
         Offline runs will not have a project url.
         """
@@ -1106,7 +1107,7 @@ class Run:
         return self._settings.project_url
 
     def get_sweep_url(self) -> Optional[str]:
-        """Returns the url for the sweep associated with the run, if there is one."""
+        """Return the url for the sweep associated with the run, if there is one."""
         if self._settings._offline:
             wandb.termwarn("URL not available in offline run")
             return None
@@ -1115,13 +1116,13 @@ class Run:
     @property
     @_run_decorator._attach
     def url(self) -> Optional[str]:
-        """Returns the W&B url associated with the run."""
+        """The W&B url associated with the run."""
         return self.get_url()
 
     @property
     @_run_decorator._attach
     def entity(self) -> str:
-        """Returns the name of the W&B entity associated with the run.
+        """The name of the W&B entity associated with the run.
 
         Entity can be a user name or the name of a team or organization.
         """
@@ -1217,7 +1218,7 @@ class Run:
 
     @_run_decorator._attach
     def display(self, height: int = 420, hidden: bool = False) -> bool:
-        """Displays this run in jupyter."""
+        """Display this run in jupyter."""
         if self._settings._jupyter and ipython.in_jupyter():
             ipython.display_html(self.to_html(height, hidden))
             return True
@@ -1227,7 +1228,7 @@ class Run:
 
     @_run_decorator._attach
     def to_html(self, height: int = 420, hidden: bool = False) -> str:
-        """Generates HTML containing an iframe displaying the current run."""
+        """Generate HTML containing an iframe displaying the current run."""
         url = self._settings.run_url + "?jupyter=true"
         style = f"border:none;width:100%;height:{height}px;"
         prefix = ""
@@ -1454,7 +1455,7 @@ class Run:
     def _add_singleton(
         self, data_type: str, key: str, value: Dict[Union[int, str], str]
     ) -> None:
-        """Stores a singleton item to wandb config.
+        """Store a singleton item to wandb config.
 
         A singleton in this context is a piece of data that is continually
         logged with the same value in each history step, but represented
@@ -1525,7 +1526,7 @@ class Run:
         commit: Optional[bool] = None,
         sync: Optional[bool] = None,
     ) -> None:
-        """Logs a dictonary of data to the current run's history.
+        """Log a dictonary of data to the current run's history.
 
         Use `wandb.log` to log data from runs, such as scalars, images, video,
         histograms, plots, and tables.
@@ -1538,7 +1539,7 @@ class Run:
         the summary values for these metrics.
 
         Visualize logged data in the workspace at [wandb.ai](https://wandb.ai),
-        or locally on a [self-hosted instance](https://docs.wandb.ai/self-hosted)
+        or locally on a [self-hosted instance](https://docs.wandb.ai/guides/hosting)
         of the W&B app, or export data to visualize and explore locally, e.g. in
         Jupyter notebooks, with [our API](https://docs.wandb.ai/guides/track/public-api-guide).
 
@@ -1548,7 +1549,7 @@ class Run:
         Logged values don't have to be scalars. Logging any wandb object is supported.
         For example `wandb.log({"example": wandb.Image("myimage.jpg")})` will log an
         example image which will be displayed nicely in the W&B UI.
-        See the [reference documentation](https://docs.wandb.com/library/reference/data_types)
+        See the [reference documentation](https://docs.wandb.com/ref/python/data-types)
         for all of the different supported types or check out our
         [guides to logging](https://docs.wandb.ai/guides/track/log) for examples,
         from 3D molecular structures and segmentation masks to PR curves and histograms.
@@ -1749,7 +1750,6 @@ class Run:
         base_path: Optional[str] = None,
         policy: "PolicyName" = "live",
     ) -> Union[bool, List[str]]:
-
         if policy not in ("live", "end", "now"):
             raise ValueError(
                 'Only "live" "end" and "now" policies are currently supported.'
@@ -1833,7 +1833,7 @@ class Run:
     def finish(
         self, exit_code: Optional[int] = None, quiet: Optional[bool] = None
     ) -> None:
-        """Marks a run as finished, and finishes uploading all data.
+        """Mark a run as finished, and finishe uploading all data.
 
         This is used when creating multiple runs in the same process. We automatically
         call this method when your script exits or if you use the run context manager.
@@ -1875,7 +1875,7 @@ class Run:
     @_run_decorator._noop
     @_run_decorator._attach
     def join(self, exit_code: Optional[int] = None) -> None:
-        """Deprecated alias for `finish()` - please use finish."""
+        """Deprecated alias for `finish()` - use finish instead."""
         deprecate.deprecate(
             field_name=deprecate.Deprecated.run__join,
             warning_message=(
@@ -1915,7 +1915,7 @@ class Run:
         fields: Dict[str, Any],
         string_fields: Optional[Dict[str, Any]] = None,
     ) -> CustomChart:
-        """Creates a custom plot on a table.
+        """Create a custom plot on a table.
 
         Arguments:
             vega_spec_name: the name of the spec for the plot
@@ -1962,7 +1962,7 @@ class Run:
             console = self._settings._console
         # only use raw for service to minimize potential changes
         if console == SettingsConsole.WRAP:
-            if self._settings._require_service:
+            if not self._settings._disable_service:
                 console = SettingsConsole.WRAP_RAW
             else:
                 console = SettingsConsole.WRAP_EMU
@@ -2129,7 +2129,6 @@ class Run:
             self._output_writer = None
 
     def _on_init(self) -> None:
-
         if self._backend and self._backend.interface:
             logger.info("communicating current version")
             version_handle = self._backend.interface.deliver_check_version(
@@ -2501,6 +2500,7 @@ class Run:
     def unwatch(self, models=None) -> None:  # type: ignore
         wandb.unwatch(models=models)
 
+    # TODO(kdg): remove all artifact swapping logic
     def _swap_artifact_name(self, artifact_name: str, use_as: Optional[str]) -> str:
         artifact_key_string = use_as or artifact_name
         replacement_artifact_info = self._launch_artifact_mapping.get(
@@ -2516,10 +2516,6 @@ class Run:
                 )
             return f"{entity}/{project}/{new_name}"
         elif replacement_artifact_info is None and use_as is None:
-            wandb.termwarn(
-                f"Could not find {artifact_name} in launch artifact mapping. "
-                f"Searching for unique artifacts with sequence name: {artifact_name}"
-            )
             sequence_name = artifact_name.split(":")[0].split("/")[-1]
             unique_artifact_replacement_info = (
                 self._unique_launch_artifact_sequence_names.get(sequence_name)
@@ -2535,14 +2531,8 @@ class Run:
                 return f"{entity}/{project}/{new_name}"
 
         else:
-            wandb.termwarn(
-                f"Could not find swappable artifact at key: {use_as}. Using {artifact_name}"
-            )
             return artifact_name
 
-        wandb.termwarn(
-            f"Could not find {artifact_key_string} in launch artifact mapping. Using {artifact_name}"
-        )
         return artifact_name
 
     def _detach(self) -> None:
@@ -2555,7 +2545,7 @@ class Run:
         target_path: str,
         aliases: Optional[List[str]] = None,
     ) -> None:
-        """Links the given artifact to a portfolio (a promoted collection of artifacts).
+        """Link the given artifact to a portfolio (a promoted collection of artifacts).
 
         The linked artifact will be visible in the UI for the specified portfolio.
 
@@ -2692,7 +2682,7 @@ class Run:
             else:
                 raise ValueError(
                     'You must pass an artifact name (e.g. "pedestrian-dataset:v1"), '
-                    "an instance of `wandb.Artifact`, or `wandb.Api().artifact()` to `use_artifact`"  # noqa: E501
+                    "an instance of `wandb.Artifact`, or `wandb.Api().artifact()` to `use_artifact`"
                 )
         if self._backend and self._backend.interface:
             self._backend.interface.publish_use_artifact(artifact)
@@ -3011,7 +3001,7 @@ class Run:
 
     @_run_decorator._attach
     def mark_preempting(self) -> None:
-        """Marks this run as preempting.
+        """Mark this run as preempting.
 
         Also tells the internal process to immediately report this to server.
         """
@@ -3045,7 +3035,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         if not check_version or settings._offline:
             return
 
@@ -3080,7 +3069,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         # printer = printer or get_printer(settings._jupyter)
         if settings._offline:
             printer.display(
@@ -3104,7 +3092,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         if settings._offline or settings.silent:
             return
 
@@ -3120,13 +3107,11 @@ class Run:
         # printer = printer or get_printer(settings._jupyter)
         if printer._html:
             if not wandb.jupyter.maybe_display():
-
                 run_line = f"<strong>{printer.link(run_url, run_name)}</strong>"
                 project_line, sweep_line = "", ""
 
                 # TODO(settings): make settings the source of truth
                 if not wandb.jupyter.quiet():
-
                     doc_html = printer.link(wburls.get("doc_run"), "docs")
 
                     project_html = printer.link(project_url, "Weights & Biases")
@@ -3224,7 +3209,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         if settings.silent:
             return
 
@@ -3320,7 +3304,6 @@ class Run:
         *,
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         # todo: is this same as settings._offline?
         if not all(poll_exit_responses):
             return
@@ -3374,7 +3357,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         if settings.silent:
             return
 
@@ -3410,7 +3392,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         if (quiet or settings.quiet) or settings.silent:
             return
 
@@ -3430,7 +3411,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         if (quiet or settings.quiet) or settings.silent:
             return
 
@@ -3502,7 +3482,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         if (quiet or settings.quiet) or settings.silent:
             return
 
@@ -3531,7 +3510,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         if (quiet or settings.quiet) or settings.silent:
             return
 
@@ -3555,7 +3533,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         if not check_version:
             return
 
@@ -3584,7 +3561,6 @@ class Run:
         settings: "Settings",
         printer: Union["PrinterTerm", "PrinterJupyter"],
     ) -> None:
-
         if (quiet or settings.quiet) or settings.silent:
             return
 
@@ -3615,7 +3591,7 @@ def restore(
     replace: bool = False,
     root: Optional[str] = None,
 ) -> Union[None, TextIO]:
-    """Downloads the specified file from cloud storage.
+    """Download the specified file from cloud storage.
 
     File is placed into the current directory or run directory.
     By default, will only download the file if it doesn't already exist.
@@ -3673,7 +3649,7 @@ except AttributeError:
 
 
 def finish(exit_code: Optional[int] = None, quiet: Optional[bool] = None) -> None:
-    """Marks a run as finished, and finishes uploading all data.
+    """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.
@@ -3686,25 +3662,33 @@ def finish(exit_code: Optional[int] = None, quiet: Optional[bool] = None) -> Non
         wandb.run.finish(exit_code=exit_code, quiet=quiet)
 
 
-class _LazyArtifact(ArtifactInterface):
+class InvalidArtifact:
+    """An "artifact" that raises an error when any properties are accessed."""
+
+    def __init__(self, base_artifact: "ArtifactInterface"):
+        super().__setattr__("base_artifact", base_artifact)
+
+    def __getattr__(self, __name: str) -> Any:
+        raise ArtifactNotLoggedError(artifact=self.base_artifact, attr=__name)
 
+    def __setattr__(self, __name: str, __value: Any) -> None:
+        raise ArtifactNotLoggedError(artifact=self.base_artifact, attr=__name)
+
+    def __bool__(self) -> bool:
+        return False
+
+
+class _LazyArtifact(ArtifactInterface):
     _api: PublicApi
-    _instance: Optional[ArtifactInterface] = None
+    _instance: Union[ArtifactInterface, InvalidArtifact]
     _future: Any
 
     def __init__(self, api: PublicApi, future: Any):
         self._api = api
+        self._instance = InvalidArtifact(self)
         self._future = future
 
-    def _assert_instance(self) -> ArtifactInterface:
-        if not self._instance:
-            raise ValueError(
-                "Must call wait() before accessing logged artifact properties"
-            )
-        return self._instance
-
     def __getattr__(self, item: str) -> Any:
-        self._assert_instance()
         return getattr(self._instance, item)
 
     def wait(self, timeout: Optional[int] = None) -> ArtifactInterface:
@@ -3727,131 +3711,101 @@ class _LazyArtifact(ArtifactInterface):
 
     @property
     def id(self) -> Optional[str]:
-        return self._assert_instance().id
+        return self._instance.id
 
     @property
     def source_version(self) -> Optional[str]:
-        return self._assert_instance().source_version
+        return self._instance.source_version
 
     @property
     def version(self) -> str:
-        return self._assert_instance().version
+        return self._instance.version
 
     @property
     def name(self) -> str:
-        return self._assert_instance().name
+        return self._instance.name
 
     @property
     def type(self) -> str:
-        return self._assert_instance().type
+        return self._instance.type
 
     @property
     def entity(self) -> str:
-        return self._assert_instance().entity
+        return self._instance.entity
 
     @property
     def project(self) -> str:
-        return self._assert_instance().project
+        return self._instance.project
 
     @property
     def manifest(self) -> "ArtifactManifest":
-        return self._assert_instance().manifest
+        return self._instance.manifest
 
     @property
     def digest(self) -> str:
-        return self._assert_instance().digest
+        return self._instance.digest
 
     @property
     def state(self) -> str:
-        return self._assert_instance().state
+        return self._instance.state
 
     @property
     def size(self) -> int:
-        return self._assert_instance().size
+        return self._instance.size
 
     @property
     def commit_hash(self) -> str:
-        return self._assert_instance().commit_hash
+        return self._instance.commit_hash
 
     @property
     def description(self) -> Optional[str]:
-        return self._assert_instance().description
+        return self._instance.description
 
     @description.setter
     def description(self, desc: Optional[str]) -> None:
-        self._assert_instance().description = desc
+        self._instance.description = desc
 
     @property
     def metadata(self) -> dict:
-        return self._assert_instance().metadata
+        return self._instance.metadata
 
     @metadata.setter
     def metadata(self, metadata: dict) -> None:
-        self._assert_instance().metadata = metadata
+        self._instance.metadata = metadata
 
     @property
     def aliases(self) -> List[str]:
-        return self._assert_instance().aliases
+        return self._instance.aliases
 
     @aliases.setter
     def aliases(self, aliases: List[str]) -> None:
-        self._assert_instance().aliases = aliases
+        self._instance.aliases = aliases
 
     def used_by(self) -> List["wandb.apis.public.Run"]:
-        return self._assert_instance().used_by()
+        return self._instance.used_by()
 
     def logged_by(self) -> "wandb.apis.public.Run":
-        return self._assert_instance().logged_by()
-
-    # Commenting this block out since this code is unreachable since LocalArtifact
-    # overrides them and therefore untestable.
-    # Leaving behind as we may want to support these in the future.
-
-    # def new_file(self, name: str, mode: str = "w") -> Any:  # TODO: Refine Type
-    #     return self._assert_instance().new_file(name, mode)
-
-    # def add_file(
-    #     self,
-    #     local_path: str,
-    #     name: Optional[str] = None,
-    #     is_tmp: Optional[bool] = False,
-    # ) -> Any:  # TODO: Refine Type
-    #     return self._assert_instance().add_file(local_path, name, is_tmp)
-
-    # def add_dir(self, local_path: str, name: Optional[str] = None) -> None:
-    #     return self._assert_instance().add_dir(local_path, name)
-
-    # def add_reference(
-    #     self,
-    #     uri: Union["ArtifactManifestEntry", str],
-    #     name: Optional[str] = None,
-    #     checksum: bool = True,
-    #     max_objects: Optional[int] = None,
-    # ) -> Any:  # TODO: Refine Type
-    #     return self._assert_instance().add_reference(uri, name, checksum, max_objects)
-
-    # def add(self, obj: "WBValue", name: str) -> Any:  # TODO: Refine Type
-    #     return self._assert_instance().add(obj, name)
+        return self._instance.logged_by()
 
     def get_path(self, name: str) -> "ArtifactManifestEntry":
-        return self._assert_instance().get_path(name)
+        return self._instance.get_path(name)
 
     def get(self, name: str) -> "WBValue":
-        return self._assert_instance().get(name)
+        return self._instance.get(name)
 
     def download(
         self, root: Optional[str] = None, recursive: bool = False
     ) -> util.FilePathStr:
-        return self._assert_instance().download(root, recursive)
+        return self._instance.download(root, recursive)
 
     def checkout(self, root: Optional[str] = None) -> str:
-        return self._assert_instance().checkout(root)
+        return self._instance.checkout(root)
 
     def verify(self, root: Optional[str] = None) -> Any:
-        return self._assert_instance().verify(root)
+        return self._instance.verify(root)
 
     def save(self) -> None:
-        return self._assert_instance().save()
+        self._instance.save()
 
     def delete(self) -> None:
-        return self._assert_instance().delete()
+        self._instance.delete()