From f81a51a764a19959a038c222f2fadba9c1efebcb Mon Sep 17 00:00:00 2001 From: Max Woolf Date: Sun, 29 Apr 2018 20:27:48 -0700 Subject: [PATCH] pep8 --- textgenrnn/textgenrnn.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/textgenrnn/textgenrnn.py b/textgenrnn/textgenrnn.py index 0094c27..5b73147 100644 --- a/textgenrnn/textgenrnn.py +++ b/textgenrnn/textgenrnn.py @@ -225,7 +225,9 @@ def save(self, weights_path="textgenrnn_weights_saved.hdf5"): self.model.save_weights(weights_path) def load(self, weights_path): - self.model = textgenrnn_model(self.num_classes, cfg=self.config, weights_path=weights_path) + self.model = textgenrnn_model(self.num_classes, + cfg=self.config, + weights_path=weights_path) def reset(self): self.config = self.default_config.copy()