From ff2eb1fef9e2d90dacec507632d42e1c72e34476 Mon Sep 17 00:00:00 2001 From: franflame Date: Tue, 10 Oct 2023 10:14:03 -0700 Subject: [PATCH] Handle pylint error --- foqus_lib/framework/surrogate/pytorch_nn.py | 2 +- foqus_lib/framework/surrogate/scikit_nn.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/foqus_lib/framework/surrogate/pytorch_nn.py b/foqus_lib/framework/surrogate/pytorch_nn.py index 0402d4d1f..ccd8f21eb 100644 --- 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 index d3207c304..1f9bc885e 100644 --- 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