Skip to content

Commit 7c3470e

Browse files
author
Jan Michelfeit
committed
#625 fix even even more pre-commit errors
1 parent 412550d commit 7c3470e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/imitation/algorithms/preference_comparisons.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,9 @@ def has_pretraining(self) -> bool:
8585
8686
By default, True is returned if the unsupervised_pretrain() method is not
8787
overriden, bud subclasses may choose to override this behavior.
88+
89+
Returns:
90+
True if this generator has a pre-training phase, False otherwise
8891
"""
8992
orig_impl = TrajectoryGenerator.unsupervised_pretrain
9093
return type(self).unsupervised_pretrain != orig_impl
@@ -105,7 +108,7 @@ def unsupervised_pretrain(self, steps: int, **kwargs: Any) -> None:
105108
f"{steps} timesteps allocated for unsupervised pre-training:"
106109
" Trajectory generators without pre-training implementation should"
107110
" not consume any timesteps (otherwise the total number of"
108-
" timesteps executed may be misleading)"
111+
" timesteps executed may be misleading)",
109112
)
110113

111114
def train(self, steps: int, **kwargs: Any) -> None:

0 commit comments

Comments
 (0)