🎨 pep8 conforming fixes

whylabs · Feb 16, 2021 · 1b065c6 · 1b065c6
1 parent c1eee7c
commit 1b065c6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 7 deletions.
diff --git a/src/whylogs/core/datasetprofile.py b/src/whylogs/core/datasetprofile.py
@@ -4,7 +4,7 @@
 import datetime
 import io
 import logging
-from typing import Dict, List
+from typing import Dict
 from collections import OrderedDict
 from uuid import uuid4
 
@@ -176,7 +176,7 @@ def session_timestamp_ms(self):
         return time.to_utc_ms(self.session_timestamp)
 
     def add_output_field(self, field):
-        if isinstance(fields, list):
+        if isinstance(field, list):
             for field_name in field:
                 self.model_profile.add_output_field(field)
         else:

diff --git a/src/whylogs/core/metrics/confusion_matrix.py b/src/whylogs/core/metrics/confusion_matrix.py
@@ -131,7 +131,6 @@ def _merge_CM(old_conf_Matrix, new_conf_Matrix):
         old_conf_Matrix.labels, new_conf_Matrix.labels)
     old_indxes = enconde_to_integers(
         old_conf_Matrix.labels, old_conf_Matrix.labels)
-    confusion_matrix = new_conf_Matrix.confusion_matrix
 
     for old_row_idx, each_row_indx in enumerate(new_indxes):
         for old_column_idx, each_column_inx in enumerate(new_indxes):

diff --git a/src/whylogs/core/model_profile.py b/src/whylogs/core/model_profile.py
@@ -2,12 +2,9 @@
 
 from sklearn.utils.multiclass import type_of_target
 import numpy as np
-import pandas as pd
 
-from whylogs.proto import ModelProfileMessage
 
-from whylogs.core.metrics.confusion_matrix import ConfusionMatrix
-from whylogs.core import ColumnProfile
+from whylogs.proto import ModelProfileMessage
 from whylogs.core.metrics.model_metrics import ModelMetrics
 
 SUPPORTED_TYPES = ("binary", "multiclass")