Add compatibility for python 3.6+

src/model.py

@@ -58,7 +58,7 @@ class ModelArguments:
     )
 
 
-@lru_cache
+@lru_cache(maxsize=None)
 def get_classifier_vectorizer(classifier_args):
     classifier_path = os.path.join(
         classifier_args.classifier_dir, classifier_args.classifier_file)

src/utils.py

@@ -157,7 +157,10 @@ class InterruptibleTaskPool:
             t = time()
             # Send SIGINT if process doesn't exit quickly enough, and kill it as last resort
             # .is_alive() also implicitly joins the process (good practice in linux)
-            while alive_procs := [p for p in procs if p.is_alive()]:
+            while True:
+                alive_procs = [p for p in procs if p.is_alive()]
+                if not alive_procs:
+                    break
                 if time() > t + self.grace_period:
                     for p in alive_procs:
                         os.kill(p.pid, signal.SIGINT)