import tensorflow as tf

import time
from itertools import product, combinations
from mpl_toolkits.mplot3d import Axes3D
from mpl_toolkits.mplot3d.art3d import Poly3DCollection
from plotting import newfig, savefig
from mpl_toolkits.axes_grid1 import make_axes_locatable
import matplotlib.gridspec as gridspec

np.random.seed(1234)
tf.set_random_seed(1234)

class PhysicsInformedNN:
    from setup_PINN_ns import __init__ 
    from initialize_PINN_ns import initialize_NN
    from xavier_init_ns import xavier_init
    from def_NN_ns import neural_net
    from def_Net_NS import net_NS 
    from func_call_ns import callback
