fix linter errors

Author: zStupan

niaarm/__init__.py

@@ -1,3 +1,8 @@
 from niaarm.niaarm import NiaARM
+from niaarm.dataset import Dataset
+from niaarm.stats import Stats
+
+
+__all__ = [NiaARM, Dataset, Stats]
 
 __version__ = "0.1.0"

niaarm/tests/test_cut_point.py

@@ -137,7 +137,7 @@ def test_cut_pointB(self):
 
         self.assertEqual(antecedent, [[0.2620357326, 0.4989950842], [0.5636729279999999, 1.13]])
         self.assertEqual(consequent, ['NO', 'NO', 'NO', 'NO', [0.34108412769999996, 0.56784007355], ['I'],
-                                       [0.13678483190000001, 0.44964727704]])
+                                      [0.13678483190000001, 0.44964727704]])
 
     def test_cut_pointC(self):
         arm = AssociationRule(self.features)
@@ -235,4 +235,4 @@ def test_cut_pointC(self):
 
         self.assertEqual(antecedent, [[0.2620357326, 0.4989950842], [0.5636729279999999, 1.13], 'NO', 'NO'])
         self.assertEqual(consequent, ['NO', 'NO', [0.34108412769999996, 0.56784007355], ['I'],
-                                       [0.13678483190000001, 0.44964727704]])
+                                      [0.13678483190000001, 0.44964727704]])