codacy fixes

hackingmaterials · Feb 2, 2019 · 42918d2 · 42918d2
1 parent a957d67
commit 42918d2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/automatminer/analytics/core.py b/automatminer/analytics/core.py
@@ -1,8 +1,6 @@
-import os
 import warnings
 
 import matplotlib.pyplot as plt
-from matminer.datasets import load_dataset
 
 try:
     from skater.core.explanations import Interpretation

diff --git a/automatminer/featurization/sets.py b/automatminer/featurization/sets.py
@@ -15,7 +15,7 @@
 try:
     import torch
     import cgcnn
-except:
+except ImportError:
     torch, cgcnn = None, None
 
 __authors__ = ["Alex Dunn", "Alex Ganose"]

diff --git a/automatminer/pipeline.py b/automatminer/pipeline.py
@@ -217,7 +217,7 @@ def benchmark(self, df, target, kfold, fold_subset=None):
                                 "may be extended!")
             results = []
             fold = 0
-            for train_ix, test_ix in kfold.split(df):
+            for _, test_ix in kfold.split(df):
                 if fold in fold_subset:
                     self.logger.info("Training on fold index {}".format(fold))
                     # Split and identify test set
@@ -240,7 +240,7 @@ def benchmark(self, df, target, kfold, fold_subset=None):
                              "dataframe for benchmark.")
             df = self.autofeaturizer.fit_transform(df, target)
 
-            for train_ix, test_ix in kfold.split(df):
+            for _, test_ix in kfold.split(df):
                 if fold in fold_subset:
                     self.logger.info("Training on fold index {}".format(fold))