nshtrainer 0.14.0__tar.gz → 0.14.2__tar.gz

{nshtrainer-0.14.0 → nshtrainer-0.14.2}/PKG-INFO

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

{nshtrainer-0.14.0 → nshtrainer-0.14.2}/pyproject.toml

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

{nshtrainer-0.14.0 → nshtrainer-0.14.2}/src/nshtrainer/runner.py

@@ -3,10 +3,9 @@ import functools
 from collections.abc import Callable, Iterable, Mapping, Sequence
 from typing import Generic
 
-from nshrunner import RunInfo
+from nshrunner import RunInfo, Snapshot
 from nshrunner import Runner as _Runner
 from nshrunner._submit import screen
-from nshrunner.snapshot import SnapshotArgType
 from typing_extensions import TypeVar, TypeVarTuple, Unpack, override
 
 from .model.config import BaseConfig
@@ -93,7 +92,7 @@ class Runner(
         options: screen.ScreenJobKwargs = {},
         n_batches: int = 1,
         *,
-        snapshot: SnapshotArgType,
+        snapshot: Snapshot,
         setup_commands: Sequence[str] | None = None,
         env: Mapping[str, str] | None = None,
         transforms: list[

{nshtrainer-0.14.0 → nshtrainer-0.14.2}/src/nshtrainer/trainer/trainer.py

@@ -244,7 +244,7 @@ class Trainer(LightningTrainer):
             log.critical(f"Disabling logger because {config.trainer.logging.enabled=}.")
             kwargs["logger"] = False
         else:
-            _update_kwargs(logger=config.trainer.logging.create_loggers(config))
+            _update_kwargs(logger=list(config.trainer.logging.create_loggers(config)))
 
         if config.trainer.auto_determine_num_nodes:
             # When num_nodes is auto, we need to detect the number of nodes.