pep8

minimaxir · Apr 30, 2018 · f81a51a · f81a51a
1 parent 5b7a92c
commit f81a51a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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()