wandb 0.20.0__py3-none-musllinux_1_2_aarch64.whl → 0.20.1__py3-none-musllinux_1_2_aarch64.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,7 +1,7 @@
 package_readme.md,sha256=U9047nyMDICgctm1HLm4HfXwFnFKsEn2m77hsYPUZ1I,4298
 wandb/jupyter.py,sha256=ADLvYDYK4mSJrsxmwVuE6IBx5o8prFMQvCizm-13_bg,17350
 wandb/__main__.py,sha256=gripuDgB7J8wMMeJt4CIBRjn1BMSFr5zvsrt585Pnj4,64
-wandb/__init__.py,sha256=Q0K6cKpsIpkEcwlfEagt0w2vTeQpORSgjX-51m9GzL4,7072
+wandb/__init__.py,sha256=voeAUjikukV1wUbzWpSRKezgf6AEE6NODckD2qsuHCU,7072
 wandb/wandb_agent.py,sha256=kdlK_NZRZRNkjXRUha7Q-1fDOARf7v0TyIr_a-aRXKk,20868
 wandb/trigger.py,sha256=PaitU3sX6ekGkd2R8iD6d_VtI72ypF7LaPBXh3rXY7Q,615
 wandb/wandb_run.py,sha256=CNh9S6uubFk8FphQjzkbvedyyGCN9aBEsRBKjy8tqqs,155
@@ -12,7 +12,7 @@ wandb/_iterutils.py,sha256=NX6MaIE3JzPXsqKNXUKdnpIlX0spGG8HwGIQt5VASrk,2400
 wandb/wandb_controller.py,sha256=SksJdgwn14PpnUoIaBjJ9Ki4Nksl9BpQGGn42hT0xZg,24936
 wandb/data_types.py,sha256=M-wqAO0FtIqvj28556u3h4nzSwlRcbxhXoX0B9jlJSo,2283
 wandb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wandb/__init__.pyi,sha256=kZo96SFeW-OLP6hIGhyi7224rxhs_79UxnRET5v3lk4,48123
+wandb/__init__.pyi,sha256=nWUVGrUbymxXWlYwbkoakcN5EbXF-15ZJkXThTGKXTg,48123
 wandb/_pydantic/__init__.py,sha256=p1Rwdo5z3rgoQgVDz6pgcWzs5eitB2JlgVN6QeFyN0Y,662
 wandb/_pydantic/base.py,sha256=qOgf1LEMPO3o1kiYli9kOf2ZQLuP0GEMa8Qx2pDnJGY,4277
 wandb/_pydantic/utils.py,sha256=XcnRAEOdjANLxvr_FMnMPKmqv4f2HwjbNo-5J_JKKe0,2810
@@ -278,7 +278,7 @@ wandb/sdk/data_types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 wandb/sdk/data_types/saved_model.py,sha256=OKk1IB3q1HC-VRQMD2CTNP1h_Ca_15NRAMY5LXLjO7o,16310
 wandb/sdk/data_types/graph.py,sha256=_P_z2dNj2mCJeSvusIfVHkSkVB9AL0AEO-6KaguC1ys,12089
 wandb/sdk/data_types/table_decorators.py,sha256=2D0dlnv7sBvrHyYH61LCCMhowmg8CENqyvth9BID-VQ,3727
-wandb/sdk/data_types/image.py,sha256=Al6Jtgm8_uDcSy9JnQf0L3MSmYb72Ep4u8eKQVxO0Io,34314
+wandb/sdk/data_types/image.py,sha256=xdYth860StgcqoUJ0o4NPncj83XB5BRWI7ypY5PbrUY,34811
 wandb/sdk/data_types/trace_tree.py,sha256=i0LaWCnZsmTY560byG6_K4D9a6LTWDYJDKrnPe2YhwE,14838
 wandb/sdk/data_types/video.py,sha256=JQ9ZJ-IqocQ7fI9iGswG_JwB0gUk89JXgMSUOemVak8,9835
 wandb/sdk/data_types/object_3d.py,sha256=83ordMmzmmFun2fp2z9P3dffNJ0fW_Hv7MfpDZROYcc,16282
@@ -738,7 +738,7 @@ wandb/errors/util.py,sha256=YqO_fpZeXubXWBUAVQ6gmuLAiz3Q9pYVE5LbQ2Rq5Ck,1711
 wandb/errors/links.py,sha256=sNwJ74e9qb0w4GRZfnbPXK5ZdpIqc5lkuaT4T2Snnqw,2520
 wandb/errors/warnings.py,sha256=kyLP3bfXSmlztp8nOepLtfTdM-03N-i7Ho1Y568BOtk,57
 wandb/errors/term.py,sha256=kH_noQvl9BDsGBXm7fQfH5u0ERmI6bOZa8Rp49DoAPk,12207
-wandb/bin/wandb-core,sha256=ca2O7obBEckcsgOTgjuZmHhv0eFXz8U4SrSCktYiSMc,54591672
+wandb/bin/wandb-core,sha256=IlVdPAML3Y_pzmFa0-hIln3NLaflbnv6AZXxM1t32Ug,54591672
 wandb/bin/gpu_stats,sha256=KxttnJk3Rf3ytDyAD58578Alpqde9n42woJqifS0uuk,10416408
 wandb/apis/__init__.py,sha256=uPZKqlX8Y9YcHoERJshWRGS3ukMOIVWi4sDnkO5_CYY,1338
 wandb/apis/internal.py,sha256=Whl1x29iH_DdIu47vUVvDSyOavEVaovdmlqXmu4-yOw,7622
@@ -907,8 +907,8 @@ wandb/proto/v3/wandb_server_pb2.py,sha256=LQCUmuy0NFDo20N9uCUiE5VPyhP5r929WKtnia
 wandb/proto/v3/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wandb/proto/v3/wandb_settings_pb2.py,sha256=_zutPZ7EPxQjX5umK5bNRyyhnVxcBYJk0-GbZeS7HFY,21524
 wandb/proto/v3/wandb_internal_pb2.py,sha256=xKTg9CalkBOznziFa5fA9kirA9kqg72QbEANoMvcSDI,118397
-wandb-0.20.0.dist-info/RECORD,,
-wandb-0.20.0.dist-info/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
-wandb-0.20.0.dist-info/WHEEL,sha256=ZCaGMInsobf_X8ibOue5yAC0a46KTbFsnGjtmL2skyU,106
-wandb-0.20.0.dist-info/METADATA,sha256=7aUxNBHbW0VdFy4_MAq7vLOn0usyft9ttAQ0J2QWucI,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/entry_points.txt,sha256=v4FCOZ9gW7Pc6KLsmgQqpCiKTrA1wh2XHmNf-NUP1-I,67
+wandb-0.20.1.dist-info/WHEEL,sha256=ZCaGMInsobf_X8ibOue5yAC0a46KTbFsnGjtmL2skyU,106
+wandb-0.20.1.dist-info/METADATA,sha256=EcYDqPeeoZYy-r-YORdy-49B7a9AswwzGfIU1lA61-k,10307
+wandb-0.20.1.dist-info/licenses/LICENSE,sha256=izOKRJpGOx1PrJiGOKR0HsNdlB5JdH2d0Z4P7a7ssTc,1081