nshtrainer 0.5.1__tar.gz → 0.5.2__tar.gz

{nshtrainer-0.5.1 → nshtrainer-0.5.2}/PKG-INFO

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

{nshtrainer-0.5.1 → nshtrainer-0.5.2}/pyproject.toml

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

{nshtrainer-0.5.1 → nshtrainer-0.5.2}/src/nshtrainer/runner.py

@@ -1,9 +1,9 @@
 import copy
 import functools
-from collections.abc import Mapping, Sequence
+from collections.abc import Callable, Iterable, Mapping, Sequence
 from typing import Generic
 
-from nshrunner import RunInfo, RunList, TransformFn
+from nshrunner import RunInfo
 from nshrunner import Runner as _Runner
 from nshrunner.snapshot import SnapshotArgType
 from typing_extensions import TypeVar, TypeVarTuple, Unpack, override
@@ -46,11 +46,14 @@ class Runner(
 
     def fast_dev_run(
         self,
-        runs: RunList[TConfig, Unpack[TArguments]],
+        runs: Iterable[tuple[TConfig, Unpack[TArguments]]],
         n_batches: int = 1,
         *,
         env: Mapping[str, str] | None = None,
-        transforms: list[TransformFn[TConfig, Unpack[TArguments]]] | None = None,
+        transforms: list[
+            Callable[[TConfig, Unpack[TArguments]], tuple[TConfig, Unpack[TArguments]]]
+        ]
+        | None = None,
     ):
         transforms = transforms or []
         transforms.append(
@@ -64,11 +67,14 @@ class Runner(
 
     def fast_dev_run_generator(
         self,
-        runs: RunList[TConfig, Unpack[TArguments]],
+        runs: Iterable[tuple[TConfig, Unpack[TArguments]]],
         n_batches: int = 1,
         *,
         env: Mapping[str, str] | None = None,
-        transforms: list[TransformFn[TConfig, Unpack[TArguments]]] | None = None,
+        transforms: list[
+            Callable[[TConfig, Unpack[TArguments]], tuple[TConfig, Unpack[TArguments]]]
+        ]
+        | None = None,
     ):
         transforms = transforms or []
         transforms.append(
@@ -82,13 +88,16 @@ class Runner(
 
     def fast_dev_run_session(
         self,
-        runs: RunList[TConfig, Unpack[TArguments]],
+        runs: Iterable[tuple[TConfig, Unpack[TArguments]]],
         n_batches: int = 1,
         *,
         snapshot: SnapshotArgType,
         setup_commands: Sequence[str] | None = None,
         env: Mapping[str, str] | None = None,
-        transforms: list[TransformFn[TConfig, Unpack[TArguments]]] | None = None,
+        transforms: list[
+            Callable[[TConfig, Unpack[TArguments]], tuple[TConfig, Unpack[TArguments]]]
+        ]
+        | None = None,
         activate_venv: bool = True,
         session_name: str = "nshrunner",
         attach: bool = True,