From dcfa044de1ab2cd0a320629dfcd759af9ffc5f82 Mon Sep 17 00:00:00 2001
From: davidADSP <david@adsp.ai>
Date: Sat, 15 Apr 2023 23:39:29 +0200
Subject: [PATCH] musegan formatting

---
 notebooks/11_music/02_musegan/musegan.ipynb | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/notebooks/11_music/02_musegan/musegan.ipynb b/notebooks/11_music/02_musegan/musegan.ipynb
index f259945..ca0ce63 100644
--- a/notebooks/11_music/02_musegan/musegan.ipynb
+++ b/notebooks/11_music/02_musegan/musegan.ipynb
@@ -139,7 +139,6 @@
     "data_binary = np.eye(N_PITCHES)[two_bars]\n",
     "data_binary[data_binary == 0] = -1\n",
     "data_binary = data_binary.transpose([0, 1, 2, 4, 3])\n",
-    "INPUT_DIM = data_binary.shape[1:]\n",
     "print(f\"Data binary shape {data_binary.shape}\")"
    ]
   },
@@ -325,7 +324,10 @@
    "outputs": [],
    "source": [
     "def Critic():\n",
-    "    critic_input = layers.Input(shape=INPUT_DIM, name=\"critic_input\")\n",
+    "    critic_input = layers.Input(\n",
+    "        shape=(N_BARS, N_STEPS_PER_BAR, N_PITCHES, N_TRACKS),\n",
+    "        name=\"critic_input\",\n",
+    "    )\n",
     "\n",
     "    x = critic_input\n",
     "\n",