nshtrainer 0.40.0__tar.gz → 0.40.2__tar.gz

{nshtrainer-0.40.0 → nshtrainer-0.40.2}/PKG-INFO

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

{nshtrainer-0.40.0 → nshtrainer-0.40.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshtrainer"
-version = "0.40.0"
+version = "0.40.2"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"

{nshtrainer-0.40.0 → nshtrainer-0.40.2}/src/nshtrainer/loggers/wandb.py

@@ -92,7 +92,7 @@ class WandbLoggerConfig(CallbackConfigBase, BaseLoggerConfig):
     - "none" or False: Do not log any checkpoints
     """
 
-    log_code: WandbUploadCodeConfig | None = None
+    log_code: WandbUploadCodeConfig | None = WandbUploadCodeConfig()
     """WandB code upload configuration. Used to upload code to WandB."""
 
     watch: WandbWatchConfig | None = WandbWatchConfig()

{nshtrainer-0.40.0 → nshtrainer-0.40.2}/src/nshtrainer/model/config.py

@@ -77,6 +77,16 @@ class BaseConfig(C.Config):
         return self.directory.resolve_subdirectory(self.id, subdirectory)
 
     # region Helper methods
+    def fast_dev_run(self, value: int | bool, /):
+        """
+        Enables fast_dev_run mode for the trainer.
+        This will run the training loop for a specified number of batches,
+        if an integer is provided, or for a single batch if True is provided.
+        """
+        config = copy.deepcopy(self)
+        config.trainer.fast_dev_run = value
+        return config
+
     def with_project_root_(self, project_root: str | Path | os.PathLike) -> Self:
         """
         Set the project root directory for the trainer.