nshtrainer 1.0.0b24__py3-none-any.whl → 1.0.0b25__py3-none-any.whl

nshtrainer/model/mixins/logger.py

@@ -64,6 +64,32 @@ class LoggerLightningModuleMixin(mixin_base_type(LightningModule)):
 
         self._logger_prefix_stack = deque[_LogContextKwargs]()
 
+    @property
+    def logging_enabled(self) -> bool:
+        # Logging is disabled in barebones mode.
+        if (trainer := self._trainer) is not None and trainer.barebones:
+            # Warn the user once that logging is disabled in barebones mode.
+            if not hasattr(self, "_barebones_logging_warned"):
+                rank_zero_warn(
+                    "Logging is disabled in barebones mode. "
+                    "This is to reduce the overhead of logging in barebones mode. "
+                    "If you want to enable logging, set `barebones=False` in the Trainer.",
+                )
+                self._barebones_logging_warned = True
+            return False
+
+        # If no loggers are registered, then logging is disabled.
+        if not self.logger:
+            return False
+
+        # Check if the topmost non-null context is disabled
+        for context in reversed(self._logger_prefix_stack):
+            if context.disabled is not None:
+                return not context.disabled
+
+        # Otherwise, logging is enabled.
+        return True
+
     @contextmanager
     def log_context(
         self,
@@ -121,22 +147,14 @@ class LoggerLightningModuleMixin(mixin_base_type(LightningModule)):
         metric_attribute: str | None = None,
         rank_zero_only: bool = False,
     ) -> None:
+        # If logging is disabled, then do nothing.
+        if not self.logging_enabled:
+            return
+
         # join all prefixes
         prefix = "".join(c.prefix for c in self._logger_prefix_stack if c.prefix)
         name = f"{prefix}{name}"
 
-        # check for disabled context:
-        # if the topmost non-null context is disabled, then we don't log
-        for c in reversed(self._logger_prefix_stack):
-            if c.disabled is not None:
-                if c.disabled:
-                    rank_zero_warn(
-                        f"Skipping logging of {name} due to disabled context"
-                    )
-                    return
-                else:
-                    break
-
         fn_kwargs = _LogContextKwargs()
         for c in self._logger_prefix_stack:
             fn_kwargs = fn_kwargs.copy_from(c)

{nshtrainer-1.0.0b24.dist-info → nshtrainer-1.0.0b25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshtrainer
-Version: 1.0.0b24
+Version: 1.0.0b25
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshtrainer-1.0.0b24.dist-info → nshtrainer-1.0.0b25.dist-info}/RECORD

@@ -106,7 +106,7 @@ nshtrainer/model/__init__.py,sha256=3G-bwPPSRStWdsdwG9-rn0bXcRpEiP1BiQpF_qavtls,
 nshtrainer/model/base.py,sha256=JL3AmH17GQjQIoMrZl3O0vUI7dj5ZsO5iEJgoLPyzHw,10356
 nshtrainer/model/mixins/callback.py,sha256=Ea_legORzs0N078j0N9RJivDVeWH5KtXDpdJS75IwIo,3098
 nshtrainer/model/mixins/debug.py,sha256=1LX9KzeFX9JDPs_a6YCdYDZXLhEk_5rBO2aCqlfBy7w,2087
-nshtrainer/model/mixins/logger.py,sha256=LQDJJbiv30PlWX6rTT_EhjNBNfUFfcvGz5sX4MnOCzI,5330
+nshtrainer/model/mixins/logger.py,sha256=27H99FuLaxc6_dDLG2pid4E_5E0-eLGnc2Ifpt0HYIM,6066
 nshtrainer/nn/__init__.py,sha256=sANhrZpeN5syLKOsmXMwhaFl2SBFPWcLaEe1EH22TWQ,1463
 nshtrainer/nn/mlp.py,sha256=2W8bzE96DzCMzGm6WPiPhNFQfhqaoG3GXPn_oKBnlUM,5988
 nshtrainer/nn/module_dict.py,sha256=9plb8aQUx5TUEPhX5jI9u8LrpTeKe7jZAHi8iIqcN8w,2365
@@ -135,6 +135,6 @@ nshtrainer/util/seed.py,sha256=diMV8iwBKN7Xxt5pELmui-gyqyT80_CZzomrWhNss0k,316
 nshtrainer/util/slurm.py,sha256=HflkP5iI_r4UHMyPjw9R4dD5AHsJUpcfJw5PLvGYBRM,1603
 nshtrainer/util/typed.py,sha256=Xt5fUU6zwLKSTLUdenovnKK0N8qUq89Kddz2_XeykVQ,164
 nshtrainer/util/typing_utils.py,sha256=MjY-CUX9R5Tzat-BlFnQjwl1PQ_W2yZQoXhkYHlJ_VA,442
-nshtrainer-1.0.0b24.dist-info/METADATA,sha256=kJJwb8hMUOHfif2qiEB70hCSllpUOt48s-n8SRFkRJY,937
-nshtrainer-1.0.0b24.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-1.0.0b24.dist-info/RECORD,,
+nshtrainer-1.0.0b25.dist-info/METADATA,sha256=ocf8xtYy2GBLpulyx5XepoiHYvjtHGnmMrLhDhvDQAQ,937
+nshtrainer-1.0.0b25.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-1.0.0b25.dist-info/RECORD,,