Update timeseries_dataset.py

Fixed the lint errors

tf_keras/utils/timeseries_dataset.py

@@ -118,15 +118,15 @@ def timeseries_dataset_from_array(
         input_data, targets, sequence_length=sequence_length)
     for batch in dataset:
       inputs, targets = batch
-      assert np.array_equal(inputs[0], data[:sequence_length])  # First sequence: steps [0-9]
+      # First sequence: steps [0-9]
+      assert np.array_equal(inputs[0], data[:sequence_length])
       # Corresponding target: step 10
       assert np.array_equal(targets[0], data[sequence_length])
       break
     # To view the generated dataset
     for batch in dataset.as_numpy_iterator():
       input, label  = batch
       print(f"Input:{input}, target:{label}")
-    
     ```
 
     Example 3: Temporal regression for many-to-many architectures.