nshtrainer 0.10.17__tar.gz → 0.10.19__tar.gz

{nshtrainer-0.10.17 → nshtrainer-0.10.19}/PKG-INFO

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

{nshtrainer-0.10.17 → nshtrainer-0.10.19}/pyproject.toml

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

{nshtrainer-0.10.17 → nshtrainer-0.10.19}/src/nshtrainer/_checkpoint/metadata.py

@@ -92,7 +92,7 @@ def _write_checkpoint_metadata(
     except Exception as e:
         log.warning(f"Failed to write metadata to {checkpoint_path}: {e}")
     else:
-        log.info(f"Checkpoint metadata written to {checkpoint_path}")
+        log.debug(f"Checkpoint metadata written to {checkpoint_path}")
 
     # Write the hparams to the checkpoint directory
     try:
@@ -101,7 +101,7 @@ def _write_checkpoint_metadata(
     except Exception as e:
         log.warning(f"Failed to write hparams to {checkpoint_path}: {e}")
     else:
-        log.info(f"Checkpoint metadata written to {checkpoint_path}")
+        log.debug(f"Checkpoint metadata written to {checkpoint_path}")
 
 
 def _remove_checkpoint_metadata(checkpoint_path: Path):
@@ -112,7 +112,7 @@ def _remove_checkpoint_metadata(checkpoint_path: Path):
         except Exception as e:
             log.warning(f"Failed to remove {path}: {e}")
         else:
-            log.info(f"Removed {path}")
+            log.debug(f"Removed {path}")
 
 
 def _link_checkpoint_metadata(checkpoint_path: Path, linked_checkpoint_path: Path):
@@ -133,7 +133,7 @@ def _link_checkpoint_metadata(checkpoint_path: Path, linked_checkpoint_path: Pat
         except Exception as e:
             log.warning(f"Failed to link {path} to {linked_path}: {e}")
         else:
-            log.info(f"Linked {path} to {linked_path}")
+            log.debug(f"Linked {path} to {linked_path}")
 
 
 def _checkpoint_sort_key_fn(key: Callable[[CheckpointMetadata, Path], Any]):

{nshtrainer-0.10.17 → nshtrainer-0.10.19}/src/nshtrainer/model/config.py

@@ -315,7 +315,7 @@ class WandbLoggerConfig(CallbackConfigBase, BaseLoggerConfig):
                 if pkg_resources.parse_version(
                     wandb.__version__
                 ) < pkg_resources.parse_version("0.17.5"):
-                    raise ValueError(
+                    log.warning(
                         "The version of WandB installed does not support the `wandb-core` backend "
                         f"(expected version >= 0.17.5, found version {wandb.__version__}). "
                         "Please either upgrade to a newer version of WandB or disable the `use_wandb_core` option."