updated reqs

tests/test_models.py

@@ -7,6 +7,7 @@ import numpy as np
 import warnings
 import torch
 import os
+import shutil
 
 SEED = 2334
 torch.manual_seed(SEED)
@@ -40,7 +41,7 @@ def clear_tt_dir():
     root_dir = os.path.dirname(os.path.realpath(__file__))
     tt_dir = os.path.join(root_dir, 'tests_tt_dir')
     if os.path.exists(tt_dir):
-        os.rmdir(tt_dir)
+        shutil.rmtree(tt_dir)
 
 
 def assert_ok_acc(trainer):