wandb 0.20.0__py3-none-macosx_11_0_arm64.whl → 0.20.1__py3-none-macosx_11_0_arm64.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,15 +1,15 @@
 package_readme.md,sha256=U9047nyMDICgctm1HLm4HfXwFnFKsEn2m77hsYPUZ1I,4298
-wandb-0.20.0.dist-info/RECORD,,
-wandb-0.20.0.dist-info/WHEEL,sha256=B_HTF0nESsbkeA2UeB8YgKrtPrxyKFORYURx7Qxu8Xw,101
-wandb-0.20.0.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
-wandb-0.20.0.dist-info/METADATA,sha256=54GhP4I8qsLq7vu3FjlR_sxav0qC3_sCSraynNk2kSg,10307
-wandb-0.20.0.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081
+wandb-0.20.1.dist-info/RECORD,,
+wandb-0.20.1.dist-info/WHEEL,sha256=B_HTF0nESsbkeA2UeB8YgKrtPrxyKFORYURx7Qxu8Xw,101
+wandb-0.20.1.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
+wandb-0.20.1.dist-info/METADATA,sha256=86RafFiYjMqMeso0ph6f-0hXGZoVAQUy9jUW33HH-EE,10307
+wandb-0.20.1.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081
 wandb/env.py,sha256=G9VGOMvSPAAmKxIfTrJuYs1fVtHvPqnalg7KjJ7fSFY,14139
-wandb/__init__.pyi,sha256=kZo96SFeW-OLP6hIGhyi7224rxhs_79UxnRET5v3lk4,48123
+wandb/__init__.pyi,sha256=nWUVGrUbymxXWlYwbkoakcN5EbXF-15ZJkXThTGKXTg,48123
 wandb/util.py,sha256=fCPwfepeZVKoR9eVgEntygISAmP6Mz5q7KknhF7cn50,63114
 wandb/wandb_run.py,sha256=CNh9S6uubFk8FphQjzkbvedyyGCN9aBEsRBKjy8tqqs,155
 wandb/_iterutils.py,sha256=NX6MaIE3JzPXsqKNXUKdnpIlX0spGG8HwGIQt5VASrk,2400
-wandb/__init__.py,sha256=Q0K6cKpsIpkEcwlfEagt0w2vTeQpORSgjX-51m9GzL4,7072
+wandb/__init__.py,sha256=voeAUjikukV1wUbzWpSRKezgf6AEE6NODckD2qsuHCU,7072
 wandb/data_types.py,sha256=M-wqAO0FtIqvj28556u3h4nzSwlRcbxhXoX0B9jlJSo,2283
 wandb/wandb_controller.py,sha256=SksJdgwn14PpnUoIaBjJ9Ki4Nksl9BpQGGn42hT0xZg,24936
 wandb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -103,7 +103,7 @@ wandb/proto/v3/wandb_base_pb2.py,sha256=_nsr_HW4Fdz62-KiXGo6Dw0_9bwdXz07auZkkH2Q
 wandb/proto/v3/wandb_internal_pb2.py,sha256=xKTg9CalkBOznziFa5fA9kirA9kqg72QbEANoMvcSDI,118397
 wandb/proto/v3/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/proto/v3/wandb_telemetry_pb2.py,sha256=IOk3VMF7uOKOpl8C5163Vuzxt32xbR_ph_2TsKJU3NQ,14350
-wandb/bin/wandb-core,sha256=7yJWXx7zH9cSSvi7Rc2q9qa7R3sW33KUbkFcNruxKxQ,56596866
+wandb/bin/wandb-core,sha256=fylEgIC7Evu_MpIko6DMXNdX-Kamwx50NkEC96cVstQ,56596866
 wandb/bin/gpu_stats,sha256=Kewea_STGOMcTnHflaerBOPvsTM2-2LUw4FM_QrSdpI,9146736
 wandb/integration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/integration/yolov8/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -307,7 +307,7 @@ wandb/sdk/data_types/bokeh.py,sha256=RuMsi7v_yJyMEuYjfrwTL1q1B8svdCW9t_eNge9aEJA
 wandb/sdk/data_types/saved_model.py,sha256=OKk1IB3q1HC-VRQMD2CTNP1h_Ca_15NRAMY5LXLjO7o,16310
 wandb/sdk/data_types/plotly.py,sha256=LEIHRqtHHpSou04AEUsyH8_UMXP4adrEHHaKoEtQkxo,2913
 wandb/sdk/data_types/video.py,sha256=JQ9ZJ-IqocQ7fI9iGswG_JwB0gUk89JXgMSUOemVak8,9835
-wandb/sdk/data_types/image.py,sha256=Al6Jtgm8_uDcSy9JnQf0L3MSmYb72Ep4u8eKQVxO0Io,34314
+wandb/sdk/data_types/image.py,sha256=xdYth860StgcqoUJ0o4NPncj83XB5BRWI7ypY5PbrUY,34811
 wandb/sdk/data_types/_private.py,sha256=zp2NRarTlIq4Hk3R2xp7j_qPGNzBMnaGHrZUN82shaY,299
 wandb/sdk/data_types/base_types/media.py,sha256=U-JmWYqd2uykHTMjPi_BdL53NMf3kgdlbpr-Dg_z_3Y,14510
 wandb/sdk/data_types/base_types/json_metadata.py,sha256=oKpimndUQvDW30n15tb2pbyyyKhhWwm8wsYNBEot060,1553