snapshot...

modules/deeplearning/icing.py

@@ -426,7 +426,7 @@ class IcingIntensityNN:
 
     @tf.function
     def train_step(self, mini_batch):
-        inputs = [mini_batch[0], mini_batch[1], mini_batch[2]]
+        inputs = [mini_batch[0], mini_batch[1]]
         labels = mini_batch[2]
         with tf.GradientTape() as tape:
             pred = self.model(inputs, training=True)