wandb 0.17.5__py3-none-any.whl → 0.17.6__py3-none-any.whl

wandb/__init__.py

@@ -11,7 +11,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.5"
+__version__ = "0.17.6"
 
 
 # Used with pypi checks and other messages related to pip
@@ -74,6 +74,7 @@ from wandb.data_types import Video
 from wandb.data_types import Audio
 from wandb.data_types import Table
 from wandb.data_types import Html
+from wandb.data_types import box3d
 from wandb.data_types import Object3D
 from wandb.data_types import Molecule
 from wandb.data_types import Histogram
@@ -245,6 +246,7 @@ __all__ = (
     "Audio",
     "Table",
     "Html",
+    "box3d",
     "Object3D",
     "Molecule",
     "Histogram",

wandb/apis/public/api.py

@@ -821,7 +821,7 @@ class Api:
             order: (str) Order can be `created_at`, `heartbeat_at`, `config.*.value`, or `summary_metrics.*`.
                 If you prepend order with a + order is ascending.
                 If you prepend order with a - order is descending (default).
-                The default order is run.created_at from newest to oldest.
+                The default order is run.created_at from oldest to newest.
             per_page: (int) Sets the page size for query pagination.
             include_sweeps: (bool) Whether to include the sweep runs in the results.
 

wandb/apis/public/jobs.py

@@ -65,6 +65,7 @@ class Job:
         self._entrypoint = source_info.get("entrypoint")
         self._dockerfile = source_info.get("dockerfile")
         self._build_context = source_info.get("build_context")
+        self._base_image = source_info.get("base_image")
         self._args = source_info.get("args")
         self._partial = self._job_info.get("_partial", False)
         self._requirements_file = os.path.join(self._fpath, "requirements.frozen.txt")
@@ -131,6 +132,8 @@ class Job:
             launch_project.set_job_dockerfile(self._dockerfile)
         if self._build_context:
             launch_project.set_job_build_context(self._build_context)
+        if self._base_image:
+            launch_project.set_job_base_image(self._base_image)
 
     def _configure_launch_project_artifact(self, launch_project):
         artifact_string = self._job_info.get("source", {}).get("artifact")
@@ -152,6 +155,8 @@ class Job:
             launch_project.set_job_dockerfile(self._dockerfile)
         if self._build_context:
             launch_project.set_job_build_context(self._build_context)
+        if self._base_image:
+            launch_project.set_job_base_image(self._base_image)
 
     def _configure_launch_project_container(self, launch_project):
         launch_project.docker_image = self._job_info.get("source", {}).get("image")

wandb/data_types.py

@@ -44,7 +44,7 @@ from .sdk.data_types.histogram import Histogram
 from .sdk.data_types.html import Html
 from .sdk.data_types.image import Image
 from .sdk.data_types.molecule import Molecule
-from .sdk.data_types.object_3d import Object3D
+from .sdk.data_types.object_3d import Object3D, box3d
 from .sdk.data_types.plotly import Plotly
 from .sdk.data_types.saved_model import _SavedModel
 from .sdk.data_types.trace_tree import WBTraceTree
@@ -65,6 +65,7 @@ __all__ = [
     "Html",
     "Image",
     "Molecule",
+    "box3d",
     "Object3D",
     "Plotly",
     "Video",

wandb/env.py

@@ -161,6 +161,12 @@ def is_debug(default: Optional[str] = None, env: Optional[Env] = None) -> bool:
     return _env_as_bool(DEBUG, default=default, env=env)
 
 
+def is_offline(env: Optional[Env] = None) -> bool:
+    if env is None:
+        env = os.environ
+    return env.get(MODE) == "offline"
+
+
 def error_reporting_enabled() -> bool:
     return _env_as_bool(ERROR_REPORTING, default="True")
 

wandb/integration/lightning/fabric/logger.py

@@ -8,7 +8,7 @@ from typing_extensions import override
 
 import wandb
 from wandb import Artifact
-from wandb.sdk.lib import RunDisabled, telemetry
+from wandb.sdk.lib import telemetry
 from wandb.sdk.wandb_run import Run
 
 try:
@@ -293,7 +293,7 @@ class WandbLogger(Logger):
         anonymous: Optional[bool] = None,
         project: Optional[str] = None,
         log_model: Union[Literal["all"], bool] = False,
-        experiment: Union["Run", "RunDisabled", None] = None,
+        experiment: Optional["Run"] = None,
         prefix: str = "",
         checkpoint_name: Optional[str] = None,
         log_checkpoint_on: Union[Literal["success"], Literal["all"]] = "success",
@@ -360,7 +360,7 @@ class WandbLogger(Logger):
 
     @property
     @rank_zero_experiment
-    def experiment(self) -> Union["Run", "RunDisabled"]:
+    def experiment(self) -> "Run":
         r"""Actual wandb object.
 
         To use wandb features in your :class:`~lightning.pytorch.core.LightningModule`, do the
@@ -393,7 +393,7 @@ class WandbLogger(Logger):
                 self._experiment = wandb.init(**self._wandb_init)
 
                 # define default x-axis
-                if isinstance(self._experiment, (Run, RunDisabled)) and getattr(
+                if isinstance(self._experiment, Run) and getattr(
                     self._experiment, "define_metric", None
                 ):
                     self._experiment.define_metric("trainer/global_step")