fix style checks and unit tests

Signed-off-by: Farhan Ahmed <[email protected]>

Author: f4str

art/estimators/certification/randomized_smoothing/pytorch.py

@@ -23,10 +23,9 @@
 from __future__ import absolute_import, division, print_function, unicode_literals
 
 import logging
-from typing import List, Optional, Tuple, Union, Any, TYPE_CHECKING
+from typing import List, Optional, Tuple, Union, TYPE_CHECKING
 
 import warnings
-import random
 from tqdm import tqdm
 import numpy as np
 

art/estimators/regression/pytorch.py

@@ -24,7 +24,6 @@
 import copy
 import logging
 import os
-import random
 import time
 from typing import Any, Dict, List, Optional, Tuple, Union, TYPE_CHECKING
 
@@ -375,7 +374,7 @@ def fit(  # pylint: disable=W0221
                     loss.backward()
 
                 self._optimizer.step()
-            
+
             if scheduler is not None:
                 scheduler.step()
 

tests/classifiersFrameworks/test_pytorch.py

@@ -253,7 +253,7 @@ def forward(self, x):
             n_samples=10000, n_features=20, n_informative=5, n_redundant=2, n_repeated=0, n_classes=2
         )
         train_x, test_x, train_y, test_y = sklearn.model_selection.train_test_split(x, y, test_size=0.2)
-        train_x = test_x.astype(np.float32)
+        train_x = train_x.astype(np.float32)
         train_y = train_y.astype(np.float32)
         test_x = test_x.astype(np.float32)
         model = BasicModel()