Merge remote-tracking branch 'origin/main'

LarsKue · Mar 23, 2023 · ad125e9 · ad125e9
2 parents 99c7d62 + 310c8d7
commit ad125e9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/lightning_trainable/__init__.py b/src/lightning_trainable/__init__.py
@@ -4,5 +4,5 @@
 from . import modules
 from . import trainable
 
-from .trainable import Trainable, TrainableHParams
+from .trainable import Trainable, TrainableHParams, SkipBatch
 from .hparams import HParams
diff --git a/src/lightning_trainable/trainable/__init__.py b/src/lightning_trainable/trainable/__init__.py
@@ -1 +1 @@
-from .trainable import Trainable, TrainableHParams
+from .trainable import Trainable, TrainableHParams, SkipBatch
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		from .trainable import Trainable, TrainableHParams
		from .trainable import Trainable, TrainableHParams, SkipBatch