🐛 drop tf.python imports (#117)

* drop `tf.python` imports

* do not use tf logging, but python logging

Author: MiWeiss

uncertainty_wizard/models/_stochastic/_sequential_stochastic.py

@@ -1,5 +1,6 @@
+import logging
+
 import tensorflow as tf
-from tensorflow.python.platform import tf_logging as logging
 
 from uncertainty_wizard.models._stochastic._abstract_stochastic import Stochastic
 from uncertainty_wizard.models._stochastic._stochastic_mode import StochasticMode

uncertainty_wizard/models/stochastic_utils/layers.py

@@ -5,7 +5,6 @@
 import warnings
 
 import tensorflow as tf
-from tensorflow.python.keras.layers import GaussianNoise
 
 from uncertainty_wizard.internal_utils import UncertaintyWizardWarning
 from uncertainty_wizard.models._stochastic._stochastic_mode import StochasticMode
@@ -215,7 +214,7 @@ def from_keras_layer(
 
     # docstr-coverage:inherited
     def get_config(self):
-        config = super(GaussianNoise, self).get_config()
+        config = super(tf.keras.layers.GaussianNoise, self).get_config()
         config["name"] = "UwGaussianNoise"
         # No custom config yet.
         return config