Handle pylint error

CCSI-Toolset · Oct 10, 2023 · ff2eb1f · ff2eb1f
1 parent 88dea3a
commit ff2eb1f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/foqus_lib/framework/surrogate/pytorch_nn.py b/foqus_lib/framework/surrogate/pytorch_nn.py
@@ -52,7 +52,7 @@
 from multiprocessing.connection import Client
 
 import random as rn
-import torch
+import torch  # pylint: disable=import-error
 import torch.nn as nn  # pylint: disable=import-error
 
 # custom class to define Keras NN layers

diff --git a/foqus_lib/framework/surrogate/scikit_nn.py b/foqus_lib/framework/surrogate/scikit_nn.py
@@ -52,7 +52,7 @@
 from multiprocessing.connection import Client
 
 import random as rn
-from sklearn.neural_network import MLPRegressor
+from sklearn.neural_network import MLPRegressor  # pylint: disable=import-error
 import pickle
 from types import SimpleNamespace