kaiko-eva 0.3.2__py3-none-any.whl → 0.3.3__py3-none-any.whl

eva/core/data/dataloaders/dataloader.py

@@ -77,7 +77,9 @@ class DataLoader:
             shuffle=self.shuffle,
             sampler=sampler or self.sampler,
             batch_sampler=self.batch_sampler,
-            num_workers=self.num_workers or multiprocessing.cpu_count(),
+            num_workers=(
+                multiprocessing.cpu_count() if self.num_workers is None else self.num_workers
+            ),
             collate_fn=self.collate_fn,
             pin_memory=self.pin_memory,
             drop_last=self.drop_last,

eva/core/loggers/log/__init__.py

@@ -2,5 +2,6 @@
 
 from eva.core.loggers.log.image import log_image
 from eva.core.loggers.log.parameters import log_parameters
+from eva.core.loggers.log.table import log_table
 
-__all__ = ["log_image", "log_parameters"]
+__all__ = ["log_image", "log_parameters", "log_table"]

eva/core/loggers/log/table.py

@@ -0,0 +1,73 @@
+"""Table log functionality."""
+
+import functools
+from typing import List
+
+import pandas as pd
+
+from eva.core.loggers import loggers
+from eva.core.loggers.log import utils
+
+
+@functools.singledispatch
+def log_table(
+    logger,
+    tag: str,
+    columns: List[str] | None = None,
+    data: List[List[str]] | None = None,
+    dataframe: pd.DataFrame | None = None,
+    step: int = 0,
+) -> None:
+    """Adds a a table to the logger.
+
+    The table can be defined either with `columns` and `data` or with `dataframe`.
+
+    Args:
+        logger: The logger to log the table to.
+        tag: The log tag.
+        columns: The column names of the table.
+        data: The data of the table as a list of lists.
+        dataframe: A pandas DataFrame to log.
+        step: The global step of the log.
+    """
+    utils.raise_not_supported(logger, "table")
+
+
+@log_table.register
+def _(
+    loggers: list,
+    tag: str,
+    columns: List[str] | None = None,
+    data: List[List[str]] | None = None,
+    dataframe: pd.DataFrame | None = None,
+    step: int = 0,
+) -> None:
+    """Adds a table to a list of supported loggers."""
+    for logger in loggers:
+        log_table(
+            logger,
+            tag=tag,
+            columns=columns,
+            data=data,
+            dataframe=dataframe,
+            step=step,
+        )
+
+
+@log_table.register
+def _(
+    logger: loggers.WandbLogger,
+    tag: str,
+    columns: List[str] | None = None,
+    data: List[List[str]] | None = None,
+    dataframe: pd.DataFrame | None = None,
+    step: int = 0,
+) -> None:
+    """Adds a table to a Wandb logger."""
+    logger.log_table(
+        key=tag,
+        columns=columns,
+        data=data,
+        dataframe=dataframe,
+        step=step,
+    )

{kaiko_eva-0.3.2.dist-info → kaiko_eva-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kaiko-eva
-Version: 0.3.2
+Version: 0.3.3
 Summary: Evaluation Framework for oncology foundation models.
 Keywords: machine-learning,evaluation-framework,oncology,foundation-models
 Author-Email: Ioannis Gatopoulos <ioannis@kaiko.ai>, =?utf-8?q?Nicolas_K=C3=A4nzig?= <nicolas@kaiko.ai>, Roman Moser <roman@kaiko.ai>

{kaiko_eva-0.3.2.dist-info → kaiko_eva-0.3.3.dist-info}/RECORD

@@ -19,7 +19,7 @@ eva/core/data/__init__.py,sha256=yG3BeOWhp1EjVYMFqx8M_TBWFDyfIwwksQGQmMdSPaI,340
 eva/core/data/dataloaders/__init__.py,sha256=0AvpsPOdh4NX5rwkX9Th1M_rzxZVVzTPTdC5oTGFd5w,194
 eva/core/data/dataloaders/collate_fn/__init__.py,sha256=CfSAVrPD36shpyYAkob2ny05VWymb95MutawQcZkbuo,134
 eva/core/data/dataloaders/collate_fn/collate.py,sha256=oBdxaDCIaXBB6H8LB90Qsi2Inw1tyHGF4kAFBINPOeQ,689
-eva/core/data/dataloaders/dataloader.py,sha256=amRIwbmI6Hi2qS-ciKkv7ehgkBbMCHx7XUaEMlttWkw,2688
+eva/core/data/dataloaders/dataloader.py,sha256=v_UL5p78EiyLqXUhWsS0EbfW0iO-DAWOMv7-WTclOvA,2750
 eva/core/data/datamodules/__init__.py,sha256=qZchYbgxo9lxYnGoqdk0C6MfS2IbF0WItO0kCdP9Mqc,229
 eva/core/data/datamodules/call.py,sha256=jjj9w3UXYuQB-qyCcw1EZpRJW10OC1I3dvgvsuQWLck,940
 eva/core/data/datamodules/datamodule.py,sha256=_pK59oXDe53oDkmv6eoJUvfl44WlFkrbC8KXSRMs_20,5533
@@ -52,9 +52,10 @@ eva/core/interface/interface.py,sha256=EeBrIqUlO497BOZGZrtO-fZnc_BhIJrrqyQmRfqWX
 eva/core/loggers/__init__.py,sha256=4YMLNlN9LnuKqhBI1R1keh69dmMD-2lcH3HKwwyn380,266
 eva/core/loggers/dummy.py,sha256=Y7ypH0ecSAIkkZ5LzTmNNEzlKkqeaHfUNMCDKVOg6D4,1204
 eva/core/loggers/experimental_loggers.py,sha256=p5uCK_9QCYufRhE-LZQUJWbhGElyobX_zRM78yX4p2o,230
-eva/core/loggers/log/__init__.py,sha256=2nXYmR5_0XW0N8BcC918uvdvNyNgQLN_x1-papMprBk,189
+eva/core/loggers/log/__init__.py,sha256=TdsUwcaB2jW0--HVGIr7_polTfmr7iOmXtSmr1wK9_c,251
 eva/core/loggers/log/image.py,sha256=iUwntQCdRNLtkSdqu8CvV34l06zPYVo4NAW2gUeiJIM,1490
 eva/core/loggers/log/parameters.py,sha256=7Xi-I5gQvEVv71d58bwdZ-Hb4287NXxaUyMfriq_KDU,1634
+eva/core/loggers/log/table.py,sha256=HULCo5icDB6UOIXXMci0eo32Pl1-YRWzaOAB-ZBVUak,1726
 eva/core/loggers/log/utils.py,sha256=k4Q7uKpAQctfDv0EEYPnPv6wt9LnckEeqGvbYSLfKO0,415
 eva/core/loggers/loggers.py,sha256=igHxdxJSotWSg6nEOKnfFuBszzblHgi8T7sBrE00FEs,166
 eva/core/loggers/utils/wandb.py,sha256=GdwzEeFTAng5kl_kIVRxKL7rvwqyicQHSaZS8VSMXvU,747
@@ -275,8 +276,8 @@ eva/vision/utils/io/image.py,sha256=IdOkr5MYqhYHz8U9drZ7wULTM3YHwCWSjZlu_Qdl4GQ,
 eva/vision/utils/io/mat.py,sha256=qpGifyjmpE0Xhv567Si7-zxKrgkgE0sywP70cHiLFGU,808
 eva/vision/utils/io/nifti.py,sha256=TFMgNhLqIK3sl3RjIRXEABM7FmSQjqVOwk1vXkuvX2w,4983
 eva/vision/utils/io/text.py,sha256=qYgfo_ZaDZWfG02NkVVYzo5QFySqdCCz5uLA9d-zXtI,701
-kaiko_eva-0.3.2.dist-info/METADATA,sha256=3OdB75bdgEKDkAhIh75c3WcpevnOsemYcMfBEG0MKy8,25704
-kaiko_eva-0.3.2.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-kaiko_eva-0.3.2.dist-info/entry_points.txt,sha256=6CSLu9bmQYJSXEg8gbOzRhxH0AGs75BB-vPm3VvfcNE,88
-kaiko_eva-0.3.2.dist-info/licenses/LICENSE,sha256=e6AEzr7j_R-PYr2qLO-JwLn8y70jbVD3U2mxbRmwcI4,11338
-kaiko_eva-0.3.2.dist-info/RECORD,,
+kaiko_eva-0.3.3.dist-info/METADATA,sha256=3-qTJLt0hRJswZbPVixj9e9Bt0gFXDjOp6YbNf6Ohd4,25704
+kaiko_eva-0.3.3.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+kaiko_eva-0.3.3.dist-info/entry_points.txt,sha256=6CSLu9bmQYJSXEg8gbOzRhxH0AGs75BB-vPm3VvfcNE,88
+kaiko_eva-0.3.3.dist-info/licenses/LICENSE,sha256=e6AEzr7j_R-PYr2qLO-JwLn8y70jbVD3U2mxbRmwcI4,11338
+kaiko_eva-0.3.3.dist-info/RECORD,,