Linting

SeldonIO · May 24, 2019 · 258b376 · 258b376
1 parent 44d7a27
commit 258b376
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alibi/explainers/counterfactual.py b/alibi/explainers/counterfactual.py
@@ -253,7 +253,7 @@ def __init__(self,
 
         # return templates
         self.instance_dict = dict.fromkeys(['X', 'distance', 'lambda', 'index', 'class', 'proba', 'loss'])
-        self.return_dict = {'cf': None, 'all': {i:[] for i in range(self.max_lam_steps)}, 'orig_class': None,
+        self.return_dict = {'cf': None, 'all': {i: [] for i in range(self.max_lam_steps)}, 'orig_class': None,
                             'orig_proba': None}  # type: dict
 
     def _initialize(self, X: np.ndarray) -> np.ndarray:
@@ -302,7 +302,7 @@ def explain(self, X: np.ndarray) -> Dict:
 
         return_dict = self.return_dict.copy()
         self.instance_dict = dict.fromkeys(['X', 'distance', 'lambda', 'index', 'class', 'proba', 'loss'])
-        self.return_dict = {'cf': None, 'all': {i:[] for i in range(self.max_lam_steps)}, 'orig_class': None,
+        self.return_dict = {'cf': None, 'all': {i: [] for i in range(self.max_lam_steps)}, 'orig_class': None,
                             'orig_proba': None}
 
         return return_dict