Add requirements and travis config

bartpy/data.py

@@ -1,5 +1,5 @@
 from collections import namedtuple
-from typing import Any, MutableMapping, Set, Optional
+from typing import Any, Set
 
 import pandas as pd
 import numpy as np

bartpy/mutation.py

@@ -1,12 +1,4 @@
-from abc import abstractmethod, ABC
-from typing import List, Set, Generator, Optional, Union
-
-import numpy as np
-import pandas as pd
-
-from bartpy.data import Data
-from bartpy.errors import NoSplittableVariableException, NoPrunableNodeException
-from bartpy.split import Split, sample_split_condition, SplitCondition
+from abc import ABC
 from bartpy.node import TreeNode, DecisionNode, LeafNode
 
 

bartpy/node.py

@@ -1,5 +1,4 @@
 from abc import abstractmethod, ABC
-from typing import List, Set, Generator, Optional, Union
 
 import numpy as np
 import pandas as pd

bartpy/proposer.py

@@ -1,5 +1,4 @@
 from abc import abstractclassmethod
-from copy import deepcopy
 from typing import Callable
 
 import numpy as np

bartpy/sampler.py

@@ -5,7 +5,7 @@
 
 from bartpy.model import Model
 from bartpy.mutation import TreeMutation, GrowMutation, ChangeMutation, PruneMutation
-from bartpy.node import DecisionNode, LeafNode, TreeNode
+from bartpy.node import LeafNode, TreeNode
 from bartpy.proposer import Proposer
 from bartpy.sigma import Sigma
 from bartpy.tree import Tree, n_splittable_leaf_nodes, n_prunable_decision_nodes, mutate

bartpy/split.py

@@ -1,5 +1,5 @@
 from abc import ABC
-from typing import List, Optional, Union
+from typing import List, Optional
 from copy import deepcopy
 
 from bartpy.data import Data

requirements.txt

@@ -0,0 +1,4 @@
+numpy
+pandas
+matplotlib
+scipy

travis.yaml

@@ -0,0 +1,7 @@
+language: python
+python:
+  - "3.6"
+script:
+  - pytest
+install:
+  - pip install -r requirements.txt