Merge pull request #68 from NatLabRockies/bnb/obs_model_4dim_fix

Bnb/obs model 4dim fix

Author: bnb32

phygnn/__init__.py

@@ -6,11 +6,10 @@
 from tensorflow.keras.utils import get_custom_objects
 
 from ._version import __version__
-
 from .base import CustomNetwork, GradientUtils
 from .layers import HiddenLayers, Layers
 from .layers.custom_layers import GaussianAveragePooling2D
-from .model_interfaces import PhygnnModel, RandomForestModel, TfModel
+from .model_interfaces import PhygnnModel, TfModel
 from .phygnn import PhysicsGuidedNeuralNetwork
 from .utilities import PreProcess, tf_isin, tf_log10
 

phygnn/model_interfaces/__init__.py

@@ -3,6 +3,6 @@
 operations, e.g. on-the-fly normalization during forward pass,
 one-hot-encoding, and saving/loading models.
 """
+
 from .phygnn_model import PhygnnModel
-from .random_forest_model import RandomForestModel
 from .tf_model import TfModel