Fix linting problems

lm_eval/tasks/race.py

@@ -23,7 +23,8 @@ class RACE(HFTask):
         return True
 
     def _collate_data(self, set):
-        if set in self.cache: return self.cache[set]
+        if set in self.cache:
+            return self.cache[set]
         # One big issue with HF's implementation of this dataset: it makes a
         # separate document for each question; meanwhile, in the GPT3 paper it
         # is shown that one document is made per passage.

lm_eval/utils_stream.py

@@ -5,11 +5,13 @@ from tqdm import tqdm
 import json
 
 
-class ExitCodeError(Exception): pass
+class ExitCodeError(Exception):
+    pass
 
 
 def sh(x):
-    if os.system(x): raise ExitCodeError()
+    if os.system(x):
+        raise ExitCodeError()
 
 def ls(x):
     return [x + '/' + fn for fn in os.listdir(x)]
@@ -64,7 +66,8 @@ class join:
         self.sep = sep
 
     def __rrshift__(self, other):
-        if other is None: return
+        if other is None:
+            return
         try:
             return self.sep.join(other)
         except: