For support different python versions (#1828)

* change, and be compatible with different python versions

* update by comments

src/sdk/pynni/nni/feature_engineering/__init__.py

+

src/sdk/pynni/nni/feature_engineering/gradient_selector/fginitialize.py

@@ -33,8 +33,8 @@ from torch.utils.data import DataLoader, Dataset
 # pylint: disable=E0611
 from torch.utils.data.dataloader import _DataLoaderIter, _utils
 
-import nni.feature_engineering.gradient_selector.constants as constants
-import nni.feature_engineering.gradient_selector.syssettings as syssettings
+from . import constants
+from . import syssettings
 
 torch.set_default_tensor_type(syssettings.torch.tensortype)
 sparsetensor = syssettings.torch.sparse.tensortype

src/sdk/pynni/nni/feature_engineering/gradient_selector/fgtrain.py

@@ -26,10 +26,10 @@ import torch
 from sklearn.feature_selection import SelectKBest, \
     f_classif, mutual_info_classif, f_regression, mutual_info_regression
 
-import nni.feature_engineering.gradient_selector.constants as constants
-import nni.feature_engineering.gradient_selector.syssettings as syssettings
-from nni.feature_engineering.gradient_selector.learnability import Solver
-from nni.feature_engineering.gradient_selector.utils import EMA
+from . import constants
+from . import syssettings
+from .learnability import Solver
+from .utils import EMA
 
 torch.set_default_tensor_type(syssettings.torch.tensortype)
 

src/sdk/pynni/nni/feature_engineering/gradient_selector/gradient_selector.py

@@ -30,9 +30,9 @@ from sklearn.utils.validation import check_is_fitted
 import torch
 
 from nni.feature_engineering.feature_selector import FeatureSelector
-import nni.feature_engineering.gradient_selector.constants as constants
-from nni.feature_engineering.gradient_selector.fginitialize import PrepareData
-from nni.feature_engineering.gradient_selector.fgtrain import _train
+from . import constants
+from .fginitialize import PrepareData
+from .fgtrain import _train
 
 
 class FeatureGradientSelector(FeatureSelector, BaseEstimator, SelectorMixin):

src/sdk/pynni/nni/feature_engineering/gradient_selector/learnability.py

@@ -25,9 +25,9 @@ import scipy.special
 import torch
 import torch.nn as nn
 
-import nni.feature_engineering.gradient_selector.constants as constants
-import nni.feature_engineering.gradient_selector.syssettings as syssettings
-from nni.feature_engineering.gradient_selector.fginitialize import ChunkDataLoader
+from . import constants
+from . import syssettings
+from .fginitialize import ChunkDataLoader
 
 torch.set_default_tensor_type(syssettings.torch.tensortype)
 sparsetensor = syssettings.torch.sparse.tensortype