[plotting] used fixef numpy/one_hot

plotting.py

@@ -177,7 +177,7 @@ def figure_node_activations(
     for process in range(n_p):
         if disjoint:
             max_activations = np.argmax(activations, axis=-1)
-            one_hot_max_activations = one_hot(max_activations)
+            one_hot_max_activations = one_hot(max_activations, n=n_p)
             values = (activations * one_hot_max_activations)[truth[:, process]].swapaxes(0, 1)
             values[values == 0] = -10000
         else: