Revert back to using numpy

Jax reports errors, type mismatch

Numpy issue may be able to be resolve here: tensorflow/models#9706

	modified:   astronet/metrics.py

astronet/metrics.py

@@ -3,7 +3,6 @@
 import pandas as pd
 import sys
 import tensorflow as tf
-import jax.numpy as jnp
 
 from tensorflow import keras
 
@@ -86,7 +85,7 @@ def mywloss(y_true, y_pred):
         Where:
         wtable - is a numpy 1d array with (the number of times class y_true occur in the data set)/(size of data set)
         """
-        wtable = jnp.sum(y_true, axis=0) / y_true.shape[0]
+        wtable = np.sum(y_true, axis=0) / y_true.shape[0]
 
         yc = tf.clip_by_value(y_pred, 1e-15, 1 - 1e-15)