add set_seed to __init__.py (huggingface#127)

* add  to __init__.py

* update examples

* fix style

examples/sentiment/scripts/gpt2-sentiment.py

@@ -20,8 +20,8 @@
 from transformers import pipeline, AutoTokenizer
 from datasets import load_dataset
 
-from trl import PPOTrainer, PPOConfig, AutoModelForCausalLMWithValueHead
-from trl.core import LengthSampler, set_seed
+from trl import PPOTrainer, PPOConfig, AutoModelForCausalLMWithValueHead, set_seed
+from trl.core import LengthSampler
 
 ########################################################################
 # This is a fully working simple example to use trl with accelerate.

examples/sentiment/scripts/t5-sentiment.py

@@ -19,8 +19,8 @@
 from transformers import pipeline, AutoTokenizer
 from datasets import load_dataset
 
-from trl import PPOTrainer, PPOConfig, AutoModelForSeq2SeqLMWithValueHead
-from trl.core import LengthSampler, set_seed
+from trl import PPOTrainer, PPOConfig, AutoModelForSeq2SeqLMWithValueHead, set_seed
+from trl.core import LengthSampler
 
 ########################################################################
 # This is a fully working simple example to use trl with accelerate.

tests/trainer/test_ppo_trainer.py

@@ -97,7 +97,6 @@ def tearDownClass(cls):
                 pass
 
     def setUp(self):
-
         # model_id
         model_id = "gpt2"
 

trl/__init__.py

@@ -2,6 +2,7 @@
 
 __version__ = "0.2.2.dev0"
 
+from .core import set_seed
 from .models import (
     AutoModelForCausalLMWithValueHead,
     AutoModelForSeq2SeqLMWithValueHead,