From cdb787dcc48ebaedcb9d557ae66a770ccdfbaf1d Mon Sep 17 00:00:00 2001 From: youliang Date: Fri, 24 May 2024 18:51:21 -0700 Subject: [PATCH] single liner precommit-nit Signed-off-by: youliang --- serl_launcher/serl_launcher/networks/reward_classifier.py | 1 + 1 file changed, 1 insertion(+) diff --git a/serl_launcher/serl_launcher/networks/reward_classifier.py b/serl_launcher/serl_launcher/networks/reward_classifier.py index b2109ff..6cdf3c9 100644 --- a/serl_launcher/serl_launcher/networks/reward_classifier.py +++ b/serl_launcher/serl_launcher/networks/reward_classifier.py @@ -12,6 +12,7 @@ from serl_launcher.common.encoding import EncodingWrapper from flax.core.frozen_dict import freeze, unfreeze + class BinaryClassifier(nn.Module): encoder_def: nn.Module hidden_dim: int = 256