diff --git a/data/gan.py b/data/gan.py index 0d449d2..621cea9 100644 --- a/data/gan.py +++ b/data/gan.py @@ -609,7 +609,7 @@ class Predict(GNet): # let's get the missing rows (if any) ... # ii = df.apply(lambda row: np.sum(row) == 0 ,axis=1) - if ii : + if ii.shape[0] == 0 : # #@TODO Have this be a configurable variable missing = np.repeat(0, np.where(ii==1)[0].size)