Cleanup imports

tests/test_models.py

@@ -1,13 +1,12 @@
 import itertools
-from inspect import getmembers, isclass
 
 import numpy as np
 import pytest
 import tensorflow as tf
 from keras.engine.functional import Functional
 from numpy.testing import assert_array_equal
 
-from mlpp_lib import models, probabilistic_layers
+from mlpp_lib import models
 
 
 FCN_OPTIONS = dict(