wandb 0.20.0__py3-none-win32.whl → 0.20.1__py3-none-win32.whl

wandb/__init__.py

@@ -10,7 +10,7 @@ For reference documentation, see https://docs.wandb.com/ref/python.
 """
 from __future__ import annotations
 
-__version__ = "0.20.0"
+__version__ = "0.20.1"
 
 
 from wandb.errors import Error

wandb/__init__.pyi

@@ -106,7 +106,7 @@ if TYPE_CHECKING:
     import wandb
     from wandb.plot import CustomChart
 
-__version__: str = "0.20.0"
+__version__: str = "0.20.1"
 
 run: Run | None
 config: wandb_config.Config

wandb/sdk/data_types/image.py

@@ -58,25 +58,39 @@ def _warn_on_invalid_data_range(
         )
 
 
-def _normalize(data: "np.ndarray") -> "np.ndarray":
-    """Normalizes and converts image pixel values to uint8 in the range [0, 255]."""
-    np = util.get_module(
-        "numpy",
-        required="wandb.Image requires numpy if not supplying PIL Images: pip install numpy",
-    )
+def _guess_and_rescale_to_0_255(data: "np.ndarray") -> "np.ndarray":
+    """Guess the image's format and rescale its values to the range [0, 255].
+
+    This is an unfortunate design flaw carried forward for backward
+    compatibility. A better design would have been to document the expected
+    data format and not mangle the data provided by the user.
+
+    If given data in the range [0, 1], we multiply all values by 255
+    and round down to get integers.
+
+    If given data in the range [-1, 1], we rescale it by mapping -1 to 0 and
+    1 to 255, then round down to get integers.
+
+    We clip and round all other data.
+    """
+    try:
+        import numpy as np
+    except ImportError:
+        raise wandb.Error(
+            "wandb.Image requires numpy if not supplying PIL images: pip install numpy"
+        ) from None
 
-    # if an image has negative values, set all values to be in the range [0, 1]
-    # This can lead to inconsistent behavior when an image has only a single negative value
-    if np.min(data) < 0:
-        data = data - np.min(data)
+    data_min: float = data.min()
+    data_max: float = data.max()
 
-    if np.ptp(data) != 0:
-        data = data / np.ptp(data)
+    if 0 <= data_min and data_max <= 1:
+        return (data * 255).astype(np.uint8)
 
-    if np.max(data) <= 1.0:
-        data = (255 * data).astype(np.int32)
+    elif -1 <= data_min and data_max <= 1:
+        return (255 * 0.5 * (data + 1)).astype(np.uint8)
 
-    return data.clip(0, 255)
+    else:
+        return data.clip(0, 255).astype(np.uint8)
 
 
 def _convert_to_uint8(data: "np.ndarray") -> "np.ndarray":
@@ -393,7 +407,7 @@ class Image(BatchableMedia):
 
             _warn_on_invalid_data_range(data, normalize)
 
-            data = _normalize(data) if normalize else data  # type: ignore [arg-type]
+            data = _guess_and_rescale_to_0_255(data) if normalize else data  # type: ignore [arg-type]
             data = _convert_to_uint8(data)
 
             if data.ndim > 2:
@@ -413,7 +427,7 @@ class Image(BatchableMedia):
             _warn_on_invalid_data_range(data, normalize)  # type: ignore [arg-type]
 
             mode = mode or self.guess_mode(data, file_type)
-            data = _normalize(data) if normalize else data  # type: ignore [arg-type]
+            data = _guess_and_rescale_to_0_255(data) if normalize else data  # type: ignore [arg-type]
             data = _convert_to_uint8(data)  # type: ignore [arg-type]
             self._image = pil_image.fromarray(
                 data,

{wandb-0.20.0.dist-info → wandb-0.20.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: wandb
-Version: 0.20.0
+Version: 0.20.1
 Summary: A CLI and library for interacting with the Weights & Biases API.
 Project-URL: Source, https://github.com/wandb/wandb
 Project-URL: Bug Reports, https://github.com/wandb/wandb/issues

{wandb-0.20.0.dist-info → wandb-0.20.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 package_readme.md,sha256=XGlaq8rMFcoBb21rCr2d5qeSM79ZI4WslLmXqRimTGQ,4395
-wandb/__init__.py,sha256=vkJpQB53unFHVkg9O3j1cHd-IJdfWGaviwchAlYfwpA,7321
-wandb/__init__.pyi,sha256=1MjC035r5bv9JFNtsGs31WVSmFXuFYci0kbsG2keM-c,49375
+wandb/__init__.py,sha256=vvYTPs3lKelQTsFwnHU_3Axn6V_EqLAq7IvadREhS8Y,7321
+wandb/__init__.pyi,sha256=AnFHXNQ-qSVPWRju2psy8Kl9n7iX8tUSioHEhWAPujQ,49375
 wandb/__main__.py,sha256=uHY6OxHT6RtTH34zC8_UC1GsCTkndgbdsHXv-t7dOMI,67
 wandb/_iterutils.py,sha256=_ZUvfdQ8YeKrT_0HKaAKjkVyLHQdTYRcCh0l-zGxTMM,2465
 wandb/data_types.py,sha256=DdCkf7Dh_j86Q74FWzh3M20EW_hzKpNagexjo03qv-A,2349
@@ -86,8 +86,8 @@ wandb/automations/_generated/operations.py,sha256=V0n3PZ_e754Ym8IraYjFzLWg0wqnA_
 wandb/automations/_generated/slack_integrations_by_entity.py,sha256=95llPwkGxKrwX4R6ze9IxzmMuyLpIWoperLUy-hKsE0,558
 wandb/automations/_generated/update_automation.py,sha256=9SvGNXYK7fEZdhztonzRcMiJ8Yqps-42hKB16CUOFYI,357
 wandb/beta/workflows.py,sha256=t35UZZ0fovG8qmE0wIINKpU0ynHHXQ2DjE7XdzWD__4,10648
-wandb/bin/gpu_stats.exe,sha256=mRrKMPkznAz1oLK-1uaLLl9wij1c0WaPCqzEv0RVyTI,8159232
-wandb/bin/wandb-core,sha256=fcZMk-hjT2g6H0qqh2ir_lk5IWwiIMV2DsjBJ8bwHk8,58521600
+wandb/bin/gpu_stats.exe,sha256=DKxl5tCYGwH0gWrUKbY_j_dT_7THYMLgqxNBmn2XOc4,8159232
+wandb/bin/wandb-core,sha256=-n93BcaIvaZRQMFRnTqYknHu7-fhhUCwvTkrXv2dKTs,58521600
 wandb/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/cli/beta.py,sha256=_wbqhk6J2yjnNuVQAGlYbYZ9-j0sFE73zMIXVkQYlXA,5619
 wandb/cli/cli.py,sha256=wGObFtUSgm_IyesGiuAKRt3nmY0lwN3QD_-OTCkyt4M,94802
@@ -326,7 +326,7 @@ wandb/sdk/data_types/bokeh.py,sha256=ykxVAha5whA7uQyrYYHalS4NXha5tcFgOVRxFXUqCwo
 wandb/sdk/data_types/graph.py,sha256=eHCo-Svlzvihf_nri_50rcD-NohqHuE80PR-otiueYQ,12494
 wandb/sdk/data_types/histogram.py,sha256=Icb6ox2dXP_OjaXjU6rZzkk8AfRQevQE9BHdQiedELo,3239
 wandb/sdk/data_types/html.py,sha256=eMgZyJ5VYcT6AH2gb34CA09Y0s3ul8en_FD48nFRU6M,4523
-wandb/sdk/data_types/image.py,sha256=EhjfERF4r9Bizppnn8Tm_faWUqRqSW_caF1uhrr4w7M,35238
+wandb/sdk/data_types/image.py,sha256=lRaEChoi9ZGL3iQK-XDbR1WIZIP2_Oz9RiNnVMrsbmg,35749
 wandb/sdk/data_types/molecule.py,sha256=JY3Hr-gI9scBk82UVTMdoZHWiAgrpL7tzDSROvCpPww,8786
 wandb/sdk/data_types/object_3d.py,sha256=6zrw8cdunfdlVNKWZpPOQDd0TGaWovRrAS_5h-GRTGs,16758
 wandb/sdk/data_types/plotly.py,sha256=UNYwRUUCODSyGvD3WEiF0p6jEseNfYiGnm8mD_tOoiI,2995
@@ -907,8 +907,8 @@ wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/importlib2.py,sha256=kX0rdVmTDL
 wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/platform.py,sha256=7fpTDfxSYvSRtHvyog-plRdLR5A6k1QVY_AL0gVhhPM,1563
 wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/unicode_paths.py,sha256=xzyQmuba2gns1s3Qemu9SXaKV5zeTL3TP9--xOi541g,2254
 wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/win32stat.py,sha256=R48kuuEIi7XzCJBJ6Xo7v6DJIbOP5EwcsWaPf5Axn_g,3951
-wandb-0.20.0.dist-info/METADATA,sha256=54GhP4I8qsLq7vu3FjlR_sxav0qC3_sCSraynNk2kSg,10307
-wandb-0.20.0.dist-info/WHEEL,sha256=F7luzo77gHneG2SMTaD6iIlqNT8oVz1uotn5Zigp4o8,89
-wandb-0.20.0.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
-wandb-0.20.0.dist-info/licenses/LICENSE,sha256=rJ7p1acqNi17WFOAJ9WqsImXZtKZDA3i_gzdDVGRuFQ,1102
-wandb-0.20.0.dist-info/RECORD,,
+wandb-0.20.1.dist-info/METADATA,sha256=86RafFiYjMqMeso0ph6f-0hXGZoVAQUy9jUW33HH-EE,10307
+wandb-0.20.1.dist-info/WHEEL,sha256=F7luzo77gHneG2SMTaD6iIlqNT8oVz1uotn5Zigp4o8,89
+wandb-0.20.1.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
+wandb-0.20.1.dist-info/licenses/LICENSE,sha256=rJ7p1acqNi17WFOAJ9WqsImXZtKZDA3i_gzdDVGRuFQ,1102
+wandb-0.20.1.dist-info/RECORD,,