wandb 0.20.0__py3-none-any.whl → 0.20.1__py3-none-any.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.4
 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=U9047nyMDICgctm1HLm4HfXwFnFKsEn2m77hsYPUZ1I,4298
-wandb/__init__.py,sha256=Q0K6cKpsIpkEcwlfEagt0w2vTeQpORSgjX-51m9GzL4,7072
-wandb/__init__.pyi,sha256=kZo96SFeW-OLP6hIGhyi7224rxhs_79UxnRET5v3lk4,48123
+wandb/__init__.py,sha256=voeAUjikukV1wUbzWpSRKezgf6AEE6NODckD2qsuHCU,7072
+wandb/__init__.pyi,sha256=nWUVGrUbymxXWlYwbkoakcN5EbXF-15ZJkXThTGKXTg,48123
 wandb/__main__.py,sha256=gripuDgB7J8wMMeJt4CIBRjn1BMSFr5zvsrt585Pnj4,64
 wandb/_iterutils.py,sha256=NX6MaIE3JzPXsqKNXUKdnpIlX0spGG8HwGIQt5VASrk,2400
 wandb/data_types.py,sha256=M-wqAO0FtIqvj28556u3h4nzSwlRcbxhXoX0B9jlJSo,2283
@@ -325,7 +325,7 @@ wandb/sdk/data_types/bokeh.py,sha256=RuMsi7v_yJyMEuYjfrwTL1q1B8svdCW9t_eNge9aEJA
 wandb/sdk/data_types/graph.py,sha256=_P_z2dNj2mCJeSvusIfVHkSkVB9AL0AEO-6KaguC1ys,12089
 wandb/sdk/data_types/histogram.py,sha256=syx3fUDoP6vYqA3r5m7bHXK6MAWo2MkRhpCKasihwvg,3145
 wandb/sdk/data_types/html.py,sha256=HTh2nNlJsncLgou1gXz6e-f_loaHInMEJerzlDLqKzc,4381
-wandb/sdk/data_types/image.py,sha256=Al6Jtgm8_uDcSy9JnQf0L3MSmYb72Ep4u8eKQVxO0Io,34314
+wandb/sdk/data_types/image.py,sha256=xdYth860StgcqoUJ0o4NPncj83XB5BRWI7ypY5PbrUY,34811
 wandb/sdk/data_types/molecule.py,sha256=_eVzkFEO-uydch8dsw6GjaW_hNGrZwUFfU_inMj70gM,8549
 wandb/sdk/data_types/object_3d.py,sha256=83ordMmzmmFun2fp2z9P3dffNJ0fW_Hv7MfpDZROYcc,16282
 wandb/sdk/data_types/plotly.py,sha256=LEIHRqtHHpSou04AEUsyH8_UMXP4adrEHHaKoEtQkxo,2913
@@ -906,8 +906,8 @@ wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/importlib2.py,sha256=cJIaJ2EQso
 wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/platform.py,sha256=UORYTNVcUSE2NpFcq9UVLIS-tsS0TS_Qw8akhKxn2eY,1506
 wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/unicode_paths.py,sha256=UWX8DB97ygkEeSxWQUYCHR4MahNilux7vl5TCTQtPPk,2190
 wandb/vendor/watchdog_0_9_0/wandb_watchdog/utils/win32stat.py,sha256=ZOevOTbSo8NRiIxkuBVGaG4yigWnPoO0goxAi-jsBkM,3828
-wandb-0.20.0.dist-info/METADATA,sha256=7aUxNBHbW0VdFy4_MAq7vLOn0usyft9ttAQ0J2QWucI,10307
-wandb-0.20.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-wandb-0.20.0.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
-wandb-0.20.0.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081
-wandb-0.20.0.dist-info/RECORD,,
+wandb-0.20.1.dist-info/METADATA,sha256=EcYDqPeeoZYy-r-YORdy-49B7a9AswwzGfIU1lA61-k,10307
+wandb-0.20.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+wandb-0.20.1.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
+wandb-0.20.1.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081
+wandb-0.20.1.dist-info/RECORD,,